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.

Plos One
|December 26, 2025
PubMed

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.

Related Concept Videos

Coronary Artery Disease I: Introduction01:30

Coronary Artery Disease I: Introduction

Coronary Artery Disease (CAD): An Overview with Scientific InsightsCoronary Artery Disease (CAD), often referred to as C-A-D, is a prevalent blood vessel disorder classified under the broader category of atherosclerosis. Atherosclerosis is a pathological process characterized by the hardening and narrowing of arteries due to the accumulation of atherosclerotic plaques. These plaques are composed of cholesterol, fatty substances, inflammatory cells, calcium, and fibrin, reducing blood flow to...
846
Coronary Artery Disease IV: Preventive Measures01:26

Coronary Artery Disease IV: Preventive Measures

Effective preventive measures for coronary artery disease (CAD) focus on controlling modifiable risk factors, including cholesterol abnormalities and lifestyle changes.Cholesterol ManagementFirst, the Mediterranean diet and the American Heart Association advocate for maintaining low-density lipoprotein (LDL) cholesterol levels below 100 mg/dL, with a more stringent recommendation of below 70 mg/dL for individuals at high risk. LDL cholesterol, often termed "bad cholesterol," can lead to the...
564
Coronary Artery Disease II: Pathophysiology01:26

Coronary Artery Disease II: Pathophysiology

Coronary Artery Disease (CAD) originates from a series of events that impair the function of coronary arteries, the blood vessels responsible for delivering oxygen-rich blood to the heart muscle. The pathophysiology of CAD is closely linked to atherosclerosis, a chronic inflammatory and lipid-driven condition affecting the vascular endothelium.1. Endothelial DamageThe process begins with damage to the vascular endothelium, which serves as a protective barrier between the blood and the vessel...
336
Cardiomyopathy III: Hypertrophic Cardiomyopathy01:29

Cardiomyopathy III: Hypertrophic Cardiomyopathy

Hypertrophic cardiomyopathy, or HCM, is an autosomal dominant genetic disorder characterized by asymmetric left ventricular hypertrophy without ventricular dilation. It is more common in men and is typically diagnosed in young, athletic adults.EtiologyHCM is primarily genetic and is caused by mutations in genes encoding sarcomeric proteins. Researchers have identified over 1400 mutations across at least 11 different genes. Among these, the most frequently occurring mutations are found in the...
360
Coronary Artery Disease III: Clinical Manifestations01:30

Coronary Artery Disease III: Clinical Manifestations

Coronary Artery Disease (CAD) is a primary health risk worldwide, leading to significant morbidity and mortality. The condition arises from the buildup of atherosclerotic plaques within the coronary arteries, resulting in diminished blood supply to the heart muscle.The clinical manifestations of CAD vary widely, from asymptomatic stages to severe, life-threatening conditions. Understanding these manifestations is crucial for early diagnosis and effective management.Angina Pectoris: The Warning...
299
Coronary Artery Disease V: Interprofessional Care01:27

Coronary Artery Disease V: Interprofessional Care

Interprofessional care for coronary artery disease includes pharmacological therapy and revascularization procedures.Pharmacological therapy for Coronary Artery Disease (CAD) aims to manage symptoms, prevent complications, and improve patient outcomes through various classes of medications:Antiplatelet Agents:Aspirin and Clopidogrel: These medications inhibit platelet aggregation, preventing blood clots, which is crucial for avoiding heart attacks and strokes. Doctors often prescribe these...
203