[en] Cardiovascular diseases (CVDs) have surpassed cancer and become the major cause of death worldwide. An electrocardiogram (ECG) is a non-invasive and quicker method for diagnosing abnormal heart conditions. While research has extensively focused on ECG analysis for disease classification, it has been primarily directed toward binary classification or classification of Arrhythmias, highlighting the dire need for detailed classification models. This study utilises the extensive PTB-XL database ECG records to develop a robust method for classifying various heart abnormalities. The data with unique labels is filtered through the Butterworth bandpass filter and Discrete Wavelet Transform (DWT) db-8. The R-peaks of the clean signal were used to detect the subsequent morphological features, i.e., P-QRS-T intervals and amplitudes. The feature set was balanced using the Synthetic Minority Oversampling Technique for Nominal and Continuous (SMOTE-NC) and fed into Convolutional Neural Network (CNN) and Deep Neural Network (DNN) with 5-fold cross-validation. The models classified the ECG records into one normal and four abnormal classes: Conduction Disturbance (CD), Myocardial Infarction (MI), Hypertrophy (HYP), and ST-T Changes (STTC). Performance metrics such as F1 score, recall, precision, and accuracy were evaluated for each model. The CNN model achieved a mean accuracy of 81% ± 0.03, while the DNN model achieved a mean accuracy of 84% ± 0.01. One key finding is that Hypertrophy (HYP) was consistently classified with up to 98% accuracy. Thus, the study demonstrates the effectiveness of combining advanced signal processing and deep learning techniques for precise multi-class heart disease classification using P-QRS-T features, paving the way for future real-time clinical applications.
Disciplines :
Engineering, computing & technology: Multidisciplinary, general & others
Author, co-author :
Nayyab, Rida ; Department of Biomedical Engineering and Sciences, School of Mechanical and Manufacturing Engineering, National University of Sciences and Technology (NUST), Islamabad, Pakistan
Waris, Asim; Department of Biomedical Engineering and Sciences, School of Mechanical and Manufacturing Engineering, National University of Sciences and Technology (NUST), Islamabad, Pakistan
Zaheer, Iqra; Department of Biomedical Engineering and Sciences, School of Mechanical and Manufacturing Engineering, National University of Sciences and Technology (NUST), Islamabad, Pakistan
Khan, Muhammad Jawad; Department of Electrical Engineering, School of Engineering, Prince Sattam Bin Abdul Aziz University, Saudi Arabia
Hazzazi, Fawwaz; Department of Electrical Engineering, College of Engineering, Prince Sattam Bin Abdul Aziz University, Al-Kharj, Saudi Arabia
Ijaz, Muhammad Adeel ; Department of Biomedical Engineering and Sciences, School of Mechanical and Manufacturing Engineering, National University of Sciences and Technology (NUST), Islamabad, Pakistan
Ashraf, Hassan ; Université de Liège - ULiège > Aérospatiale et Mécanique (A&M)
Gilani, Syed Omer ; Department of Electrical, Computer, and Biomedical Engineering, Abu Dhabi University, Abu Dhabi, United Arab Emirates
Language :
English
Title :
Enhancing ECG disease detection accuracy through deep learning models and P-QRS-T waveform features.
World Heart Report 2023: confronting the World’s number one killer. Geneva, Switzerland: World Heart Federation; 2023.
Lopez EO, Ballard BD, Jan A. Cardiovascular disease. Nursing. 2023. Accessed: 2024 June 09. Available: https://www.ncbi.nlm.nih.gov/books/NBK535419/
Śmigiel S, Pałczyński K, Ledziński D. ECG signal classification using deep learning techniques based on the PTB-XL dataset. Entropy (Basel). 2021;23(9):1121. https://doi.org/10.3390/e23091121 PMID: 34573746
Ko S-Y, Wang K-M, Lian W-C, Kao C-H. A portable ECG recorder. In: 2012 2nd International conference on consumer electronics, communications and networks (CECNet). 2012. p. 3063–7. https://doi.org/10.1109/cecnet.2012.6201460
Kamaruddin NH, Murugappan M, Mohammad Iqbal O. Early prediction of cardiovascular diseases using ECG signal: review. In: 2012 IEEE student conference on research and development (SCOReD). 2012. p. 48–53. https://doi.org/10.1109/scored.2012.6518609
Kirti, Sohal H, Jain S. Interpretation of cardio vascular diseases using electrocardiogram: a study. In: 2018 Fifth International conference on parallel, distributed and grid computing (PDGC). 2018. p. 159–64. https://doi.org/10.1109/pdgc.2018.8745999
Liu X, Wang H, Li Z, Qin L. Deep learning in ECG diagnosis: a review. Knowledge-Based Syst. 2021;227:107187. https://doi.org/10.1016/j.knosys.2021.107187
Śmigiel S, Pałczyński K, Ledziński D. Deep learning techniques in the classification of ECG signals using R-peak detection based on the PTB-XL dataset. Sensors (Basel). 2021;21(24):8174. https://doi.org/10.3390/s21248174 PMID: 34960267
Rawi AA, Elbashir MK, Ahmed AM. Deep learning models for multilabel ECG abnormalities classification: a comparative study using TPE optimization. J Intel Syst. 2023;32(1). https://doi.org/10.1515/JISYS-2023-0002
Pałczyński K, Śmigiel S, Ledziński D, Bujnowski S. Study of the few-shot learning for ECG classification based on the PTB-XL dataset. Sensors (Basel). 2022;22(3):904. https://doi.org/10.3390/s22030904 PMID: 35161650
Tadejko P, Rakowski W. Mathematical Morphology Based ECG Feature Extraction for the Purpose of Heartbeat Classification. In: 6th International Conference on Computer Information Systems and Industrial Management Applications (CISIM’07). 2007. p. 322–7. https://doi.org/10.1109/cisim.2007.47
Jovic A, Bogunovic N. Feature extraction for ECG time-series mining based on chaos theory. In: 29th International conference on information technology interfaces. 2007. p. 63–8. https://doi.org/10.1109/iti.2007.4283745
Chouhan VS, Mehta SS. Detection of QRS complexes in 12-lead ECG using adaptive quantized threshold. Inter J Comp Sci Netw Secur. 2008;8(1).
Olvera FE. Electrocardiogram waveform feature extraction using the matched filter. Accessed 2024 July 25. https://scholar.google.com/scholar?hl=en&as_sdt=0%2C5&q=F.E.%20Olvera%2C%20%E2%80%9CElectrocardiogram%20Waveform%20Feature%20Extraction%20Using%20the%20Matched%20Filter%2C%E2%80%9D%202006.&btnG=
Naseer N, Nazeer H. Classification of normal and abnormal ECG signals based on their PQRST intervals. In: 2017 International conference on mechanical, system and control engineering (ICMSC). 2017. p. 388–91. https://doi.org/10.1109/icmsc.2017.7959507
Srivastava VK, Prasad D. Dwt-based feature extraction from ecg Signal. Am J Eng Res. 2013. Accessed: 2024 July 25. www.ajer.us
Haddadi R, Abdelmounim E, El Hanine M, Belaguid A. Discrete wavelet transform based algorithm for recognition of QRS complexes. In: 2014 International conference on multimedia computing and systems (ICMCS). 2014. p. 375–9. https://doi.org/10.1109/icmcs.2014.6911261
Complete pan tompkins implementation ECG QRS detector - file exchange - MATLAB central. Accessed 2024 June 9. https://www.mathworks.com/matlabcentral/fileexchange/45840-complete-pan-tompkins-implementation-ecg-qrs-detector
Rumelhart DE, Hinton GE, Williams RJ. Learning representations by back-propagating errors. Nature. 1986;323(6088):533–6. https://doi.org/10.1038/323533a0
Feyisa DW, Debelee TG, Ayano YM, Kebede SR, Assore TF. Lightweight multireceptive field CNN for 12-lead ECG signal classification. 2022. https://doi.org/10.1155/2022
Hannun AY, Rajpurkar P, Haghpanahi M, Tison GH, Bourn C, Turakhia MP, et al. Cardiologist-level arrhythmia detection and classification in ambulatory electrocardiograms using a deep neural network. Nat Med. 2019;25(1):65–9. https://doi.org/10.1038/s41591-018-0268-3 PMID: 30617320
Avanzato R, Beritelli F. Automatic ECG diagnosis using convolutional neural network. Electronics. 2020;9(6):951. https://doi.org/10.3390/electronics9060951
Hanbay K. Deep neural network based approach for ECG classification using hybrid differential features and active learning. IET signal process. 2019;13(2):165–75. https://doi.org/10.1049/iet-spr.2018.5103
Bhoi AK, Sherpa KS, Khandelwal B. Ischemia and arrhythmia classification using time-frequency domain features of QRS complex. Procedia Comp Sci. 2018;132:606–13. https://doi.org/10.1016/j.procs.2018.05.014
Marinho LB, Nascimento N de MM, Souza JWM, Gurgel MV, Rebouças Filho PP, de Albuquerque VHC. A novel electrocardiogram feature extraction approach for cardiac arrhythmia classification. Fut Gener Comp Syst. 2019;97:564–77. https://doi.org/10.1016/j.future.2019.03.025
Anbalagan T, Nath MK. AF identification from time–frequency analysis of ECG signal using deep neural networks. IEEE Sens Lett. 2024;8(9):1–4. https://doi.org/10.1109/lsens.2024.3435009
Anbalagan T, Nath MK, Anbalagan A. Detection of atrial fibrillation from ECG signal using efficient feature selection and classification. Circuits Syst Signal Process. 2024;43(9):5782–808. https://doi.org/10.1007/S00034-024-02727-W
Sharma A, Garg N, Patidar S, San Tan R, Acharya UR. Automated pre-screening of arrhythmia using hybrid combination of fourier-bessel expansion and LSTM. Comput Biol Med. 2020;120:103753. https://doi.org/10.1016/j.compbiomed.2020.103753 PMID: 32421653
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. Inform Fusion. 2020;53:174–82. https://doi.org/10.1016/j.inffus.2019.06.024
Hasan NI, Bhattacharjee A. Deep learning approach to cardiovascular disease classification employing modified ECG signal from empirical mode decomposition. Biomed Signal Process Control. 2019;52:128–40. https://doi.org/10.1016/j.bspc.2019.04.005
LeCun Y, Bengio Y, Hinton G. Deep learning. Nature. 2015;521(7553):436–44. https://doi.org/10.1038/nature14539 PMID: 26017442
Asgharzadeh-Bonab A, Amirani MC, Mehri A. Spectral entropy and deep convolutional neural network for ECG beat classification. Biocyber Biomed Eng. 2020;40(2):691–700. https://doi.org/10.1016/j.bbe.2020.02.004
Mohonta SC, Motin MA, Kumar DK. Electrocardiogram based arrhythmia classification using wavelet transform with deep learning model. Sens Bio-Sens Res. 2022;37:100502. https://doi.org/10.1016/j.sbsr.2022.100502
Oktivasari P, Hasyim M, Amy HS, Freddy H, Suprijadi. A simple real-time system for detection of normal and myocardial ischemia in the ST segment and T wave ECG signal. In: 2019 International conference on information and communications technology (ICOIACT). 2019. p. 327–31. https://doi.org/10.1109/icoiact46704.2019.8938461