Abstract
The timely prediction of heart diseases with an automated system reduces the mortality rate of cardiac disease patients. However, detecting cardiac disease is one of the difficult tasks due to the small variations in the ECG signal that cannot be easily visible to the human eyes. To overcome this issue, many techniques have been introduced to effectively classify the variation in beats. However, those techniques face high error and fail to learn the spatiotemporal features, which badly affects the accuracy performance. Hence, a novel hybridized DL technique is introduced, which analyzes the spatio-temporal features and performs the heartbeat classification accurately with less error rate. At the initial stage, the signal from the raw dataset is smoothened to enhance the accuracy performance. The pre-processed samples are then balanced using the synthetic minority oversampling (SMOTE) technique to avoid over-fitting issues. Then, spatiotemporal features are extracted using a novel hybridized DL based One-Dimensional Residual Deep Convolutional Auto-Encoder (1D-RDCAE) technique. Finally, ML based extreme gradient boosting (XGB) classifier is introduced to classify the ECG heartbeats effectively. The proposed model is implemented via PYTHON and processed with the MIT-BIH arrhythmia dataset. Performance measures like accuracy, sensitivity, specificity, and false negative rate are analyzed and compared with existing techniques. In the experimental section, the proposed model obtains an accuracy of 99.9% and a specificity of 99.8%. Compared to other existing models, the proposed model shows better outcomes. Consequently, clinical cardiac care systems may benefit from this strategy as well.
Similar content being viewed by others
Explore related subjects
Discover the latest articles, news and stories from top researchers in related subjects.Data availability
Data sharing is not applicable to this article.
References
Akdağ S, Kuncan F, Kaya Y (2022) A new approach for congestive heart failure and arrhythmia classification using downsampling local binary patterns with LSTM. Turk J Electr Eng Comput Sci 30(6):2145–2164
Jiao Y, Qi H, Wu J (2022) Capsule network assisted electrocardiogram classification model for smart healthcare. Biocybern Biomed Eng 42(2):543–555
Rashed-Al-Mahfuz M, Moni MA, Lio P, Islam SMS, Berkovsky S, Khushi M, Quinn JM (2021) Deep convolutional neural networks based ECG beats classification to diagnose cardiovascular conditions. Biomed Eng Lett 11:147–162
Tripathi PM, Kumar A, Kumar M, Komaragiri R (2022) Multilevel classification and detection of cardiac arrhythmias with high-resolution superlet transform and deep convolution neural network. IEEE Trans Instrum Meas 71:1–13
Çalışkan A (2022) A new ensemble approach for congestive heart failure and arrhythmia classification using shifted one-dimensional local binary patterns with long short-term memory. Comput J 65(9):2535–2546
Sivapalan G, Nundy KK, Dev S, Cardiff B, John D (2022) ANNet: A lightweight neural network for ECG anomaly detection in IoT edge sensors. IEEE Trans Biomed Circuits Syst 16(1):24–35
Liu P, Sun X, Han Y, He Z, Zhang W, Wu C (2022) Arrhythmia classification of LSTM autoencoder based on time series anomaly detection. Biomed Signal Process Control 71:103228
Fang Y, Shi J, Huang Y, Zeng T, Ye Y, Su L, Zhu D, Huang J (2022) Electrocardiogram signal classification in the diagnosis of heart disease based on RBF neural network. Comput Math Methods Med 2022:9251225
Udawat AS, Singh P (2022) An automated detection of atrial fibrillation from single-lead ECG using HRV features and machine learning. J Electrocardiol 75:70–81
Li H, Lin Z, An Z, Zuo S, Zhu W, Zhang Z, García JDP (2022) Automatic electrocardiogram detection and classification using bidirectional long short-term memory network improved by Bayesian optimization. Biomed Signal Process Control 73:103424
Houssein EH, Abdelminaam DS, Ibrahim IE, Hassaballah M, Wazery YM (2021) A hybrid heartbeats classification approach based on marine predators algorithm and convolution neural networks. IEEE Access 9:86194–86206
Mathunjwa BM, Lin YT, Lin CH, Abbod MF, Shieh JS (2021) ECG arrhythmia classification by using a recurrence plot and convolutional neural network. Biomed Signal Process Control 64:102262
Essa E, Xie X (2021) An ensemble of deep learning-based multi-model for ECG heartbeats arrhythmia classification. IEEE Access 9:103452–103464
Dias FM, Monteiro HL, Cabral TW, Naji R, Kuehni M, Luz EJDS (2021) Arrhythmia classification from single-lead ECG signals using the inter-patient paradigm. Comput Methods Programs Biomed 202:105948
Rajendran VG, Jayalalitha S, Thalaimalaichamy M, Raj TN (2021) Classification of heart disease from ECG signals using machine learning. In: 2021 International Conference on Recent Trends on Electronics, Information, Communication & Technology (RTEICT), Bangalore, India. IEEE, pp 606–609
Nguyen TN, Nguyen TH (2021) Deep learning framework with ECG feature-based kernels for heart disease classification. Elektronika ir Elektrotechnika 27(1):48–59
Wu M, Lu Y, Yang W, Wong SY (2021) A study on arrhythmia via ECG signal classification using the convolutional neural network. Front Comput Neurosci 14:564015
Pławiak P, Acharya UR (2020) Novel deep genetic ensemble of classifiers for arrhythmia detection using ECG signals. Neural Comput Appl 32(15):11137–11161
Shaker AM, Tantawi M, Shedeed HA, Tolba MF (2020) Generalization of convolutional neural networks for ECG classification using generative adversarial networks. IEEE Access 8:35592–35605
Aziz S, Ahmed S, Alouini MS (2021) ECG-based machine-learning algorithms for heartbeat classification. Sci Rep 11(1):18738
Çınar A, Tuncer SA (2021) Classification of normal sinus rhythm, abnormal arrhythmia and congestive heart failure ECG signals using LSTM and hybrid CNN-SVM deep neural networks. Comput Methods Biomech Biomed Engin 24(2):203–214
Chen C, Hua Z, Zhang R, Liu G, Wen W (2020) Automated arrhythmia classification based on a combination network of CNN and LSTM. Biomed Signal Process Control 57:101819
Rath A, Mishra D, Panda G, Satapathy SC (2021) Heart disease detection using deep learning methods from imbalanced ECG samples. Biomed Signal Process Control 68:102820
Madan P, Singh V, Singh DP, Diwakar M, Pant B, Kishor A (2022) A hybrid deep learning approach for ECG-based arrhythmia classification. Bioengineering 9(4):152
Jyotishi D, Dandapat S (2023) An attentive spatio-temporal learning-based network for cardiovascular disease diagnosis. IEEE Trans Syst Man Cybern: Syst
Hao CZ, Nugroho H (2022) A spatio-temporal approach with transformer network for heart disease classification with 12-lead electrocardiogram signals. In: Control, instrumentation and mechatronics: theory and practice, vol 921. Springer Nature Singapore, Singapore, pp 673–684
Zhao Z (2023) Transforming ECG diagnosis: An in-depth review of transformer-based deeplearning models in cardiovascular disease detection. arXiv preprint arXiv:2306.01249
Ojha MK, Wadhwani S, Wadhwani AK, Shukla A (2022) Automatic detection of arrhythmias from an ECG signal using an auto-encoder and SVM classifier. Phys Eng Sci Med 45(2):665–674
Singh P, Sharma A (2022) Attention-based convolutional denoising autoencoder for two-lead ECG denoising and arrhythmia classification. IEEE Trans Instrum Meas 71:1–10
Wang Z, Stavrakis S, Yao B (2023) Hierarchical deep learning with generative adversarial network for automatic cardiac diagnosis from ECG signals. Comput Biol Med 155:106641
Sowmya S, Jose D (2022) Contemplate on ECG signals and classification of arrhythmia signals using CNN-LSTM deep learning model. Meas: Sens 24:100558
Liu Y, Chen J, Bao N, Gupta BB, Lv Z (2021) Survey on atrial fibrillation detection from a single-lead ECG wave for Internet of Medical Things. Computer Communications 178:245–258
Funding
No funding is provided for the preparation of the manuscript.
Author information
Authors and Affiliations
Contributions
All authors read and approved the final manuscript.
Corresponding author
Ethics declarations
Conflict of interest
Authors declare that they have no conflict of interest.
Ethical approval
This article does not contain any studies with human participants or animals performed by any of the authors.
Consent to participate
All the authors involved have agreed to participate in this submitted article.
Consent to publish
All the authors involved in this manuscript give full consent for publication of this submitted article.
Additional information
Publisher's note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Parveen, N., Gupta, M., Kasireddy, S. et al. ECG based one-dimensional residual deep convolutional auto-encoder model for heart disease classification. Multimed Tools Appl 83, 66107–66133 (2024). https://doi.org/10.1007/s11042-023-18009-7
Received:
Revised:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s11042-023-18009-7