A comparative study and analysis of LSTM deep neural networks for heartbeats classification

Abstract

Heart diseases and their diagnosis has become a predominant topic in Healthcare systems as the heart is one of the pivotal parts of the human body. Electrocardiogram (ECG) signal-based diagnosis and classification have been experimented with various computational techniques which have demonstrated early detection and treatment of heart disease. Deep learning (DL) is the current interest of different Healthcare applications that includes the heartbeat classification based on ECG signals. There are various studies conducted with different DL models, such as Convolutional Neural Network (CNN), Recurrent Neural Network (RNN) and Gated Recurrent Unit (GRU) for the heartbeat classification using MIT-BIH arrhythmia dataset. This paper aims to provide a comprehensive analysis of Long-Short Term Memory (LSTM) based DL models with multiple performance metrics on the MIT-BIH arrhythmia dataset for the heartbeat classification. The different variants of the LSTM DL model are proposed for the purpose of the classification. Among the variants, the bi-directional LSTM DL model shows high accuracy in the classification of Normal beats (97%), Premature ventricular contractions (PVC) beats (98%), Atrial Premature Complex (APC) beats (98%), and Paced Beats (PB) beats (99%). The comparative analysis of the bi-directional LSTM DL model with the existing works shows 95% sensitivity and 98% specificity in the classification of heartbeats. The results evidently show that the LSTM DL models are appropriate for the classification of heartbeats.

This is a preview of subscription content, access via your institution.

We’re sorry, something doesn't seem to be working properly.

Please try refreshing the page. If that doesn't work, please contact support so we can address the problem.

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6

