Enhancing cardiac diagnostics: a deep learning ensemble approach for precise ECG image classification
Abstract Cardiovascular diseases are a global health challenge that necessitates improvements in diagnostic accuracy and efficiency. This study examines the potential of deep learning (DL) models for the classification of electrocardiogram (ECG) images to assist in the identification of various card...
Saved in:
Main Authors: | , , , , , , , |
---|---|
Format: | Article |
Language: | English |
Published: |
SpringerOpen
2025-01-01
|
Series: | Journal of Big Data |
Subjects: | |
Online Access: | https://doi.org/10.1186/s40537-025-01070-4 |
Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
_version_ | 1832594695337803776 |
---|---|
author | Ahmed Alsayat Alshimaa Abdelraof Mahmoud Saad Alanazi Ayman Mohamed Mostafa Nasser Alshammari Majed Abdullah Alrowaily Hosameldeen Shabana Mohamed Ezz |
author_facet | Ahmed Alsayat Alshimaa Abdelraof Mahmoud Saad Alanazi Ayman Mohamed Mostafa Nasser Alshammari Majed Abdullah Alrowaily Hosameldeen Shabana Mohamed Ezz |
author_sort | Ahmed Alsayat |
collection | DOAJ |
description | Abstract Cardiovascular diseases are a global health challenge that necessitates improvements in diagnostic accuracy and efficiency. This study examines the potential of deep learning (DL) models for the classification of electrocardiogram (ECG) images to assist in the identification of various cardiac conditions. We initiated a two-tiered experimental framework to investigate the effectiveness of several neural network architectures in this medical application. In the first experiment, eight distinct neural network models were selected based on their top-5 accuracy on the ImageNet validation dataset and were fine-tuned using transfer learning techniques. These models were assessed using a cross-validation scheme, focusing on balanced accuracy, precision, recall, and the F1-score to evaluate their classification capabilities across four cardiac conditions: Myocardial Infarction (MI), abnormal heartbeat, historical MI, and normal ECG patterns. The second experiment extended our inquiry into the power of ensemble learning. By testing all possible combinations of the chosen models, we explored 120 ensemble configurations. The resulting analysis identified the best-performing ensemble set, which did not include the least effective model based on F1 score rankings. The most effective ensemble, composed of Inception, MobileNet, and NASNetLarge, achieved an F1 score of 0.9651 and a balanced accuracy of 0.9640, indicating a superior predictive performance. The ROC curve analysis yielded near-perfect Area Under the Curve (AUC) values for all classes, underscoring the ensemble’s proficiency in distinguishing between the specified cardiac conditions. The outcomes of this research highlight the synergistic benefit of ensembles in DL applications for medical imaging and suggest a promising approach for the early detection and diagnosis of cardiac diseases, potentially improving clinical outcomes and patient care. |
format | Article |
id | doaj-art-66a6b97234ed467d97ebe0eaef9305a9 |
institution | Kabale University |
issn | 2196-1115 |
language | English |
publishDate | 2025-01-01 |
publisher | SpringerOpen |
record_format | Article |
series | Journal of Big Data |
spelling | doaj-art-66a6b97234ed467d97ebe0eaef9305a92025-01-19T12:26:39ZengSpringerOpenJournal of Big Data2196-11152025-01-0112112810.1186/s40537-025-01070-4Enhancing cardiac diagnostics: a deep learning ensemble approach for precise ECG image classificationAhmed Alsayat0Alshimaa Abdelraof Mahmoud1Saad Alanazi2Ayman Mohamed Mostafa3Nasser Alshammari4Majed Abdullah Alrowaily5Hosameldeen Shabana6Mohamed Ezz7Department of Computer Science, College of Computer and Information Sciences, Jouf UniversityDepartment of Information Systems, MCI AcademyDepartment of Computer Science, College of Computer and Information Sciences, Jouf UniversityInformation Systems Department, College of Computer and Information Sciences, Jouf UniversityDepartment of Computer Engineering & Networks, College of Computer and Information Sciences, Jouf UniversityDepartment of Computer Science, College of Computer and Information Sciences, Jouf UniversityCollege of Medicine, Shaqra UniversityDepartment of Computer Science, College of Computer and Information Sciences, Jouf UniversityAbstract Cardiovascular diseases are a global health challenge that necessitates improvements in diagnostic accuracy and efficiency. This study examines the potential of deep learning (DL) models for the classification of electrocardiogram (ECG) images to assist in the identification of various cardiac conditions. We initiated a two-tiered experimental framework to investigate the effectiveness of several neural network architectures in this medical application. In the first experiment, eight distinct neural network models were selected based on their top-5 accuracy on the ImageNet validation dataset and were fine-tuned using transfer learning techniques. These models were assessed using a cross-validation scheme, focusing on balanced accuracy, precision, recall, and the F1-score to evaluate their classification capabilities across four cardiac conditions: Myocardial Infarction (MI), abnormal heartbeat, historical MI, and normal ECG patterns. The second experiment extended our inquiry into the power of ensemble learning. By testing all possible combinations of the chosen models, we explored 120 ensemble configurations. The resulting analysis identified the best-performing ensemble set, which did not include the least effective model based on F1 score rankings. The most effective ensemble, composed of Inception, MobileNet, and NASNetLarge, achieved an F1 score of 0.9651 and a balanced accuracy of 0.9640, indicating a superior predictive performance. The ROC curve analysis yielded near-perfect Area Under the Curve (AUC) values for all classes, underscoring the ensemble’s proficiency in distinguishing between the specified cardiac conditions. The outcomes of this research highlight the synergistic benefit of ensembles in DL applications for medical imaging and suggest a promising approach for the early detection and diagnosis of cardiac diseases, potentially improving clinical outcomes and patient care.https://doi.org/10.1186/s40537-025-01070-4Cardiovascular diseasesDeep learningECG classificationNeural network architecturesTransfer learningEnsemble learning |
spellingShingle | Ahmed Alsayat Alshimaa Abdelraof Mahmoud Saad Alanazi Ayman Mohamed Mostafa Nasser Alshammari Majed Abdullah Alrowaily Hosameldeen Shabana Mohamed Ezz Enhancing cardiac diagnostics: a deep learning ensemble approach for precise ECG image classification Journal of Big Data Cardiovascular diseases Deep learning ECG classification Neural network architectures Transfer learning Ensemble learning |
title | Enhancing cardiac diagnostics: a deep learning ensemble approach for precise ECG image classification |
title_full | Enhancing cardiac diagnostics: a deep learning ensemble approach for precise ECG image classification |
title_fullStr | Enhancing cardiac diagnostics: a deep learning ensemble approach for precise ECG image classification |
title_full_unstemmed | Enhancing cardiac diagnostics: a deep learning ensemble approach for precise ECG image classification |
title_short | Enhancing cardiac diagnostics: a deep learning ensemble approach for precise ECG image classification |
title_sort | enhancing cardiac diagnostics a deep learning ensemble approach for precise ecg image classification |
topic | Cardiovascular diseases Deep learning ECG classification Neural network architectures Transfer learning Ensemble learning |
url | https://doi.org/10.1186/s40537-025-01070-4 |
work_keys_str_mv | AT ahmedalsayat enhancingcardiacdiagnosticsadeeplearningensembleapproachforpreciseecgimageclassification AT alshimaaabdelraofmahmoud enhancingcardiacdiagnosticsadeeplearningensembleapproachforpreciseecgimageclassification AT saadalanazi enhancingcardiacdiagnosticsadeeplearningensembleapproachforpreciseecgimageclassification AT aymanmohamedmostafa enhancingcardiacdiagnosticsadeeplearningensembleapproachforpreciseecgimageclassification AT nasseralshammari enhancingcardiacdiagnosticsadeeplearningensembleapproachforpreciseecgimageclassification AT majedabdullahalrowaily enhancingcardiacdiagnosticsadeeplearningensembleapproachforpreciseecgimageclassification AT hosameldeenshabana enhancingcardiacdiagnosticsadeeplearningensembleapproachforpreciseecgimageclassification AT mohamedezz enhancingcardiacdiagnosticsadeeplearningensembleapproachforpreciseecgimageclassification |