Explainable and Interpretable Model for the Early Detection of Brain Stroke Using Optimized Boosting Algorithms.

Abstract

Background/Objectives: Stroke stands as a prominent global health issue, causing con-siderable mortality and debilitation. It arises when cerebral blood flow is compromised, leading to irreversible brain cell damage or death. Leveraging the power of machine learning, this paper presents a systematic approach to predict stroke patient survival based on a comprehensive set of factors. These factors include demographic attributes, medical history, lifestyle elements, and physiological metrics. Method: An effective random sampling method is proposed to handle the highly biased data of stroke. The stroke pre-diction using optimized boosting machine learning algorithms is supported with explainable AI using LIME and SHAP. This enables the models to discern intricate data patterns and establish correlations between selected features and patient survival. Results: The performance of three boosting algorithms is studied for stroke prediction, which include Gradient Boosting (GB), AdaBoost (ADB), and XGBoost (XGB) with XGB achieved the best outcome overall with a training accuracy of 96.97% and testing accuracy of 92.13%. Conclusions: Through this approach, the study seeks to uncover actionable insights to guide healthcare practitioners in devising personalized treatment strategies for stroke patients.