Interpretable machine learning for early neurological deterioration prediction in atrial fibrillation-related stroke.

ABSTRACT: We aimed to develop a novel prediction model for early neurological deterioration (END) based on an interpretable machine learning (ML) algorithm for atrial fibrillation (AF)-related stroke and to evaluate the prediction accuracy and feature importance of ML models. Data from multicenter prospective stroke registries in South Korea were collected. After stepwise data preprocessing, we utilized logistic regression, support vector machine, extreme gradient boosting, light gradient boosting machine (LightGBM), and multilayer perceptron models. We used the Shapley additive explanation (SHAP) method to evaluate feature importance. Of the 3,213 stroke patients, the 2,363 who had arrived at the hospital within 24 h of symptom onset and had available information regarding END were included. Of the

SUBMITTER: Kim SH 

PROVIDER: S-EPMC8523653 | biostudies-literature | 2021 Oct

REPOSITORIES: biostudies-literature