Optimizing Stroke Risk Prediction: A Machine Learning Pipeline Combining ROS-Balanced Ensembles and XAI
By: A S M Ahsanul Sarkar Akib, Raduana Khawla, Abdul Hasib
Potential Business Impact:
Predicts stroke risk with 99% accuracy.
Stroke is a major cause of death and permanent impairment, making it a major worldwide health concern. For prompt intervention and successful preventative tactics, early risk assessment is essential. To address this challenge, we used ensemble modeling and explainable AI (XAI) techniques to create an interpretable machine learning framework for stroke risk prediction. A thorough evaluation of 10 different machine learning models using 5-fold cross-validation across several datasets was part of our all-inclusive strategy, which also included feature engineering and data pretreatment (using Random Over-Sampling (ROS) to solve class imbalance). Our optimized ensemble model (Random Forest + ExtraTrees + XGBoost) performed exceptionally well, obtaining a strong 99.09% accuracy on the Stroke Prediction Dataset (SPD). We improved the model's transparency and clinical applicability by identifying three important clinical variables using LIME-based interpretability analysis: age, hypertension, and glucose levels. Through early prediction, this study highlights how combining ensemble learning with explainable AI (XAI) can deliver highly accurate and interpretable stroke risk assessment. By enabling data-driven prevention and personalized clinical decisions, our framework has the potential to transform stroke prediction and cardiovascular risk management.
Similar Papers
Comparative Analysis of Stroke Prediction Models Using Machine Learning
Machine Learning (CS)
Helps doctors guess who might get a stroke.
Machine learning algorithms to predict stroke in China based on causal inference of time series analysis
Quantitative Methods
Finds stroke risks by watching health changes.
Machine Learning-Based Model for Postoperative Stroke Prediction in Coronary Artery Disease
Machine Learning (CS)
Predicts heart surgery stroke risk for patients.