arrow
Return

Machine learning-based predictive model for post-stroke dementia

delete2024-11-11
delete0
delete
OA
AI
Z
Zemin Wei
M
Mengqi Li
C
Chenghui Zhang
J
Jinli Miao
王文敏 (Wenmin Wang)
H
Hong Fan *
DOI:10.1186/s12911-024-02752-4delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

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.
Keywords:
Stroke
Post-stroke dementia
Boruta algorithm
Machine learning
Prediction model
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

BMC Medical Informatics and Decision Making cover
BMC Medical Informatics and Decision Making
IF:
3.8
Papers:
4.3K
Citations:
1.2W

Organization

T
tsinghua university
Scholars:
11.8W
Papers: 10.0W
Citations: 137
S
shaoxing university
Scholars:
5.7K
Papers: 3.6K
Citations: 88