Enhanced machine learning and hybrid ensemble approaches for Coronary Heart Disease prediction
Maurice Wanyonyi1, Zakayo Ndiku Morris1, Faith Mueni Musyoka2
1Department of Mathematics and Statistics, University of Embu, Embu, Kenya.
Insights
Enhanced machine learning models significantly improve coronary heart disease (CHD) prediction accuracy. These advanced AI tools offer robust diagnostic capabilities for resource-limited settings, aiding early detection and prevention efforts.
Area of Science:
- Cardiology
- Artificial Intelligence
- Health Informatics
Background:
- Coronary heart disease (CHD) is a leading global cause of death, particularly in low- and middle-income countries with limited diagnostic resources.
- Traditional statistical models struggle with complex, high-dimensional, and imbalanced health data for accurate CHD prediction.
Purpose of the Study:
- To develop and evaluate enhanced machine learning and hybrid ensemble models for improved CHD prediction.
- Focus on enhancing diagnostic performance, interpretability, and applicability in resource-constrained environments.
Main Methods:
- Utilized a large, nationally representative dataset (253,680 individuals) from the Behavioral Risk Factor Surveillance System.
- Employed data preprocessing including normalization and Synthetic Minority Oversampling Technique (SMOTE) for balancing.
- Compared baseline models (Decision Trees, Random Forests, Gradient Boosting, SVM) against enhanced versions (ADNRT, HIRF, PGBM, ESVM) and ensemble methods (stacking, boosting, bagging, Bayesian model averaging, majority voting).
Main Results:
- Enhanced models consistently outperformed baseline models in CHD prediction.
- The Pruned Gradient Boosting Machine (PGBM) achieved the highest sensitivity (90.8%).
- The Hybrid Imbalanced Random Forest (HIRF) showed excellent calibration and balance (AUC = 0.937).
- The stacking ensemble model demonstrated the best overall performance with 87.2% accuracy, 89.6% sensitivity, 84.7% specificity, and an AUC of 0.94.
- Calibration and learning curve analyses indicated strong generalizability and minimal overfitting for ensemble models.
Conclusions:
- Hybrid ensemble machine learning models significantly outperform traditional classifiers for CHD prediction.
- These models provide high accuracy, robustness, and interpretability, crucial for clinical decision-making.
- The developed models offer a scalable framework for AI-driven diagnostics in low-resource settings, potentially revolutionizing CHD early detection and prevention.
Abstract:
Coronary heart disease (CHD) remains the leading cause of mortality worldwide, disproportionately affecting low- and middle-income countries where diagnostic resources are limited. Traditional statistical models often fail to deliver adequate predictive accuracy in complex, high-dimensional, and imbalanced health datasets. To develop and evaluate enhanced machine learning and hybrid ensemble models for the prediction of coronary heart disease, with a focus on improving diagnostic performance, interpretability, and applicability in resource-constrained settings. We utilized a nationally representative dataset of 253,680 individuals from the Behavioral Risk Factor Surveillance System. Preprocessing included normalization and balancing via the Synthetic Minority Oversampling Technique (SMOTE). Baseline models-Decision Trees, Random Forests, Gradient Boosting, and Support Vector Machines-were compared against improved versions: Adaptive Noise-Resistant Decision Tree (ADNRT), Hybrid Imbalanced Random Forest (HIRF), Pruned Gradient Boosting Machine (PGBM), and Enhanced Support Vector Machine (ESVM). Ensemble approaches (stacking, boosting, bagging, Bayesian model averaging and majority voting) were implemented and evaluated using accuracy, sensitivity, specificity, and area under the curve (AUC). Calibration and learning curves were also analyzed. Enhanced models consistently outperformed their baseline counterparts. PGBM achieved the highest sensitivity (90.8%), while HIRF demonstrated the best overall calibration and balance (AUC = 0.937; sensitivity = 88.4%; specificity = 82.9%). The stacking ensemble emerged as the best-performing model with an accuracy of 87.2%, sensitivity of 89.6%, specificity of 84.7%, and AUC of 0.94. Calibration and learning curve analyses confirmed strong generalizability and low overfitting across ensemble models. Hybrid ensemble machine learning models significantly outperform traditional classifiers in CHD prediction, offering high accuracy, robustness, and interpretability. These models present a scalable framework for implementing AI-driven diagnostic tools in low-resource environments, potentially transforming early detection and prevention of coronary heart disease.
Related Concept Videos
Coronary Artery Disease I: Introduction
Coronary Artery Disease IV: Preventive Measures
Coronary Artery Disease II: Pathophysiology
Cardiomyopathy III: Hypertrophic Cardiomyopathy
Coronary Artery Disease III: Clinical Manifestations
Coronary Artery Disease V: Interprofessional Care

