Machine Learning-based Prediction of Prolonged Intensive Care Unit Stay for Critical Patients with Spinal Cord Injury

Abstract

Study design: A retrospective cohort study.

Objective: The objective of the study was to develop machine-learning (ML) classifiers for predicting prolonged intensive care unit (ICU)-stay and prolonged hospital-stay for critical patients with spinal cord injury (SCI).

Summary of background data: Critical patients with SCI in ICU need more attention. SCI patients with prolonged stay in ICU usually occupy vast medical resources and hinder the rehabilitation deployment.

Methods: A total of 1599 critical patients with SCI were included in the study and labeled with prolonged stay or normal stay. All data were extracted from the eICU Collaborative Research Database and the Medical Information Mart for Intensive Care III-IV Database. The extracted data were randomly divided into training, validation and testing (6:2:2) subdatasets. A total of 91 initial ML classifiers were developed, and the top three initial classifiers with the best performance were further stacked into an ensemble classifier with logistic regressor. The area under the curve (AUC) was the main indicator to assess the prediction performance of all classifiers. The primary predicting outcome was prolonged ICU-stay, while the secondary predicting outcome was prolonged hospital-stay.

Results: In predicting prolonged ICU-stay, the AUC of the ensemble classifier was 0.864 ± 0.021 in the three-time five-fold cross-validation and 0.802 in the independent testing. In predicting prolonged hospital-stay, the AUC of the ensemble classifier was 0.815 ± 0.037 in the three-time five-fold cross-validation and 0.799 in the independent testing. Decision curve analysis showed the merits of the ensemble classifiers, as the curves of the top three initial classifiers varied a lot in either predicting prolonged ICU-stay or discriminating prolonged hospital-stay.

Conclusion: The ensemble classifiers successfully predict the prolonged ICU-stay and the prolonged hospital-stay, which showed a high potential of assisting physicians in managing SCI patients in ICU and make full use of medical resources.Level of Evidence: 3.