Risk factor identification and prediction models for prolonged length of stay in hospital after acute ischemic stroke using artificial neural networks

A clinical prediction model based on interpretable machine learning algorithms for prolonged hospital stay in acute ischemic stroke patients: a real-world study

Objective

Acute ischemic stroke (AIS) brings an increasingly heavier economic burden nowadays. Prolonged length of stay (LOS) is a vital factor in healthcare expenditures. The aim of this study was to predict prolonged LOS in AIS patients based on an interpretable machine learning algorithm.

Methods

We enrolled AIS patients in our hospital from August 2017 to July 2019, and divided them into the "prolonged LOS" group and the "no prolonged LOS" group. Prolonged LOS was defined as hospitalization for more than 7 days. The least absolute shrinkage and selection operator (LASSO) regression was applied to reduce the dimensionality of the data. We compared the predictive capacity of extended LOS in eight different machine learning algorithms. SHapley Additive exPlanations (SHAP) values were used to interpret the outcome, and the most optimal model was assessed by discrimination, calibration, and clinical utility.

Results

Prolonged LOS developed in 149 (22.0%) of the 677 eligible patients. In eight machine learning algorithms, prolonged LOS was best predicted by the Gaussian naive Bayes (GNB) model, which had a striking area under the curve (AUC) of 0.878 ± 0.007 in the training set and 0.857 ± 0.039 in the validation set. The variables sorted by the gap values showed that the strongest predictors were pneumonia, dysphagia, thrombectomy, and stroke severity. High net benefits were observed at 0%-76% threshold probabilities, while good agreement was found between the observed and predicted probabilities.

Conclusions

The model using the GNB algorithm proved excellent for predicting prolonged LOS in AIS patients. This simple model of prolonged hospitalization could help adjust policies and better utilize resources.

XGBoost-Based Simple Three-Item Model Accurately Predicts Outcomes of Acute Ischemic Stroke

An all-inclusive and accurate prediction of outcomes for patients with acute ischemic stroke (AIS) is crucial for clinical decision-making. This study developed extreme gradient boosting (XGBoost)-based models using three simple factors-age, fasting glucose, and National Institutes of Health Stroke Scale (NIHSS) scores-to predict the three-month functional outcomes after AIS. We retrieved the medical records of 1848 patients diagnosed with AIS and managed at a single medical center between 2016 and 2020. We developed and validated the predictions and ranked the importance of each variable. The XGBoost model achieved notable performance, with an area under the curve of 0.8595. As predicted by the model, the patients with initial NIHSS score > 5, aged over 64 years, and fasting blood glucose > 86 mg/dL were associated with unfavorable prognoses. For patients receiving endovascular therapy, fasting glucose was the most important predictor. The NIHSS score at admission was the most significant predictor for those who received other treatments. Our proposed XGBoost model showed a reliable predictive power of AIS outcomes using readily available and simple predictors and also demonstrated the validity of the model for application in patients receiving different AIS treatments, providing clinical evidence for future optimization of AIS treatment strategies.