Development of machine learning-based models for predicting risk factors in acute cerebral infarction patients: a clinical retrospective study
- PMID: 39217304
- PMCID: PMC11365171
- DOI: 10.1186/s12883-024-03818-6
Development of machine learning-based models for predicting risk factors in acute cerebral infarction patients: a clinical retrospective study
Abstract
Objectives: The aim of this study was to develop machine learning-based models for predicting acute cerebral infarction (ACI) in patients.
Methods: We extracted the data of ACI patients and non-ACI patients (as control) from two hospitals. The Lasso algorithm was employed to select the most crucial features associated with ACI. Five machine learning algorithms-based models were trained, which was performed with 10-fold cross-validation. Then, the area under the receiver operating characteristic curve (AUC), accuracy, and F1-score were calculated in the training models. Accordingly, the training models with excellent performance was selected as the final predictive model. The relative importance of variables was analyzed and ranked.
Results: A total of 150 patients were diagnosed with ACI (50.00%), with a higher proportion of males (70.67% vs. 44.00%) compared to the non-ACI patients. The logistic regression model exhibited a good performance in predicting ACI in the training set, as evidenced by its highest AUC, accuracy, sensitivity, and F1-score. Furthermore, feature importance analysis showed that blood glucose, gender, smoking history, serum homocysteine, folic acid, and C-reactive protein were the top six crucial variables of the logistic regression.
Conclusions: In our work, the ACI risk prediction model developed by the logistic regression exhibited excellent performance. This could contribute to the identification of risk variables for ACI patients and enables clinicians timely and effective interventions.
Keywords: Acute cerebral infarction; Machine learning; Prediction model; Risk factors.
© 2024. The Author(s).
Conflict of interest statement
The authors declare no competing interests.
Figures




References
-
- Suzuki H, Kanamaru H, Kawakita F, Asada R, Fujimoto M, Shiba M. Cerebrovascular pathophysiology of delayed cerebral ischemia after aneurysmal subarachnoid hemorrhage. Histol Histopathol. 2021;36(2):143–58. - PubMed
-
- Nishi H, Oishi N, Ogawa H, Natsue K, Doi K, Kawakami O, et al. Predicting cerebral infarction in patients with atrial fibrillation using machine learning: the Fushimi AF registry. J Cereb Blood flow Metabolism: Official J Int Soc Cereb Blood Flow Metabolism. 2022;42(5):746–56. 10.1177/0271678X211063802 - DOI - PMC - PubMed
MeSH terms
LinkOut - more resources
Full Text Sources
Research Materials