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
- Yanhong Peng (ORCID: https://orcid.org/0000-0002-2162-3798)
- Zhigang Ding
- Yang Jiang (ORCID: https://orcid.org/0000-0002-5364-1421)
- Cong Li (ORCID: https://orcid.org/0000-0001-7593-1692)
- Rui Liang
- Zihao Zuo
- Jiabin Xu
- Hong Jiang
Institutions
- China Three Gorges Corporation (China) (CN)
- Nantong University (CN)
- Chongqing Institute of Green and Intelligent Technology (CN)
- Chongqing University of Technology (CN)
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