Stacked Ensemble Model With Explainable AI for Early Detection of Heart Disease

Heart disease (HD) is still one of the most common causes of death around the world. Early detection is very important, but it is often hard to do because the symptoms are not specific and the models are not very clear. We propose a two-layer stacked ensemble that combines four base learners-Support Vector Machine, K-Nearest Neighbors, Naïve Bayes, and Decision Tree-with Logistic Regression as a meta-learner. This ensemble was trained on a balanced sample from five public HD datasets. The model outperforms standard baselines with 93.69% accuracy, 93% F1-score, and 94.7% AUC. To ensure that predictions are clinically interpretable, we offer global explainability through SHAP summaries, feature importance, and stacked plots, as well as ELi5 permutation-importance plots. Both identify ST slope and chest pain type as the most significant predictors, consistent with clinical understanding. We use LIME, ELi5 text explanations, SHAP force plots, and waterfall plots to illustrate how different features of a person affect their risk. We utilize SHAP interaction values to examine feature interactions and confirm nonlinear relationships using 1D and 2D partial-dependence plots. Finally, we generate counterfactual explanations to demonstrate minor, plausible changes based on domain knowledge that would affect a prediction. This is useful both for auditability and for conversations around patient-centred care. The framework combines global and local explainable artificial intelligence with strong predictive power to improve decision support and accuracy in HD risk assessment.

Authors

Institutions

Publication Details

Journal
Analytical Science Advances
Published
2026-03-03
DOI
https://doi.org/10.1002/ansa.70072
Primary Topic
Machine Learning in Healthcare
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Stacked Ensemble Model With Explainable AI for Early Detection of Heart Disease

Mayeen Uddin Khandaker, Nazmun Nahar, Md. Hasan Imam, H. J. Woo et al.
Analytical Science Advances
Machine Learning in Healthcare
article

Stacked Ensemble Model With Explainable AI for Early Detection of Heart Disease

Mayeen Uddin Khandaker, Nazmun Nahar, Md. Hasan Imam, H. J. Woo, Hamid Osman, Joy Bhowmik, Sanjatul Hasan Siam, Ayesha Nasrin Ripa
article en

Abstract

Heart disease (HD) is still one of the most common causes of death around the world. Early detection is very important, but it is often hard to do because the symptoms are not specific and the models are not very clear. We propose a two-layer stacked ensemble that combines four base learners-Support Vector Machine, K-Nearest Neighbors, Naïve Bayes, and Decision Tree-with Logistic Regression as a meta-learner. This ensemble was trained on a balanced sample from five public HD datasets. The model outperforms standard baselines with 93.69% accuracy, 93% F1-score, and 94.7% AUC. To ensure that predictions are clinically interpretable, we offer global explainability through SHAP summaries, feature importance, and stacked plots, as well as ELi5 permutation-importance plots. Both identify ST slope and chest pain type as the most significant predictors, consistent with clinical understanding. We use LIME, ELi5 text explanations, SHAP force plots, and waterfall plots to illustrate how different features of a person affect their risk. We utilize SHAP interaction values to examine feature interactions and confirm nonlinear relationships using 1D and 2D partial-dependence plots. Finally, we generate counterfactual explanations to demonstrate minor, plausible changes based on domain knowledge that would affect a prediction. This is useful both for auditability and for conversations around patient-centred care. The framework combines global and local explainable artificial intelligence with strong predictive power to improve decision support and accuracy in HD risk assessment.

Analytical Science AdvancesVol. 7(1)
Taif University (SA), Daffodil International University (BD), Noakhali Science and Technology University (BD), University of Malaya (MY), International University of Business Agriculture and Technology (BD), Korea University (JP), Sunway University (MY)
Peace, Justice and strong institutions
Openalex Percentile: Top 39%
Machine Learning in Healthcare
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.