Machine learning-based predictive model for post-stroke dementia

Zemin Wei

doi:10.1186/s12911-024-02752-4

Machine learning-based predictive model for post-stroke dementia

2024-11-11

0

OA

AI

DOI：10.1186/s12911-024-02752-4

原文链接

查看原文

摘要

En 中文

BackgroundPost-stroke dementia (PSD), a common complication, diminishes rehabilitation efficacy and affects disease prognosis in stroke patients. Many factors may be related to PSD, including demographic, comorbidities, and examination characteristics. However, most existing methods are qualitative evaluations of independent factors, which ignore the interaction amongst various factors. Therefore, the purpose of this study is to explore the applicability of machine learning (ML) methods for predicting PSD.Methods9 acceptable features were screened out by the Spearman correlation analysis and Boruta algorithm. We developed and evaluated 8 ML models: logistic regression, elastic net, k-nearest neighbors, decision tree, extreme gradient boosting, support vector machine, random forest, and multilayer perceptron.ResultsA total of 539 stroke patients were included in this study. Among the 8 models used to predict PSD, extreme gradient boosting and random forest showed the highest area under the curve (AUC) of the receiver operating characteristic curve (ROC), with values of 0.7287 and 0.7285, respectively. The most important features for predicting PSD included age, high sensitivity C-reactive protein, stroke side and location, and the occurrence of cerebral hemorrhage.ConclusionOur findings suggest that ML models, especially extreme gradient boosting, can best predict the risk of PSD.

Keyword:

Stroke

Post-stroke dementia

Boruta algorithm

Machine learning

Prediction model

Machine learning-based predictive model for post-stroke dementia

摘要

AI总结

期刊

机构