Interpretable Machine Learning Model Predicting Early Neurological Deterioration in Ischemic Stroke Patients Treated with Mechanical Thrombectomy: A Retrospective Study

Abstract

Early neurologic deterioration (END) is a common and feared complication for acute ischemic stroke (AIS) patients treated with mechanical thrombectomy (MT). This study aimed to develop an interpretable machine learning (ML) model for individualized prediction to predict END in AIS patients treated with MT. The retrospective cohort of AIS patients who underwent MT was from two hospitals. ML methods applied include logistic regression (LR), random forest (RF), support vector machine (SVM), and extreme gradient boosting (XGBoost). The area under the receiver operating characteristic curve (AUC) was the main evaluation metric used. We also used Shapley Additive Explanation (SHAP) and Local Interpretable Model-Agnostic Explanations (LIME) to interpret the result of the prediction model. A total of 985 patients were enrolled in this study, and the development of END was noted in 157 patients (15.9%). Among the used models, XGBoost had the highest prediction power (AUC = 0.826, 95% CI 0.781-0.871). The Delong test and calibration curve indicated that XGBoost significantly surpassed those of the other models in prediction. In addition, the AUC in the validating set was 0.846, which showed a good performance of the XGBoost. The SHAP method revealed that blood glucose was the most important predictor variable. The constructed interpretable ML model can be used to predict the risk probability of END after MT in AIS patients. It may help clinical decision making in the perioperative period of AIS patients treated with MT.