Abstract:Background: The prognosis, recurrence rates, and secondary prevention strategies varied significantly among different subtypes of acute ischemic stroke (AIS). Machine learning (ML) techniques can uncover intricate, non-linear relationships within medical data, enabling the identification of factors associated with etiological classification. However, there is currently a lack of research utilizing ML algorithms for predicting AIS etiology. Objective: We aimed to use interpretable ML algorithms to develop AIS etiology prediction models, identify critical factors in etiology classification, and enhance existing clinical categorization. Methods: This study involved patients with the Third China National Stroke Registry (CNSR-III). Nine models, which included Natural Gradient Boosting (NGBoost), Categorical Boosting (CatBoost), Extreme Gradient Boosting (XGBoost), Random Forest (RF), Light Gradient Boosting Machine (LGBM), Gradient Boosting Decision Tree (GBDT), Adaptive Boosting (AdaBoost), Support Vector Machine (SVM), and logistic regression (LR), were employed to predict large artery atherosclerosis (LAA), small vessel occlusion (SVO), and cardioembolism (CE) using an 80:20 randomly split training and test set. We designed an SFS-XGB with 10-fold cross-validation for feature selection. The primary evaluation metrics for the models included the area under the receiver operating characteristic curve (AUC) for discrimination and the Brier score (or calibration plots) for calibration. Results: A total of 5,213 patients were included, comprising 2,471 (47.4%) with LAA, 2,153 (41.3%) with SVO, and 589 (11.3%) with CE. In both LAA and SVO models, the AUC values of the ML models were significantly higher than that of the LR model (P < 0.001). The optimal model for predicting SVO (AUC [RF model] = 0.932) outperformed the optimal LAA model (AUC [NGB model] = 0.917) and the optimal CE model (AUC [LGBM model] = 0.846). Each model displayed relatively satisfactory calibration. Further analysis showed that the optimal CE model could identify potential CE patients in the undetermined etiology (SUE) group, accounting for 1,900 out of 4,156 (45.7%). Conclusions: The ML algorithm effectively classified patients with LAA, SVO, and CE, demonstrating superior classification performance compared to the LR model. The optimal ML model can identify potential CE patients among SUE patients. These newly identified predictive factors may complement the existing etiological classification system, enabling clinicians to promptly categorize stroke patients' etiology and initiate optimal strategies for secondary prevention.

A Multivariate Prediction Model and Its Application in Forecasting Acute Ischemic Stroke: Protocol for a Retrospective Clinical Study

Validating a Predictive Model of Acute Advanced Imaging Biomarkers in Ischemic Stroke

Prediction-Driven Decision Support for Patients With Mild Stroke: A Model Based on Machine Learning Algorithms

Prediction of Acute Myocardial Infarction in Asian Patients with Acute Ischemic Stroke: the CTRAN Score

Prediction of recurrent ischaemic stroke using radiomics data and machine learning methods in patients with acute ischaemic stroke: protocol for a multicentre, large sample, prospective observational cohort study in China

Development and validation of a machine learning-based prognostic risk stratification model for acute ischemic stroke

A machine learning model for visualization and dynamic clinical prediction of stroke recurrence in acute ischemic stroke patients: A real-world retrospective study

Prediction of large vessel occlusion for ischaemic stroke by using the machine learning model random forests

Short-Term Prognostic Risk Factors and Nomogram Predictive Model Construction for Acute Ischaemic Stroke after Endovascular Treatment

Predicting Poststroke Cognitive Impairment after Acute Ischemic Stroke Based on Admission Characteristics

Acute ischemic stroke prediction and predictive factors analysis using hematological indicators in elderly hypertensives post-transient ischemic attack

Machine learning is an effective method to predict the 3-month prognosis of patients with acute ischemic stroke

A one-year relapse prediction model for acute ischemic stroke (AIS) based on clinical big data

Predicting acute ischemic stroke using the revised Framingham stroke risk profile and multimodal magnetic resonance imaging

Risk factors and prediction models for recurrent acute ischemic stroke: a retrospective analysis

New strategy for clinical etiologic diagnosis of acute ischemic stroke and blood biomarker discovery based on machine learning

Predicting long-term outcomes for acute ischemic stroke using multi-model MRI radiomics and clinical variables

Machine Learning Models for Predicting Early Neurological Deterioration and Risk Classification of Acute Ischemic Stroke

Development of a Prediction Model for 1-Year Poor Prognosis in Patients with Acute Ischemic Stroke.

Personalized Risk Prediction of Symptomatic Intracerebral Hemorrhage after Stroke Thrombolysis Using a Machine-Learning Model

Predictive etiological classification of acute ischemic stroke through interpretable machine learning algorithms: a multicenter, prospective cohort study