JMIR Publications

ABSTRACT

Background:

Efficient allocation of healthcare resources is essential for sustainable hospital operation. Effective intensive care unit (ICU) management is essential for alleviating the financial strain on healthcare systems. Accurate prediction of length-of-stay in ICUs is vital for optimizing capacity planning and resource allocation, with the challenge of achieving early, real-time predictions.

Objective:

This study aims to develop a predictive model, namely WT-LSTM, for ICU length-of-stay using only real-time vital sign data. The model is designed for urgent care settings where demographic and historical patient data or lab results may be unavailable; the model leverages real-time inputs to deliver early and accurate ICU length-of-stay predictions.

Methods:

The proposed model integrates discrete wavelet transformation and Long Short-Term Memory (LSTM) neural networks to filter noise from patients’ vital sign series and improve length-of-stay prediction accuracy. Model performance was evaluated using the eICU database, focusing on ten common ICU admission diagnoses in the database.

Results:

The results demonstrate that WT-LSTM consistently outperforms baseline models, including linear regression, LSTM, and BiLSTM, in predicting ICU length-of-stay using vital sign data, achieving significant improvements in Mean Squared Error (MSE). Specifically, the wavelet transformation component of the model enhances the overall performance of WT-LSTM. Removing this component results in an average decrease of 3.3% in MSE; such a phenomenon is particularly pronounced in specific patient cohorts. The model's adaptability is highlighted through real-time predictions using only 3-hour, 6-hour, 12-hour, and 24-hour input data. Using only three hours of input data, the WT-LSTM model delivers competitive results across the ten most common ICU admission diagnoses, often outperforming APACHE IV, the leading ICU outcome prediction system currently implemented in clinical practice. WT-LSTM effectively captures patterns from vital signs recorded during the initial hours of a patient’s ICU stay, making it a promising tool for early prediction and resource optimization in the ICU.

Conclusions:

Our proposed WT-LSTM model, based on real-time vital sign data, offers a promising solution for ICU length-of-stay prediction. Its high accuracy and early prediction capabilities hold significant potential for enhancing clinical practice, optimizing resource allocation, and supporting critical clinical and administrative decisions in ICU management.