Advertisement

Arrhythmias Classification by Integrating Stacked Bidirectional LSTM and Two-Dimensional CNN

  • Fan LiuEmail author
  • Xingshe Zhou
  • Jinli Cao
  • Zhu Wang
  • Hua Wang
  • Yanchun Zhang
Conference paper
Part of the Lecture Notes in Computer Science book series (LNCS, volume 11440)

Abstract

Classifying different types of arrhythmias based on ECG signal is an important research topic in healthcare. Traditional methods focus on extracting varieties of features from ECG and using them to build a classifier. However, ECG usually presents high inter- and intra-subjects variability both in morphology and timing, hence, it’s difficult for predesigned features to accurately depict the fluctuation patterns of each heartbeat. To this end, we propose a novel arrhythmias classification model by integrating stacked bidirectional long short-term memory network (SB-LSTM) and two-dimensional convolutional neural network (TD-CNN). Particularly, SB-LSTM mines the long-term dependencies contained in ECG from both directions to depict the overall variation trend of ECG, while TD-CNN exploits local characteristics of ECG to characterize the short-term fluctuation patterns of ECG. Moreover, we design a discrete wavelet transform (DWT) based ECG decomposition layer and a Sum Rule based intermediate classification result fusion layer, by which ECG can be analyzed from multiple time-frequency resolutions, and the classification results of our model can be more accurate. Experimental results based on MIT-BIH arrhythmia database shows that our model outperforms 3 baseline methods, achieving 99.5% of accuracy, 99.9% of sensitivity and 98.2% specificity, respectively.

Keywords

Arrhythmias classification Stacked bidirectional LSTM Convolutional neural network Wavelet decomposition Classification result fusion 

Notes

Acknowledgements

This work was partially supported by the National Natural Science Foundation of China (No. 61332013, No. 61672161), the National Key Research and Development Program of China (No. 2016YFB1001400), and the China Scholarship Council (No. 201706290110).

