Open AccessPrognosis Prediction of Stroke based on Machine Learning and Explanation Model
Author(s) -
Qiuli Qin,
Xuehan Zhou,
Yong Jiang
Publication year - 2021
Publication title -
international journal of computers, communications and control
Language(s) - English
Resource type - Journals
SCImago Journal Rank - 0.422
H-Index - 33
eISSN - 1841-9844
pISSN - 1841-9836
DOI - 10.15837/ijccc.2021.2.4108
Subject(s) - random forest , support vector machine , logistic regression , atrial fibrillation , decision tree , artificial intelligence , feature (linguistics) , computer science , stroke (engine) , machine learning , glasgow coma scale , sample (material) , medicine , surgery , engineering , mechanical engineering , linguistics , philosophy , chemistry , chromatography
The prognosis prediction of stroke is of great significance to its prevention and treatment. This paper used machine learning to predict stroke prognosis, and use SHAP method to make feature importance and single sample analysis. Firstly, feature engineering, use Borderline-SMOTE algorithm to deal with data imbalance, use Support Vector Machine(SVM) to build a prognostic prediction model, and use Random Forest(RF), Decision Tree(DT), Logistic Regression(LR) for comparative analysis, and find the performance of SVM after feature engineering better than other models, the accuracy, specificity, F1 score, AUC value reach 0.8306, 0.8356, 0.8415 and 0.9140. Then, the model was further analyzed for explainability, and it was found that the top three causes of the disease were Glasgow Coma Score, NIHSS and atrial fibrillation. Finally, try to analysis a single sample, which is performed to determine that the patient is a low-risk patient, and suffering from atrial fibrillation is the largest potential risk factor for the patient.