TY - JOUR
T1 - Integrating explainable machine learning and user-centric model for diagnosing cardiovascular disease
T2 - A novel approach
AU - Dharmarathne, Gangani
AU - Bogahawaththa, Madhusha
AU - Rathnayake, Upaka
AU - Meddage, D. P.P.
N1 - Publisher Copyright:
© 2024 The Author(s)
PY - 2024/9
Y1 - 2024/9
N2 - Conventional machine learning techniques in diagnosing cardiovascular disease have a limitation owing to the lack of interpretability of models. This study utilised an explainable machine learning approach to predict the likelihood of having CVD. Four machine learning models were employed for CVD diagnosis: Decision Tree (DT), K-Nearest Neighbor (KNN), Random Forest (RF), and Extreme Gradient Boost (XGB). Shapley Additive Explanations (SHAP) were used to provide reasoning for the models' predictions. Using these models and explanations, a user interface was developed to assist in diagnosing CVD. All four classification models demonstrated good accuracy in diagnosing CVD, with the KNN model showcasing the best performance (Accuracy: 71 %). SHAP provided the reasoning behind KNN predictions, and the predictive interface was developed by embedding these explanations to provide transparency behind the model's decisions.
AB - Conventional machine learning techniques in diagnosing cardiovascular disease have a limitation owing to the lack of interpretability of models. This study utilised an explainable machine learning approach to predict the likelihood of having CVD. Four machine learning models were employed for CVD diagnosis: Decision Tree (DT), K-Nearest Neighbor (KNN), Random Forest (RF), and Extreme Gradient Boost (XGB). Shapley Additive Explanations (SHAP) were used to provide reasoning for the models' predictions. Using these models and explanations, a user interface was developed to assist in diagnosing CVD. All four classification models demonstrated good accuracy in diagnosing CVD, with the KNN model showcasing the best performance (Accuracy: 71 %). SHAP provided the reasoning behind KNN predictions, and the predictive interface was developed by embedding these explanations to provide transparency behind the model's decisions.
KW - Diagnose
KW - Healthcare
KW - Heart disease
KW - Machine learning
KW - Prediction
UR - http://www.scopus.com/inward/record.url?scp=85202076214&partnerID=8YFLogxK
U2 - 10.1016/j.iswa.2024.200428
DO - 10.1016/j.iswa.2024.200428
M3 - Article
AN - SCOPUS:85202076214
SN - 2667-3053
VL - 23
JO - Intelligent Systems with Applications
JF - Intelligent Systems with Applications
M1 - 200428
ER -