Interpretable Heart Disease Prediction: Optimizing Machine Learning Models via Metaheuristic Ivy Algorithm

Cardiovascular diseases remain a major global health burden, making the development of accurate and interpretable prediction models important for clinical decision support. In this study, the metaheuristic Ivy Algorithm was employed to perform two-stage hyperparameter optimization for five machine learning classifiers, namely ID3, SVM, RF, XGBoost, and LightGBM. The proposed framework was evaluated on the publicly available Cleveland and Statlog heart disease datasets using outer stratified 10-fold cross-validation. The results showed that IVYA-based optimization improved the predictive performance of all five classifiers to varying degrees. Among them, IVYA-LightGBM achieved the best overall performance, with mean AUC, Accuracy, Precision, Recall, and F1-score values of 0.945, 0.907, 0.931, 0.864, and 0.893, respectively. Paired Wilcoxon signed-rank tests based on the fold-wise results indicated that the improvements in AUC were statistically significant in most model–dataset comparisons. In addition, under consistent experimental settings, IVYA was compared with five widely used metaheuristic optimization algorithms and achieved the highest AUC, Recall, and F1-score, while requiring the shortest average runtime. To enhance model interpretability, SHAP analysis was further incorporated to quantify the contributions of different clinical features to the model predictions and improve the transparency of the prediction process. Overall, IVYA-LightGBM achieved a favorable balance among predictive performance, computational efficiency, and interpretability. Nevertheless, further validation on larger and more diverse clinical datasets is required before practical clinical application.

Authors

Institutions

Publication Details

Journal
Algorithms
Published
2026-09-14
DOI
https://doi.org/10.3390/a19090788
Primary Topic
Artificial Intelligence in Healthcare
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
article

Interpretable Heart Disease Prediction: Optimizing Machine Learning Models via Metaheuristic Ivy Algorithm

Yanhong Peng, Zhigang Ding, Yang Jiang, Cong Li et al.
Algorithms
Artificial Intelligence in Healthcare
article

Interpretable Heart Disease Prediction: Optimizing Machine Learning Models via Metaheuristic Ivy Algorithm

Yanhong Peng, Zhigang Ding, Yang Jiang, Cong Li, Rui Liang, Zihao Zuo, Jiabin Xu, Hong Jiang
article en

Abstract

Cardiovascular diseases remain a major global health burden, making the development of accurate and interpretable prediction models important for clinical decision support. In this study, the metaheuristic Ivy Algorithm was employed to perform two-stage hyperparameter optimization for five machine learning classifiers, namely ID3, SVM, RF, XGBoost, and LightGBM. The proposed framework was evaluated on the publicly available Cleveland and Statlog heart disease datasets using outer stratified 10-fold cross-validation. The results showed that IVYA-based optimization improved the predictive performance of all five classifiers to varying degrees. Among them, IVYA-LightGBM achieved the best overall performance, with mean AUC, Accuracy, Precision, Recall, and F1-score values of 0.945, 0.907, 0.931, 0.864, and 0.893, respectively. Paired Wilcoxon signed-rank tests based on the fold-wise results indicated that the improvements in AUC were statistically significant in most model–dataset comparisons. In addition, under consistent experimental settings, IVYA was compared with five widely used metaheuristic optimization algorithms and achieved the highest AUC, Recall, and F1-score, while requiring the shortest average runtime. To enhance model interpretability, SHAP analysis was further incorporated to quantify the contributions of different clinical features to the model predictions and improve the transparency of the prediction process. Overall, IVYA-LightGBM achieved a favorable balance among predictive performance, computational efficiency, and interpretability. Nevertheless, further validation on larger and more diverse clinical datasets is required before practical clinical application.

AlgorithmsVol. 19(9)
China Three Gorges Corporation (China) (CN), Nantong University (CN), Chongqing Institute of Green and Intelligent Technology (CN), Chongqing University of Technology (CN)
Peace, Justice and strong institutions
Openalex Percentile: Top 3%
Artificial Intelligence 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.