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 条
[1]  
Abdeldjouad F.Z., 2020, IMPACT DIGITAL TECHN, V12157, DOI [10.1007/978-3-030-51517-1_26, DOI 10.1007/978-3-030-51517-1_26]
[2]   Machine learning-based heart disease diagnosis: A systematic literature review [J].
Ahsan, Md Manjurul ;
Siddique, Zahed .
ARTIFICIAL INTELLIGENCE IN MEDICINE, 2022, 128
[3]  
Aiosa G. V., 2023, 2023 IEEE 19 INT C E
[4]   EXplainable AI for Decision Support to Obesity Comorbidities Diagnosis [J].
Aiosa, Grazia V. ;
Palesi, Maurizio ;
Sapuppo, Francesca .
IEEE ACCESS, 2023, 11 :107767-107782
[5]  
Alotaibi FS, 2019, INT J ADV COMPUT SC, V10, P261
[6]  
Ashwath S., 2024, P 2024 11 INT C SIGN, P199
[7]  
Barik Shekharesh, 2020, Advances in Electrical Control and Signal Systems. Select Proceedings of AECSS 2019. Lecture Notes in Electrical Engineering (LNEE 665), P879, DOI 10.1007/978-981-15-5262-5_67
[8]   Effective Heart Disease Prediction Using Machine Learning Techniques [J].
Bhatt, Chintan M. ;
Patel, Parth ;
Ghetia, Tarang ;
Mazzeo, Pier Luigi .
ALGORITHMS, 2023, 16 (02)
[9]  
Boukhatem C., 2022, 2022 ADV SCI ENG TEC, P1
[10]  
Chakraborty P., 2024, 2024 INT C DISTR COM