References

  1. 1.
    Acharya, U.R., Oh, S.L., Hagiwara, Y., et al.: A deep convolutional neural network model to classify heartbeats. Comput. Biol. Med. 89, 389–396 (2017)CrossRefGoogle Scholar
  2. 2.
    Xiong, Q., Proietti, M., Senoo, K., Lip, G.Y.H.: Asymptomatic versus symptomatic atrial fibrillation: a systematic review of age/gender differences and cardiovascular outcomes. Int. J. Cardiol. 191, 172–177 (2015)CrossRefGoogle Scholar
  3. 3.
    Huikuri, H.V., Castellanos, A., Myerburg, R.J.: Sudden death due to cardiac arrhythmias. New Engl. J. Med. 345(20), 1473–1482 (2001)CrossRefGoogle Scholar
  4. 4.
    Fuster, V., Ryden, L.E., Cannom, D.S., et al.: 2011 ACCF/AHA/HRS focused updates incorporated into the ACC/AHA/ESC 2006 guidelines for the management of patients with atrial fibrillation. J. Am. Coll. Cardiol. 57(11), e269–e367 (2011)CrossRefGoogle Scholar
  5. 5.
    ANSI/AAMI EC57: Testing and Reporting Performance Results of Cardiac Rhythm and ST Segment Measure Algorithms (2012)Google Scholar
  6. 6.
    Martis, R.J., Acharya, U.R., Adeli, H.: Current methods in electrocardiogram characterization. Comput. Biol. Med. 48, 133–149 (2014)CrossRefGoogle Scholar
  7. 7.
    Zhou, F.Y., Jin, L.P., Dong, J.: Premature ventricular contraction detection combining deep neural networks and rules inference. Artif. Intell. Med. 79, 42–51 (2017)CrossRefGoogle Scholar
  8. 8.
    Mant, J., Fitzmaurice, D.A., et al.: Accuracy of diagnosing atrial fibrillation on electrocardiogram by primary care practitioners and interpretative diagnostic software: analysis of data from screening for atrial fibrillation in the elderly (SAFE) trial. BMJ 7616, 335–380 (2007)Google Scholar
  9. 9.
    Javadi, M., Arani, S.A.A.A., Sajedin, A., Ebrahimpour, R.: Classification of ECG arrhythmia by a modular neural network based on mixture of experts and negatively correlated learning. Biomed. Signal Process. Control 8(3), 289–296 (2013)CrossRefGoogle Scholar
  10. 10.
    Chang, P.C., Lin, J.J., Hsieh, J.C., Weng, J.: Myocardial infarction classification with multi-lead ECG using hidden Markov models and Gaussian mixture models. Appl. Soft Comput. 12(10), 3165–3175 (2012)CrossRefGoogle Scholar
  11. 11.
    Kutlu, Y., Kuntalp, D.: Feature extraction for ECG heartbeats using higher order statistics of WPD coefficients. Comput. Methods Program Biomed. 105(3), 257–267 (2012)CrossRefGoogle Scholar
  12. 12.
    Elhaj, F.A., Salim, N., Harris, A.R., Swee, T.T., Ahmed, T.: Arrhythmia recognition and classification using combined linear and nonlinear features of ECG signals. Comput. Methods Program Biomed. 127, 52–63 (2016)CrossRefGoogle Scholar
  13. 13.
    Jiang, H., Zhou, R., Zhang, L., Wang, H., Zhang Y.: Sentence level topic models for associated topics extraction. World Wide Web.  https://doi.org/10.1007/s11280-018-0639-1
  14. 14.
    Peng, M., Zeng, G., Sun, Z., Huang, J., Wang, H., Tian, G.: Personalized app recommendation based on app permissions. World Wide Web 21(1), 89–104 (2018)CrossRefGoogle Scholar
  15. 15.
    Khalaf, A.F., Owis, M.I., Yassine, I.A.: A novel technique for cardiac arrhythmia classification using spectral correlation and support vector machines. Expert Syst. Appl. 42(21), 8361–8368 (2015)CrossRefGoogle Scholar
  16. 16.
    Liu, F., Zhou, X., Wang, Z., Wang, T., Ni, H., Yang, J.: Identifying obstructive sleep apnea by exploiting fine-grained BCG features based on event phase segmentation. In: IEEE BIBE, pp. 293–300 (2016)Google Scholar
  17. 17.
    Yildirim, Ö.: A novel wavelet sequence based on deep bidirectional LSTM network model for ECG signal classification. Comput. Biol. Med. 96, 189–202 (2018)CrossRefGoogle Scholar
  18. 18.
    Liu, F., Zhou, X., Wang, Z., Ni, H., Wang, T.: OSA-weigher: an automated computational framework for identifying obstructive sleep apnea based on event phase segmentation. J. Ambient Intell. Hum. Comput. (2018).  https://doi.org/10.1007/s12652-018-0787-2
  19. 19.
    Liu, F., Zhou, X., Wang, Z., Wang, T., Zhang, Y.: Identification of hypertension by mining class association rules from multi-dimensional features. In: ICPR 2018, pp. 3114–3119 (2018)Google Scholar
  20. 20.
    Yeh, Y.C., Chiou, C.W., Lin, H.J.: Analyzing ECG for cardiac arrhythmia using cluster analysis. Expert Syst. Appl. 39(1), 1000–1010 (2012)CrossRefGoogle Scholar
  21. 21.
    Goldberger, A.L., Amaral, L.A.N., Glass, L., et al.: PhysioBank, PhysioToolkit, and PhysioNet: components of a new research resource for complex physiologic signals. Circulation 101(23), e215–e220 (2000)CrossRefGoogle Scholar
  22. 22.
    Liu, F., Zhou, X., Wang, Z., et al.: A light-weight data preprocessing and integrative scheduling framework for health monitoring. In: IEEE-EMBS BHI, pp. 192–195 (2016)Google Scholar
  23. 23.
    Andreotti, F., Carr, O., Pimentel, M.A.F., Mahdi, A., Vos, M.D.: Comparing feature-based classifiers and convolutional neural networks to detect arrhythmia from short segments of ECG. Comput. Cardiol. 44, 1 (2017)Google Scholar
  24. 24.
    Krizhevsky, A., Sutskever, I., Hinton, G.E.: ImageNet classification with deep convolutional neural networks. In: NIPS, pp. 1097–1105 (2012)Google Scholar
  25. 25.
    Coşkun, M., Uçar, A., Yıldırım, Ö., et al.: Face recognition based on convolutional neural network. In: IEEE MEES, pp. 376–379 (2017)Google Scholar
  26. 26.
    Ren, S., He, K., Girshick, R., Zhang, X., Sun, J.: Object detection networks on convolutional feature maps. IEEE Trans. Pattern Anal. Mach. Intell. 39(7), 1476–1481 (2017)CrossRefGoogle Scholar
  27. 27.
    Singh, B.N., Tiwari, A.K.: Optimal selection of wavelet basis function applied to ECG signal denoising. Digit. Signal Process. 16(3), 275–287 (2006)CrossRefGoogle Scholar
  28. 28.
    Chawla, N.V., Bowyer, K.W., Hall, L.O., Kegelmeyer, W.P.: SMOTE: synthetic minority over-sampling technique. J. Artif. Intell. Res. 16, 321–357 (2002)CrossRefGoogle Scholar
  29. 29.
    Martis, R.J., Acharya, U.R., Min, L.C.: ECG beat classification using PCA, LDA, ICA and discrete wavelet transform. Biomed. Signal Process. Control 8(5), 437–448 (2013)CrossRefGoogle Scholar
  30. 30.
    Wang, Z., Zhou, X., Zhao, W., Liu, F., Ni, H., Yu, Z.: Assessing the severity of sleep apnea syndrome based on ballistocardiogram. PLoS ONE 12(4), e0175351 (2017)CrossRefGoogle Scholar
  31. 31.
    Xie, J., Wang, Z., Yu, Z., Guo, B.: Enabling efficient stroke prediction by exploring sleep related features. In: IEEE UIC, pp. 452–461 (2018)Google Scholar

Copyright information

© Springer Nature Switzerland AG 2019

Authors and Affiliations

  • Fan Liu
    • 1
    • 3
    Email author
  • Xingshe Zhou
    • 1
  • Jinli Cao
    • 2
  • Zhu Wang
    • 1
  • Hua Wang
    • 3
  • Yanchun Zhang
    • 3
  1. 1.School of Computer ScienceNorthwestern Polytechnical UniversityXi’anChina
  2. 2.Department of Computer Science and Information TechnologyLa Trobe UniversityMelbourneAustralia
  3. 3.College of Engineering and ScienceVictoria UniversityMelbourneAustralia

Personalised recommendations