A hybrid deep feature extraction and ensemble learning framework for robust multi-class ECG image classification
Abstract
ECG signals, the electrical activity of the heart, play a vital role in early identification of cardiovascular diseases (CVDs), which are currently the leading cause of death across the world. However, manual detection of ECG images can be time-consuming, prone to human error, and difficult to scale, particularly in resource-constrained environments. To address these challenges, this work proposes a hybrid classification framework to better analyze ECG images, improving reliability and efficiency. The method integrates deep learning with feature optimization and ensemble learning. An EfficientNet-B0 model is used to extract significant features from ECG images, initialized with weights trained on ImageNet data. High-dimensional features are compressed using a lightweight autoencoder to remove data redundancy while retaining significant feature pat-terns. Features obtained after compression are then applied to classify using two different machine learning algorithms: (i) Support Vector Machine with RBF kernel and (ii) Random Forest classifier. Finally, a majority-voting ensemble integrates the predictions of the CNN, SVM, and RF classifiers. Our system is tested on the Mendeley ECG Images Dataset for four classes (Normal, Myocardial Infarction (MI), Abnormal Heartbeat, and History of MI, achieving an accuracy of 92.09%, precision of 91.3%, recall of 91.1%, and F1 of 91.15%. Therefore, the proposed framework demonstrates promising classification performance with favorable computational characteristics on the evaluated dataset, highlighting its potential for computer-aided ECG image classification; however, external and prospective clinical validation is required before its clinical applicability can be established.
Related
- Multi-task deep learning framework for brain tumor classification and segmentation using ensemble feature extraction
- A Dual-Stream Feature Learning Framework with Adaptive Bio-Inspired Protozoa-Starfish Optimization for Melanoma Classification.
- A hybrid explainable deep learning framework for blood cancer classification using CNN-based feature embeddings and random forest decision models
- Multi-Platform Transcriptomic Classification of LUAD Driver Mutation Subtypes via Heterogeneous Ensemble Learning
- Development of a parallel cascaded adaptive ensemble framework with feature engineering process for the prediction of neurodegenerative disease using temporal clinical data
BibTeX
@article{nagar2026hybrid,
title = {{A hybrid deep feature extraction and ensemble learning framework for robust multi-class ECG image classification}},
author = {Sonam Nagar and Harish Sharma and Karan Verma and Sachin Singh and Kumar Shashvat},
journal = {Scientific Reports},
year = {2026},
month = oct,
doi = {10.1038/s41598-026-72309-0},
url = {https://www.nature.com/articles/s41598-026-72309-0},
note = {CC BY 4.0},
}
RIS
TY - JOUR
TI - A hybrid deep feature extraction and ensemble learning framework for robust multi-class ECG image classification
AU - Sonam Nagar
AU - Harish Sharma
AU - Karan Verma
AU - Sachin Singh
AU - Kumar Shashvat
PY - 2026
DA - 2026/10/09
JO - Scientific Reports
DO - 10.1038/s41598-026-72309-0
UR - https://www.nature.com/articles/s41598-026-72309-0
AB - ECG signals, the electrical activity of the heart, play a vital role in early identification of cardiovascular diseases (CVDs), which are currently the leading cause of death across the world. However, manual detection of ECG images can be time-consuming, prone to human error, and difficult to scale, particularly in resource-constrained environments. To address these challenges, this work proposes a hybrid classification framework to better analyze ECG images, improving reliability and efficiency. The method integrates deep learning with feature optimization and ensemble learning. An EfficientNet-B0 model is used to extract significant features from ECG images, initialized with weights trained on ImageNet data. High-dimensional features are compressed using a lightweight autoencoder to remove data redundancy while retaining significant feature pat-terns. Features obtained after compression are then applied to classify using two different machine learning algorithms: (i) Support Vector Machine with RBF kernel and (ii) Random Forest classifier. Finally, a majority-voting ensemble integrates the predictions of the CNN, SVM, and RF classifiers. Our system is tested on the Mendeley ECG Images Dataset for four classes (Normal, Myocardial Infarction (MI), Abnormal Heartbeat, and History of MI, achieving an accuracy of 92.09%, precision of 91.3%, recall of 91.1%, and F1 of 91.15%. Therefore, the proposed framework demonstrates promising classification performance with favorable computational characteristics on the evaluated dataset, highlighting its potential for computer-aided ECG image classification; however, external and prospective clinical validation is required before its clinical applicability can be established.
SN - 2045-2322
ER -