Back

A Consensus-Driven Stacking Ensemble Framework for Interpretable Cardiovascular Risk Prediction and Clinical Deployment

Sozol, S. S.; Dev Nath, B. C.; Fahim, F. M. S.; Suzana, N. N.; Mirza, J. F.; Ahmmed, S.; Zohra, F.-T.; Zafr, A. H. A.; Uddin, M. N.; Mondal, M. R. H.; Hoque, A. S. M. L.

2026-05-26 health informatics
10.64898/2026.05.18.26352989 medRxiv
Show abstract

Machine learning (ML) is being considered to help diagnose cardiovascular diseases (CVD). Still, challenges like inconsistent and limited datasets, limited infrastructure, and global inequalities lead to the need for a reliable and practicable ML solution. This paper presents an ML-driven framework for predicting CVD risk scores and classifying status. Several data preprocessing techniques, including multiple imputation by chained equations (MICE), outlier removal, are considered. In addition, hyperparameter tuning is performed with the GridSearchCV tuning technique. Moreover, a consensus-driven five-feature selection method is applied to identify optimal predictors. The dataset used in this study contains healthcare records related to future CVD risk scores, comprising 1,529 patient records with 22 features. The optimized stacked ensemble model is applied to the dataset and achieves a cross-validated coefficient of determination value of 98.13% for CVD risk score regression. Comparative evaluation with other ML models confirmed improved accuracy, efficiency, and interpretability. The explainable AI technique SHAP is applied to interpret predictions and highlight key risk factors. Moreover, a deployment-ready web platform with multi-role access has been developed that demonstrates clinical applicability. The proposed framework offers a reliable and interpretable tool for early detection of CVD and personalized risk assessment. In the future, this work can be extended to integrate longitudinal data, medical imaging, and deep learning to improve generalizability and strengthen real-world impact.

Matching journals

The top 7 journals account for 50% of the predicted probability mass.

1
Computers in Biology and Medicine
128 papers in training set
Top 0.1%
10.4%
2
Journal of Biomedical Informatics
47 papers in training set
Top 0.1%
9.5%
3
IEEE Journal of Biomedical and Health Informatics
37 papers in training set
Top 0.1%
6.6%
4
PLOS Digital Health
106 papers in training set
Top 0.8%
6.6%
5
BMC Medical Informatics and Decision Making
43 papers in training set
Top 0.3%
6.6%
6
PLOS ONE
5266 papers in training set
Top 29%
5.4%
7
Scientific Reports
3612 papers in training set
Top 17%
5.4%
50% of probability mass above
8
Frontiers in Digital Health
24 papers in training set
Top 0.2%
4.8%
9
Computer Methods and Programs in Biomedicine
28 papers in training set
Top 0.2%
3.2%
10
npj Digital Medicine
118 papers in training set
Top 2%
3.2%
11
Artificial Intelligence in Medicine
17 papers in training set
Top 0.2%
2.6%
12
Informatics in Medicine Unlocked
22 papers in training set
Top 0.5%
2.1%
13
European Heart Journal - Digital Health
18 papers in training set
Top 0.5%
1.9%
14
JAMIA Open
42 papers in training set
Top 0.8%
1.9%
15
Frontiers in Artificial Intelligence
20 papers in training set
Top 0.3%
1.7%
16
Patterns
78 papers in training set
Top 2%
1.5%
17
International Journal of Medical Informatics
26 papers in training set
Top 0.8%
1.5%
18
Biomedical Signal Processing and Control
22 papers in training set
Top 0.4%
1.5%
19
Biology Methods and Protocols
61 papers in training set
Top 2%
1.1%
20
Journal of the American Medical Informatics Association
71 papers in training set
Top 2%
1.1%
21
BMC Medical Research Methodology
47 papers in training set
Top 1%
1.1%
22
Diagnostics
50 papers in training set
Top 2%
1.0%
23
Nature Communications
5641 papers in training set
Top 54%
1.0%
24
Expert Systems with Applications
11 papers in training set
Top 0.3%
1.0%
25
Frontiers in Cardiovascular Medicine
53 papers in training set
Top 2%
1.0%
26
Sensors
43 papers in training set
Top 1%
0.9%
27
Journal of Medical Internet Research
87 papers in training set
Top 3%
0.8%
28
Communications Medicine
113 papers in training set
Top 5%
0.8%
29
Healthcare
17 papers in training set
Top 1%
0.8%
30
International Journal of Environmental Research and Public Health
128 papers in training set
Top 6%
0.6%