Integrating explainable machine learning and user-centric model for diagnosing cardiovascular disease: A novel approach

被引:9
作者
Dharmarathne, Gangani [1 ]
Bogahawaththa, Madhusha [1 ]
Rathnayake, Upaka [2 ]
Meddage, D. P. P. [1 ,3 ]
机构
[1] Univ New South Wales, Sch Engn & Informat Technol, Canberra, Australia
[2] Atlantic Technol Univ, Dept Civil & Environm Engn, Sligo, Ireland
[3] Ceylon Inst Artificial Intelligence Res CIAIR, Colombo, Sri Lanka
来源
INTELLIGENT SYSTEMS WITH APPLICATIONS | 2024年 / 23卷
关键词
Diagnose; Prediction; Heart disease; Machine learning; Healthcare; PREDICTION;
D O I
10.1016/j.iswa.2024.200428
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
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.
引用
收藏
页数:15
相关论文
共 68 条
[41]  
Nayeem M. J. N., 2022, EUR J ARTIF INTELL M, V1, P22
[42]  
Nikhar S., 2016, INT J ADV ENG MANAG, V2
[43]  
Nuttall Frank Q, 2015, Nutr Today, V50, P117
[44]  
Okasha N. M., 2024, Developments in the Built Environment
[45]  
Parthiban G., 2012, International Journal of Applied Information Systems (IJAIS), V3, P25, DOI [10.5120/ijais12-450593, DOI 10.5120/IJAIS12-450593]
[46]  
Pasha Syed Nawaz, 2020, IOP Conference Series: Materials Science and Engineering, V981, DOI 10.1088/1757-899X/981/2/022006
[47]  
Patil P.B., 2020, Journal of Critical Reviews, V7, P836
[48]  
Pouriyeh S, 2017, IEEE SYMP COMP COMMU, P204, DOI 10.1109/ISCC.2017.8024530
[49]   MACHINE LEARNING BASED HEART DISEASE PREDICTION SYSTEM [J].
Raja, M. Snehith ;
Anurag, M. ;
Reddy, Ch Prachetan ;
Sirisala, Nageswara Rao .
2021 INTERNATIONAL CONFERENCE ON COMPUTER COMMUNICATION AND INFORMATICS (ICCCI), 2021,
[50]  
Rajdhan A, 2021, Lecture Notes in Networks Syst, V203, P653, DOI [DOI 10.1007/978-981-16-0733-2_46, 10.1007/978-981-16-0733-246, DOI 10.1007/978-981-16-0733-246]