A deep learning approach for inpatient length of stay and mortality prediction

J Biomed Inform. 2023 Nov:147:104526. doi: 10.1016/j.jbi.2023.104526. Epub 2023 Oct 17.

Abstract

Purpose: Accurate prediction of the Length of Stay (LoS) and mortality in the Intensive Care Unit (ICU) is crucial for effective hospital management, and it can assist clinicians for real-time demand capacity (RTDC) administration, thereby improving healthcare quality and service levels.

Methods: This paper proposes a novel one-dimensional (1D) multi-scale convolutional neural network architecture, namely 1D-MSNet, to predict inpatients' LoS and mortality in ICU. First, a 1D multi-scale convolution framework is proposed to enlarge the convolutional receptive fields and enhance the richness of the convolutional features. Following the convolutional layers, an atrous causal spatial pyramid pooling (SPP) module is incorporated into the networks to extract high-level features. The optimized Focal Loss (FL) function is combined with the synthetic minority over-sampling technique (SMOTE) to mitigate the imbalanced-class issue.

Results: On the MIMIC-IV v1.0 benchmark dataset, the proposed approach achieves the optimum R-Square and RMSE values of 0.57 and 3.61 for the LoS prediction, and the highest test accuracy of 97.73% for the mortality prediction.

Conclusion: The proposed approach presents a superior performance in comparison with other state-of-the-art, and it can effectively perform the LoS and mortality prediction tasks.

Keywords: Atrous causal SPP; Length of stay prediction; Mortality prediction; Multi-scale convolution; SMOTE.

Publication types

  • Research Support, U.S. Gov't, Non-P.H.S.

MeSH terms

  • Deep Learning*
  • Humans
  • Inpatients
  • Intensive Care Units
  • Length of Stay
  • Neural Networks, Computer