References

  1. 1.

    Jayaraman PP, Forkan ARM, Morshed A,Haghighi PD, Kang YB. Healthcare 4.0: A review of frontiers in digital health. Wiley Interdisciplinary Reviews: Data Mining and Knowledge Discovery. 2020;10(2), e1350.

  2. 2.

    Roth GA, Mensah GA, Johnson CO, Addolorato G, Ammirati E, Baddour LM. GBD-NHLBI-JACC Global burden of cardiovascular diseases and risk factors, 1990–2019: update from the GBD 2019 study. J Am Coll Cardiol. 2020;76(25):2982–3021.

    Article  Google Scholar 

  3. 3.

    Akhtar U, Lee JW, Bilal HS, Ali T, Khan WA, Lee S. The Impact of Big Data In Healthcare Analytics. In 2020 International Conference on Information Networking (ICOIN) 2020 Jan 7 (pp. 61-63). IEEE.

  4. 4.

    Murat F, Yildirim O, Talo M, Baloglu UB, Demir Y, Acharya UR. Application of deep learning techniques for heartbeats detection using ECG signals-analysis and review. Computers in biology and medicine. 2020 Apr 8:103726.

  5. 5.

    Yao Q, Wang R, Fan X, Liu J, Li Y. Multi-class Arrhythmia detection from 12-lead varied-length ECG using Attention-based Time-Incremental Convolutional Neural Network. Information Fusion. 2020;53:174–82.

    Article  Google Scholar 

  6. 6.

    Shamshirband S, Fathi M, Dehzangi A, Chronopoulos AT, Alinejad-Rokny H. A Review on Deep Learning Approaches in Healthcare Systems: Taxonomies, Challenges, and Open Issues. J Biomed Inform. 2020 Nov 28:103627.

  7. 7.

    Faust O, Hagiwara Y, Hong TJ, Lih OS, Acharya UR. Deep learning for healthcare applications based on physiological signals: A review. Comput Methods Programs Biomed. 2018;161:1–13.

    Article  Google Scholar 

  8. 8.

    Saadatnejad S, Oveisi M, Hashemi M. LSTM-based ECG classification for continuous monitoring on personal wearable devices. IEEE J Biomed Health Inform. 2019;24(2):515–23.

    Article  Google Scholar 

  9. 9.

    Sannino G, De Pietro G. A deep learning approach for ECG-based heartbeat classification for arrhythmia detection. Futur Gener Comput Syst. 2018;86:446–55.

    Article  Google Scholar 

  10. 10.

    Hanbay K. Deep neural network based approach for ECG classification using hybrid differential features and active learning. IET Signal Proc. 2018;13(2):165–75.

    Article  Google Scholar 

  11. 11.

    Oh SL, Ng EY, San Tan R, Acharya UR. Automated diagnosis of arrhythmia using combination of CNN and LSTM techniques with variable length heart beats. Comput Biol Med. 2018;102:278–87.

    Article  Google Scholar 

  12. 12.

    Singh S, Pandey SK, Pawar U, Janghel RR. Classification of ECG arrhythmia using recurrent neural networks. Procedia computer science. 2018;132:1290–7.

    Article  Google Scholar 

  13. 13.

    Kiranyaz S, Ince T, Gabbouj M. Personalized monitoring and advance warning system for cardiac arrhythmias. Sci Rep. 2017;7(1):1–8.

    Article  Google Scholar 

  14. 14.

    Shaker AM, Tantawi M, Shedeed HA, Tolba MF. Generalization of convolutional neural networks for ECG classification using generative adversarial networks. IEEE Access. 2020;8:35592–605.

    Article  Google Scholar 

  15. 15.

    Tuncer T, Dogan S, Pławiak P, Acharya UR. Automated arrhythmia detection using novel hexadecimal local pattern and multilevel wavelet transform with ECG signals. Knowl-Based Syst. 2019;186:104923.

    Article  Google Scholar 

  16. 16.

    Huda N, Khan S, Abid R, Shuvo SB, Labib MM, Hasan T. A Low-cost, Low-energy Wearable ECG System with Cloud-Based Arrhythmia Detection. In 2020 IEEE Region 10 Symposium (TENSYMP) 2020 Jun 5 (pp. 1840-1843). IEEE.

  17. 17.

    Raj S, Ray KC. Application of variational mode decomposition and ABC optimized DAG-SVM in arrhythmia analysis. In2017 7th International Symposium on Embedded Computing and System Design (ISED) 2017 Dec 18 (pp. 1-5). IEEE.

  18. 18.

    Yildirim O, Baloglu UB, Tan RS, Ciaccio EJ, Acharya UR. A new approach for arrhythmia classification using deep coded features and LSTM networks. Comput Methods Programs Biomed. 2019;176:121–33.

    Article  Google Scholar 

  19. 19.

    Hou B, Yang J, Wang P, Yan R. LSTM-based auto-encoder model for ECG arrhythmias classification. IEEE Trans Instrum Meas. 2019;69(4):1232–40.

    Article  Google Scholar 

  20. 20.

    Ebrahimi Z, Loni M, Daneshtalab M, Gharehbaghi A. A review on deep learning methods for ECG arrhythmia classification. Expert Systems with Applications: X. 2020 Jun 20:100033.

  21. 21.

    Liu M, Kim Y. Classification of heart diseases based on ECG signals using long short-term memory. In 2018 40th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC) 2018 Jul 18 (pp. 2707-2710). IEEE.

  22. 22.

    Gers FA, Schmidhuber J, Cummins F. Learning to forget: Continual prediction with LSTM.

  23. 23.

    Goldberger A, Amaral L, Glass L, Hausdorff J, Ivanov PC, Mark R, Stanley HE. PhysioBank, PhysioToolkit, and PhysioNet: Components of a new research resource for complex physiologic signals. Circulation [Online]. 2000;101(23):e215–20.

    Google Scholar 

  24. 24.

    Moody GB, Mark RG. The impact of the MIT-BIH arrhythmia database. IEEE Engineering in Medicine and Biology Magazine. 2001;20(3):45-50. https://doi.org/10.1109/51.932724

  25. 25.

    Scipy (rftt and irftt methods)https://docs.scipy.org/doc/scipy/reference/generated/scipy.fft.rfft.html

  26. 26.

    Parikh R, Mathai A, Parikh S, Sekhar GC, Thomas R. Understanding and using sensitivity, specificity and predictive values. Indian J Ophthalmol. 2008;56(1):45.

    Article  Google Scholar 

Download references

Funding

The authors received no financial support for the research, authorship, and/or publication of this article.

Author information

Affiliations

Authors

Corresponding author

Correspondence to Srinidhi Hiriyannaiah.

Ethics declarations

Ethical approval

This paper does not contain any studies with human participants or animals performed by any of the authors.

Informed consent

This paper does not contain any studies with human participants or animals performed by any of the authors. Hence no informed consent is required.

Conflict of interest

The authors declare that they have no conflict of interest.

Additional information

This article is part of the Computer based medical systems

Rights and permissions

Reprints and Permissions

About this article

Verify currency and authenticity via CrossMark

Cite this article

Hiriyannaiah, S., G M, S., M H M, K. et al. A comparative study and analysis of LSTM deep neural networks for heartbeats classification. Health Technol. 11, 663–671 (2021). https://doi.org/10.1007/s12553-021-00552-8

Download citation

Keywords

  • Healthcare analytics
  • Deep learning
  • ECG
  • LSTM
  • DNN & Healthcare