Explainable machine learning for long-term outcome prediction in two-center stroke patients after intravenous thrombolysis

Abstract

ObjectiveNeurological outcome prediction in patients with ischemic stroke is very critical in treatment strategy and post-stroke management. Machine learning techniques with high accuracy are increasingly being developed in the medical field. We studied the application of machine learning models to predict long-term neurological outcomes in patients with after intravenous thrombolysis.MethodsA retrospective cohort study was performed to review all stroke patients with intravenous thrombolysis. Patients with modified Rankin Score (mRs) less than two at three months post-thrombolysis were considered as good outcome. The clinical features between stroke patients with good and with poor outcomes were compared using three different machine learning models (Random Forest, Support Vector Machine and Logistic Regression) to identify which performed best. Two datasets from the other stroke center were included accordingly for external verification and performed with explainable AI models.ResultsOf the 488 patients enrolled in this study, and 374 (76.6%) patients had favorable outcomes. Patients with higher mRs at 3 months had increased systolic pressure, blood glucose, cholesterol (TC), and 7-day National Institute of Health Stroke Scale (NIHSS) score compared to those with lower mRs. The predictability and the areas under the curves (AUC) for the random forest model was relatively higher than support vector machine and LR models. These findings were further validated in the external dataset and similar results were obtained. The explainable AI model identified the risk factors as well.ConclusionExplainable AI model is able to identify NIHSS_Day7 is independently efficient in predicting neurological outcomes in patients with ischemic stroke after intravenous thrombolysis.