A Machine Learning Approach to First Pass Reperfusion in Mechanical Thrombectomy: Prediction and Feature Analysis

Abstract

IntroductionNovel machine learning (ML) methods are being investigated across medicine for their predictive capabilities while boasting increased adaptability and generalizability. In our study, we compare logistic regression with machine learning for feature importance analysis and prediction in first-pass reperfusion. MethodsWe retrospectively identified cases of ischemic stroke treated with mechanical thrombectomy (MT) at our institution from 2012–2018. Significant variables used in predictive modeling were demographic characteristics, medical history, admission NIHSS, and stroke characteristics. Outcome was binarized TICI on first pass (0-2a vs 2b-3). Shapley feature importance plots were used to identify variables that strongly affected outcomes. ResultsAccuracy for the Random Forest and SVM models were 67.1% compared to 65.8% for the logistic regression model. Brier score was lower for the Random Forest model (0.329 vs 0.342) indicating better predictive capability. Other supervised learning models performed worse than the logistic regression model, with accuracy of 56.2% for Naïve Bayes and 61.6% for XGBoost. Shapley plots for the Random Forest model showed use of aspiration, hyperlipidemia, hypertension, use of stent retriever, and time between symptom onset and catheterization as the top five predictors of first pass reperfusion. ConclusionUse of machine learning models, such as Random Forest, for the study of MT outcomes, is more accurate than logistic regression for our dataset, and identifies new factors that contribute to achieving first pass reperfusion. The benefits of machine learning, such as improved predictive capabilities, integration of new data, and generalizability, establish ML as the preferred model for studying outcomes in stroke.