Comparative analysis of explainable machine learning models for cardiovascular risk stratification using clinical data and shapley additive explanations

Heart disease remains a leading cause of mortality globally, demanding timely and reliable diagnostic support in clinical settings. This study proposes an interpretable machine learning framework that leverages Random Forest (RF) models integrated with SHapley Additive exPlanations (SHAP) and Partial Dependence Plots (PDP) to predict heart disease risk using clinical data. By addressing the limitations of “black-box” predictive systems, the framework ensures transparency and trust in decision-making. Multiple machine learning classifiers were benchmarked, with RF demonstrating high performance among the classifiers under analysis in terms of accuracy and interpretability. A streamlined, user-friendly graphical interface was developed using Streamlit to facilitate real-time risk assessment, feature-level explanations, and actionable clinical insights. The system incorporates electronic health records, utilising preprocessing and imputation strategies to enhance model robustness. Experimental evaluations demonstrated that the proposed method strikes a balance between good predictive accuracy and interpretability, making it suitable for integration into clinical workflows. This work contributes to the advancement of explainable AI in healthcare engineering, supporting clinicians in early diagnosis and preventive care for cardiovascular conditions. • An interpretable machine learning framework using Random Forest is proposed for heart disease prediction. • SHAP and Partial Dependence Plots are integrated for transparent clinical feature explanations. • KNN imputation improves data quality in Electronic Health Records for robust model training. • A real-time Streamlit GUI enables interactive risk prediction and visual explanation for clinicians. • The model achieves strong predictive performance (81.3 % accuracy) while ensuring explainability.

Authors

Institutions

Publication Details

Journal
Intelligence-Based Medicine
Published
2025-01-01
DOI
https://doi.org/10.1016/j.ibmed.2025.100286
Citations
3
Primary Topic
Machine Learning in Healthcare
Type
article
Field-Weighted Citation Impact
7.44
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Comparative analysis of explainable machine learning models for cardiovascular risk stratification using clinical data and shapley additive explanations

Rajkumar Govindarajan, Jeff Murugan, N. Komal Kumar, Bindu Kolappa Pillai Vijayammal et al.
3 citations
Intelligence-Based Medicine
Machine Learning in Healthcare
7.44
article

Comparative analysis of explainable machine learning models for cardiovascular risk stratification using clinical data and shapley additive explanations

Rajkumar Govindarajan, Jeff Murugan, N. Komal Kumar, Bindu Kolappa Pillai Vijayammal, S. Siva Sathya
article en
3 citations

Abstract

Heart disease remains a leading cause of mortality globally, demanding timely and reliable diagnostic support in clinical settings. This study proposes an interpretable machine learning framework that leverages Random Forest (RF) models integrated with SHapley Additive exPlanations (SHAP) and Partial Dependence Plots (PDP) to predict heart disease risk using clinical data. By addressing the limitations of “black-box” predictive systems, the framework ensures transparency and trust in decision-making. Multiple machine learning classifiers were benchmarked, with RF demonstrating high performance among the classifiers under analysis in terms of accuracy and interpretability. A streamlined, user-friendly graphical interface was developed using Streamlit to facilitate real-time risk assessment, feature-level explanations, and actionable clinical insights. The system incorporates electronic health records, utilising preprocessing and imputation strategies to enhance model robustness. Experimental evaluations demonstrated that the proposed method strikes a balance between good predictive accuracy and interpretability, making it suitable for integration into clinical workflows. This work contributes to the advancement of explainable AI in healthcare engineering, supporting clinicians in early diagnosis and preventive care for cardiovascular conditions. • An interpretable machine learning framework using Random Forest is proposed for heart disease prediction. • SHAP and Partial Dependence Plots are integrated for transparent clinical feature explanations. • KNN imputation improves data quality in Electronic Health Records for robust model training. • A real-time Streamlit GUI enables interactive risk prediction and visual explanation for clinicians. • The model achieves strong predictive performance (81.3 % accuracy) while ensuring explainability.

Intelligence-Based MedicineVol. 12
GS Engineering (United States) (US), Saveetha University (IN), National Institute of Technology Meghalaya (IN)
Openalex Percentile: Top 3%
Machine Learning in Healthcare
7.44
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.