Enhanced Cardiovascular Disease Prediction: A Comparative Analysis of Machine Learning Models with SMOTE Balancing
Cardiovascular disease (CVD) is one of the leading causes of mortality worldwide, accounting for over 32% of deaths in several countries worldwide. Early detection of CVD is crucial for saving lives, as traditional diagnostic methods can be time-consuming and prone to inaccuracies. This study aims to evaluate various machine learning (ML) models for the accurate prediction of CVD. The models assessed include Extra Trees Classifier, AdaBoost, Random Forest, XGBoost, Light GBM, Gradient Boosting, and Stacking Classifier. The dataset is collected from Kaggle and preprocessed, followed by splitting it into training and testing subsets. Different types of data preprocessing techniques, including feature transformation, normalization, the Synthetic Minority Over-sampling Technique (SMOTE) for class balancing, and encoding, were applied to ensure a clean and well-structured dataset for training the model. Extensive experiments were conducted utilizing the CVD dataset, with all models trained and evaluated simultaneously for fair evaluation. Among these, the Stacking Classifier achieved the highest accuracy of 90.17%. Additionally, Random Forest (RF), LGBM Classifier, and XGBoost demonstrated strong performance, achieving accuracies of 85.68%, 83.70%, and 87.70%, respectively.
Authors
- Nahida Hoque
- Md. Shenuarin Bhuiyan (ORCID: https://orcid.org/0000-0003-0073-3071)
- Kazi Ferdous Hasan (ORCID: https://orcid.org/0009-0002-0475-0370)
- Md. Rafiqul Islam (ORCID: https://orcid.org/0000-0001-6976-4242)
- Alfe Suny
Institutions
- Dhaka University of Engineering & Technology (BD)
- East West University (BD)
Publication Details
- Published
- 2025-06-27
- DOI
- https://doi.org/10.1109/ncim65934.2025.11159887
- Primary Topic
- Artificial Intelligence in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 0.00