Skip to main navigation Skip to search Skip to main content

Machine Learning-Driven Prediction of Intensive Care Units Mortality and Length of Stay: A 11-Year Retrospective Study in Hong Kong Public Hospitals

  • Ying Zhao (Co-first Author)
  • , Xincheng Shu (Co-first Author)
  • , Chi-Sing Leung
  • , Eric W. M. Wong*
  • , Qi Xuan
  • , Kar-Lung Lee
  • , Anne Leung
  • , Lowell Ling
  • , Hoi-Ping Shum
  • , Wing-Lun Wan
  • , Pauline Yeung Ng
  • , Tsz-Kin Yim
  • , Wai-Ming Tang
  • , Kenny King-Chung Chan
  • , Gavin Joynt
  • *Corresponding author for this work

Research output: Journal Publications and ReviewsRGC 21 - Publication in refereed journalpeer-review

17 Downloads (CityUHK Scholars)

Abstract

This study aims to develop a machine learning (ML)-based pipeline to predict intensive care unit (ICU) mortality and length of stay (LOS). A dataset including 140,904 ICU admissions was collected from 15 public hospitals in Hong Kong over an 11-year period. The proposed pipeline deployed a suite of ML models to predict mortality and LOS. The performance of ML models was compared with the Acute Physiology and Chronic Health Evaluation (APACHE) systems on the collected dataset using five-fold cross-validation. Among all involved models, the Gradient Boosting with Categorical Features (CatBoost) achieved the highest area under the receiver operating characteristic curve (AUROC) of 0.9070 as well as the lowest Brier score of 0.0827 for mortality prediction and the lowest Mean Absolute Error (MAE) of 2.6364 for LOS prediction. The SHapley Additive exPlanations (SHAP) analysis conducted on CatBoost revealed that age, Glasgow Coma Scale (GCS) and urine output were the top-three important features for mortality prediction, whereas the top-three important features for LOS prediction were creatinine level, and the indicators for whether the lowest and highest respiratory rates were ventilator-measured. We further performed temporal validation and an in-depth analysis of CatBoost’s predictive performance across subsets grouped by age and hospital. Our results demonstrate that the proposed pipeline mitigates the overestimation of mortality predictions from APACHE systems in Hong Kong. Besides, the proposed predictive ML-based pipeline offers a transferable framework for researchers to develop models tailored to their local medical environments. © The Author(s) 2026.
Original languageEnglish
Article number31
Number of pages15
JournalJournal of Medical Systems
Volume50
Issue number1
Online published10 Mar 2026
DOIs
Publication statusOnline published - 10 Mar 2026

Funding

Open access publishing enabled by City University of Hong Kong Library's agreement with Springer Nature. This study was supported in part by the Health and Medical Research Fund of Hong Kong (16171921) and the Research Grants Council (RGC) of Hong Kong under the General Research Fund (11104620, 11102421, and 11101422), in part by the Key R&D Program of Zhejiang under Grant (2022C01018, 2024C01025), in part by the National Natural Science Foundation of China under Grants U21B2001.

Research Keywords

  • ICU
  • Machine learning
  • Mortality prediction
  • Length of stay prediction
  • Model interpretability

Publisher's Copyright Statement

  • This full text is made available under CC-BY 4.0. https://creativecommons.org/licenses/by/4.0/

RGC Funding Information

  • RGC-funded

Fingerprint

Dive into the research topics of 'Machine Learning-Driven Prediction of Intensive Care Units Mortality and Length of Stay: A 11-Year Retrospective Study in Hong Kong Public Hospitals'. Together they form a unique fingerprint.

Cite this