A ROBUST ENSEMBLE MACHINE LEARNING FRAMEWORK FOR MULTI-CLASS CARDIOVASCULAR DISEASE DETECTION USING ECG IMAGES
— Cardiovascular diseases (CVDs) continue to be the leading cause of death globally, creating an urgent need for faster and more accessible diagnostic tools. Although electrocardiograms (ECGs) play a crucial role in detecting cardiac abnormalities, their interpretation often requires specialized expertise and can be time-consuming. To address these challenges, this paper proposes a robust ensemble machine learning framework for multi-class cardiovascular disease detection using ECG image-derived signals. The system processes standard 12-lead ECG images by performing grayscale conversion, noise and gridline removal, and contour-based lead extraction to obtain one-dimensional cardiac signals. These signals are normalized and refined using Principal Component Analysis (PCA) to improve classification performance. Multiple machine learning algorithms, including K-Nearest Neighbors, Logistic Regression, Support Vector Machine, Random Forest, Naïve Bayes, and XGBoost, are trained and optimized, and their outputs are combined using a stacking-based ensemble approach.
The model classifies ECG data into four categories: Normal, Abnormal Heartbeat, Myocardial Infarction, and History of Myocardial Infarction. The final system is deployed through a Streamlit-based web application, enabling rapid and user-friendly cardiac screening. The results demonstrate the potential of the proposed approach in supporting efficient and scalable heart disease detection.
K.S, R. S., Raj.R, V. S., Prasandh, G. D. & .M, S. (2026). A Robust Ensemble Machine Learning Framework for Multi-Class Cardiovascular Disease Detection using ECG Images. International Journal of Science, Strategic Management and Technology, 02(03). https://doi.org/10.55041/ijsmt.v2i3.203
K.S, Rohit, et al.. "A Robust Ensemble Machine Learning Framework for Multi-Class Cardiovascular Disease Detection using ECG Images." International Journal of Science, Strategic Management and Technology, vol. 02, no. 03, 2026, pp. . doi:https://doi.org/10.55041/ijsmt.v2i3.203.
K.S, Rohit,Vijay Raj.R,Gali Prasandh, and Stanlywit .M. "A Robust Ensemble Machine Learning Framework for Multi-Class Cardiovascular Disease Detection using ECG Images." International Journal of Science, Strategic Management and Technology 02, no. 03 (2026). https://doi.org/https://doi.org/10.55041/ijsmt.v2i3.203.
[2] P. Rajpurkar et al., “Cardiologist-level arrhythmia detection with convolutional neural networks,” Nature Medicine, vol. 25, no. 1, pp. 65–69, 2019.
[3] M. Llamedo and J. P. Martínez, “Heartbeat classification using feature selection driven by database generalization criteria,” IEEE Transactions on Biomedical Engineering, vol. 58, no. 3, pp. 616–625, 2011.
[4] I. H. Witten, E. Frank, M. A. Hall, and C. J. Pal, Data Mining: Practical Machine Learning Tools and Techniques, 4th ed. Burlington, MA, USA: Morgan Kaufmann, 2016.
[5] T. Chen and C. Guestrin, “XGBoost: A scalable tree boosting system,” in Proc. 22nd ACM SIGKDD Int. Conf. Knowledge Discovery and Data Mining, 2016, pp. 785–794.
[6] C. Cortes and V. Vapnik, “Support-vector networks,” Machine Learning, vol. 20, no. 3, pp. 273–297, 1995.
[7] L. Breiman, “Random forests,” Machine Learning, vol. 45, no. 1, pp. 5–32, 2001.