Medical practitioners need to understand the critical features of ECG beats to diagnose and identify cardiovascular conditions accurately. This would be greatly facilitated by identifying the significant features of frequency components in temporal ECG wave-forms using computational methods. In this study, we have proposed a novel ECG beat classifier based on a customized VGG16-based Convolution Neural Network (CNN) that uses the time-frequency representation of temporal ECG, and a method to identify the contribution of interpretable ECG frequencies when classifying based on the SHapley Additive exPlanations (SHAP) values. We applied our model to the MIT-BIH arrhythmia dataset to classify the ECG beats and to characterise of the beats frequencies. This model was evaluated with two advanced time-frequency analysis methods. Our results indicated that for 2-4 classes our proposed model achieves a classification accuracy of 100% and for 5 classes it achieves a classification accuracy of 99.90%. We have also tested the proposed model using premature ventricular contraction beats from the American Heart Association (AHA) database and normal beats from Lobachevsky University Electrocardiography database (LUDB) and obtained a classification accuracy of 99.91% for the 5-classes case. In addition, SHAP value increased the interpretability of the ECG frequency features. Thus, this model could be applicable to the automation of the cardiovascular diagnosis system and could be used by clinicians.

Deep convolutional neural networks based ECG beats classification to diagnose cardiovascular conditions / Rashed-Al-Mahfuz, M.; Moni, M. A.; Lio, P.; Islam, S. M. S.; Berkovsky, S.; Khushi, M.; Quinn, J. M. W.. - In: BIOMEDICAL ENGINEERING LETTERS. - ISSN 2093-9868. - 11:2(2021), pp. 147-162. [10.1007/s13534-021-00185-w]

Deep convolutional neural networks based ECG beats classification to diagnose cardiovascular conditions

Lio P.;
2021

Abstract

Medical practitioners need to understand the critical features of ECG beats to diagnose and identify cardiovascular conditions accurately. This would be greatly facilitated by identifying the significant features of frequency components in temporal ECG wave-forms using computational methods. In this study, we have proposed a novel ECG beat classifier based on a customized VGG16-based Convolution Neural Network (CNN) that uses the time-frequency representation of temporal ECG, and a method to identify the contribution of interpretable ECG frequencies when classifying based on the SHapley Additive exPlanations (SHAP) values. We applied our model to the MIT-BIH arrhythmia dataset to classify the ECG beats and to characterise of the beats frequencies. This model was evaluated with two advanced time-frequency analysis methods. Our results indicated that for 2-4 classes our proposed model achieves a classification accuracy of 100% and for 5 classes it achieves a classification accuracy of 99.90%. We have also tested the proposed model using premature ventricular contraction beats from the American Heart Association (AHA) database and normal beats from Lobachevsky University Electrocardiography database (LUDB) and obtained a classification accuracy of 99.91% for the 5-classes case. In addition, SHAP value increased the interpretability of the ECG frequency features. Thus, this model could be applicable to the automation of the cardiovascular diagnosis system and could be used by clinicians.
2021
CNN; ECG; ECG beats classification; ECG frequencies; SHAP value; VGG16
01 Pubblicazione su rivista::01a Articolo in rivista
Deep convolutional neural networks based ECG beats classification to diagnose cardiovascular conditions / Rashed-Al-Mahfuz, M.; Moni, M. A.; Lio, P.; Islam, S. M. S.; Berkovsky, S.; Khushi, M.; Quinn, J. M. W.. - In: BIOMEDICAL ENGINEERING LETTERS. - ISSN 2093-9868. - 11:2(2021), pp. 147-162. [10.1007/s13534-021-00185-w]
File allegati a questo prodotto
File Dimensione Formato  
Rashed-Al-Mahfuz_Deep_2021.pdf

accesso aperto

Note: DOI 10.1007/s13534-021-00185-w
Tipologia: Versione editoriale (versione pubblicata con il layout dell'editore)
Licenza: Creative commons
Dimensione 2.06 MB
Formato Adobe PDF
2.06 MB Adobe PDF

I documenti in IRIS sono protetti da copyright e tutti i diritti sono riservati, salvo diversa indicazione.

Utilizza questo identificativo per citare o creare un link a questo documento: https://hdl.handle.net/11573/1723822
Citazioni
  • ???jsp.display-item.citation.pmc??? 9
  • Scopus 33
  • ???jsp.display-item.citation.isi??? 27
social impact