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 条
[21]   Prediction of Coronary Heart Disease using Machine Learning: An Experimental Analysis [J].
Gonsalves, Amanda H. ;
Thabtah, Fadi ;
Mohammad, Rami Mustafa A. ;
Singh, Gurpreet .
ICDLT 2019: 2019 3RD INTERNATIONAL CONFERENCE ON DEEP LEARNING TECHNOLOGIES, 2019, :51-56
[22]   Comparing different feature selection algorithms for cardiovascular disease prediction [J].
Hasan, Najmul ;
Bao, Yukun .
HEALTH AND TECHNOLOGY, 2021, 11 (01) :49-62
[23]  
Jagtap, 2019, International Journal of Research in Engineering, Science and Management, V2, P352
[24]  
Jonnavithula S. K., 2020, AIP Conference Proceedings, V2292
[25]   Machine Learning Outperforms ACC/AHA CVD Risk Calculator in MESA [J].
Kakadiaris, Ioannis A. ;
Vrigkas, Michalis ;
Yen, Albert A. ;
Kuznetsova, Tatiana ;
Budoff, Matthew ;
Naghavi, Morteza .
JOURNAL OF THE AMERICAN HEART ASSOCIATION, 2018, 7 (22)
[26]  
Katarya Rahul, 2020, 2020 International Conference on Electronics and Sustainable Communication Systems (ICESC). Proceedings, P302, DOI 10.1109/ICESC48915.2020.9155586
[27]   Machine Learning Techniques for Heart Disease Prediction: A Comparative Study and Analysis [J].
Katarya, Rahul ;
Meena, Sunit Kumar .
HEALTH AND TECHNOLOGY, 2021, 11 (01) :87-97
[28]  
Khan M.I.H., 2020, Int. J. Comput. Appl., V176, P46
[29]   Machine learning prediction in cardiovascular diseases: a meta-analysis [J].
Krittanawong, Chayakrit ;
Virk, Hafeez Ul Hassan ;
Bangalore, Sripal ;
Wang, Zhen ;
Johnson, Kipp W. ;
Pinotti, Rachel ;
Zhang, HongJu ;
Kaplin, Scott ;
Narasimhan, Bharat ;
Kitai, Takeshi ;
Baber, Usman ;
Halperin, Jonathan L. ;
Tang, W. H. Wilson .
SCIENTIFIC REPORTS, 2020, 10 (01)
[30]   Utilizing a Two-Stage Taguchi Method and Artificial Neural Network for the Precise Forecasting of Cardiovascular Disease Risk [J].
Lin, Chia-Ming ;
Lin, Yu-Shiang .
BIOENGINEERING-BASEL, 2023, 10 (11)