Objectives: Predicting stroke mortality is crucial for personalized care. This study aims to design and evaluate a machine learning model to predict one-year mortality after a stroke.
Materials and methods: Data from the National Multiethnic Stroke Registry was utilized. Eight machine learning (ML) models were trained and evaluated using various metrics. SHapley Additive exPlanations (SHAP) analysis was used to identify the influential predictors.
Results: The final analysis included 9840 patients diagnosed with stroke were included in the study. The XGBoost algorithm exhibited optimal performance with high accuracy (94.5%) and AUC (87.3%). Core predictors encompassed National Institutes of Health Stroke Scale (NIHSS) at admission, age, hospital length of stay, mode of arrival, heart rate, and blood pressure. Increased NIHSS, age, and longer stay correlated with higher mortality. Ambulance arrival and lower diastolic blood pressure and lower body mass index predicted poorer outcomes.
Conclusions: This model's predictive capacity emphasizes the significance of NIHSS, age, hospital stay, arrival mode, heart rate, blood pressure, and BMI in stroke mortality prediction. Specific findings suggest avenues for data quality enhancement, registry expansion, and real-world validation. The study underscores machine learning's potential for early mortality prediction, improving risk assessment, and personalized care. The potential transformation of care delivery through robust ML predictive tools for Stroke outcomes could revolutionize patient care, allowing for personalized plans and improved preventive strategies for stroke patients. However, it is imperative to conduct prospective validation to evaluate its practical clinical effectiveness and ensure its successful adoption across various healthcare environments.
Keywords: Hemorrhagic stroke; Ischemic stroke; Machine learning; Mortality; Prognosis; Stroke.
© 2024 The Authors.