Abstract:OBJECTIVE:To develop machine learning models for predicting intensive care unit (ICU) readmission using ensemble learning algorithms.METHODS:A publicly accessible American ICU database, medical information mart for intensive care (MIMIC)-Ⅲ as the data source was used, and the patients were selected by the inclusion and exclusion criteria. A set of variables that had the predictive ability of outcome including demographics, vital signs, laboratory tests, and comorbidities of patients were extracted from the dataset. We built the ICU readmission prediction models based on ensemble learning methods including random forest, adaptive boosting (AdaBoost), and gradient boosting decision tree (GBDT), and compared the prediction performance of the machine learning models with a conventional Logistic regression model. Five-fold cross validation was used to train and validate the prediction models. Average sensitivity, positive prediction value, negative prediction value, false positive rate, false negative rate, area under the receiver operating characteristic curve (AUROC) and Brier score were used as performance measures. After constructing the prediction models, top 10 predictive variables based on importance ranking were identified by the model with the best discrimination.RESULTS:Among these ICU readmission prediction models, GBDT (AUROC=0.858) had better performance than random forest (AUROC=0.827), and was slightly superior to AdaBoost (AUROC=0.851) in terms of AUROC. Compared with Logistic regression (AUROC=0.810), the discrimination of the three ensemble learning models was much better. The feature importance provided by GBDT showed that the top ranking variables included vital signs and laboratory tests. The patients with ICU readmission had higher mean arterial pressure, systolic blood pressure, diastolic blood pressure, and heart rate than the patients without ICU readmission. Meanwhile, the patients readmitted to ICU experienced lower urine output and higher serum creatinine. Overall, the patients having repeated admissions during their hospitalization showed worse heart function and renal function compared with the patients without ICU readmission.CONCLUSION:The ensemble learning based ICU readmission prediction models had better performance than Logistic regression model. Such ensemble learning models have the potential to aid ICU physicians in identifying those patients with high risk of ICU readmission and thus help improve overall clinical outcomes.

Prediction of extubation failure for intensive care unit patients using light gradient boosting machine

Prediction of Extubation Failure for Intensive Care Unit Patients Using Light Gradient Boosting Machine

A Risk Prediction Model for Efficient Intubation in the Emergency Department: A Five-Year Single-Center Retrospective Analysis

Development and Validation of a Machine-Learning Model for Prediction of Extubation Failure in Intensive Care Units

Development and Validation of a Machine-Learning Model for Prediction of Hypoxemia after Extubation in Intensive Care Units

Early prediction of noninvasive ventilation failure after extubation: development and validation of a machine-learning model

Machine learning-based in-hospital mortality risk prediction tool for intensive care unit patients with heart failure

Predicting Extubation Failure in Intensive Care: The Development of a Novel, End-to-End Actionable and Interpretable Prediction System

A New Risk Model Based on the Machine Learning Approach for Prediction of Mortality in the Respiratory Intensive Care Unit

Developing a machine-learning model for real-time prediction of successful extubation in mechanically ventilated patients using time-series ventilator-derived parameters

Development and Validation of a Deep Learning Model to Predict the Survival of Patients in ICU.

Using machine-learning models to predict extubation failure in neonates with bronchopulmonary dysplasia

Prediction model of in-hospital mortality in intensive care unit patients with heart failure: machine learning-based, retrospective analysis of the MIMIC-III database

Predicting Intubation for Intensive Care Units Patients: A Deep Learning Approach to Improve Patient Management

Development of a Machine-Learning Model for Prediction of Extubation Failure in Patients with Difficult Airways after General Anesthesia of Head, Neck, and Maxillofacial Surgeries

Dynamic prediction of late noninvasive ventilation failure in intensive care unit using a time adaptive machine model

[Prediction of Intensive Care Unit Readmission for Critically Ill Patients Based on Ensemble Learning].

Early Prediction of Extubation Failure in Severe Pneumonia: A Retrospective Cohort Study

Comprehensive breathing variability indices enhance the prediction of extubation failure in patients on mechanical ventilation

Optimizing Mortality Prediction for ICU Heart Failure Patients: Leveraging XGBoost and Advanced Machine Learning with the MIMIC-III Database

Predicting Length of Stay in Intensive Care Unit Using Ensemble Learning Methods