Automated Stroke Prediction Using Machine Learning: An Explainable and Exploratory Study With a Web Application for Early Intervention

被引:16
作者
Mridha, Krishna [1 ]
Ghimire, Sandesh [1 ]
Shin, Jungpil [2 ]
Aran, Anmol [1 ]
Uddin, Md. Mezbah [1 ]
Mridha, M. F. [3 ]
机构
[1] Marwadi Univ, Dept Comp Engn, Rajkot 360003, India
[2] Univ Aizu, Dept Comp Sci & Engn, Aizu Wakamatsu 9658580, Japan
[3] Amer Int Univ Bangladesh, Dept Comp Sci, Dhaka 1229, Bangladesh
关键词
Stroke (medical condition); Machine learning; Predictive models; Prediction algorithms; Machine learning algorithms; Medical services; Medical diagnostic imaging; Stroke prediction; data leakage; explainable machine learning; ANOVA test; SHAPE; LIME;
D O I
10.1109/ACCESS.2023.3278273
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Stroke is a dangerous medical disorder that occurs when blood flow to the brain is disrupted, resulting in neurological impairment. It is a big worldwide threat with serious health and economic implications. To solve this, researchers are developing automated stroke prediction algorithms, which would allow for early intervention and perhaps save lives. The number of people at risk for stroke is growing as the population ages, making precise and effective prediction systems increasingly critical. wo In a comparison examination with six well-known classifiers, the effectiveness of the proposed ML technique was explored in terms of metrics relating to both generalization capability and prediction accuracy. To give insight into the black-box machine learning models, we also studied two kinds of explainable techniques, namely SHAP and LIME, in this study. SHAP (Shapley Additive Explanations) and LIME (Local Interpretable Model-agnostic Explanations) are well-established and reliable approaches for explaining model decision-making, particularly in the medical industry. The findings of the experiment revealed that more complicated models outperformed simpler ones, with the top model obtaining almost 91% accuracy and the other models achieving 83-91% accuracy. The proposed framework, which includes global and local explainable methodologies, can aid in standardizing complicated models and gaining insight into their decision-making, which can enhance stroke care and treatment.
引用
收藏
页码:52288 / 52308
页数:21
相关论文
共 50 条
[31]   Enhancing accuracy and interpretability in EEG-based medical decision making using an explainable ensemble learning framework application for stroke prediction [J].
Bouazizi, Samar ;
Ltifi, Hela .
DECISION SUPPORT SYSTEMS, 2024, 178
[32]   Stroke Prediction Using Machine Learning in a Distributed Environment [J].
Rajora, Maihul ;
Rathod, Mansi ;
Naik, Nenavath Srinivas .
DISTRIBUTED COMPUTING AND INTERNET TECHNOLOGY, ICDCIT 2021, 2021, 12582 :238-252
[33]   Exploratory Study of Machine Learning Techniques for Supporting Failure Prediction [J].
Campos, Joao R. ;
Vieira, Marco ;
Costa, Ernesto .
2018 14TH EUROPEAN DEPENDABLE COMPUTING CONFERENCE (EDCC 2018), 2018, :9-16
[34]   An explainable machine learning-based web attack detection system for industrial IoT web application security [J].
Chakir, Oumaima ;
Sadqi, Yassine ;
Alaoui, El Arbi Abdellaoui .
INFORMATION SECURITY JOURNAL, 2024,
[35]   A study on road accident prediction and contributing factors using explainable machine learning models: analysis and performance [J].
Ahmed, Shakil ;
Hossain, Md Akbar ;
Ray, Sayan Kumar ;
Bhuiyan, Md Mafijul Islam ;
Sabuj, Saifur Rahman .
TRANSPORTATION RESEARCH INTERDISCIPLINARY PERSPECTIVES, 2023, 19
[36]   A Standard Baseline for Software Defect Prediction: Using Machine Learning and Explainable AI [J].
Bommi, Nitin Sai ;
Negi, Atul .
2023 IEEE 47TH ANNUAL COMPUTERS, SOFTWARE, AND APPLICATIONS CONFERENCE, COMPSAC, 2023, :1798-1803
[37]   Automatic COVID-19 prediction using explainable machine learning techniques [J].
Solayman S. ;
Aumi S.A. ;
Mery C.S. ;
Mubassir M. ;
Khan R. .
International Journal of Cognitive Computing in Engineering, 2023, 4 :36-46
[38]   Prediction of hydrogen uptake of metal organic frameworks using explainable machine learning [J].
Meduri, Sitaram ;
Nandanavanam, Jalaiah .
ENERGY AND AI, 2023, 12
[39]   Automated Machine Learning for the Early Prediction of the Severity of Acute Pancreatitis in Hospitals [J].
Yin, Minyue ;
Zhang, Rufa ;
Zhou, Zhirun ;
Liu, Lu ;
Gao, Jingwen ;
Xu, Wei ;
Yu, Chenyan ;
Lin, Jiaxi ;
Liu, Xiaolin ;
Xu, Chunfang ;
Zhu, Jinzhou .
FRONTIERS IN CELLULAR AND INFECTION MICROBIOLOGY, 2022, 12
[40]   Prediction of Long-Term Stroke Recurrence Using Machine Learning Models [J].
Abedi, Vida ;
Avula, Venkatesh ;
Chaudhary, Durgesh ;
Shahjouei, Shima ;
Khan, Ayesha ;
Griessenauer, Christoph J. ;
Li, Jiang ;
Zand, Ramin .
JOURNAL OF CLINICAL MEDICINE, 2021, 10 (06) :1-16