Synergizing gradient boosting and patient similarity: A hybrid interpretable framework for ICU length-of-stay prediction

Accurate forecasting of intensive care unit (ICU) length of stay (LOS) is important for resource planning, yet it often requires a compromise between predictive performance and interpretability. We present Gradient-Boosted Patient Similarity (GB-PS), a hybrid framework in which XGBoost-derived global feature importance is used to guide a covariance-aware k -Nearest Neighbors regressor for ICU LOS prediction. Using a retrospective cohort of 10,000 adult ICU admissions from MIMIC-IV, we compared GB-PS with a pure XGBoost baseline. GB-PS achieved a mean absolute error (MAE) of 1.65 days (RMSE 2.79, R 2 0.60), compared with 2.02 days (RMSE 2.78, R 2 0.65) for pure XGBoost. Thus, the hybrid model reduced typical patient-level absolute error while explaining slightly less overall variance. Because ICU bed planning is often driven by day-level prediction error, we treated MAE as the primary operational metric and interpreted the lower R 2 as a trade-off rather than a contradiction. Subgroup analyses showed stable performance in elderly and high-severity patients. The interpretability offered by GB-PS is example-based: predictions can be traced to clinically similar historical cases rather than to a fully explicit global rule set. Overall, GB-PS provides a pragmatic hybrid alternative that modestly improves MAE while preserving a clinically interpretable case-based reasoning pathway. • GB-PS framework synergizes gradient boosting with patient similarity. • XGBoost feature weights guide a covariance-aware k-NN distance metric. • Model achieves 1.65 days MAE on MIMIC-IV, outperforming pure XGBoost. • Case-based reasoning offers transparent explanations for clinical use. • The method is robust in high-severity patients and efficient on CPUs.

Authors

Institutions

Publication Details

Journal
Machine Learning with Applications
Published
2026-04-25
DOI
https://doi.org/10.1016/j.mlwa.2026.100907
Primary Topic
Sepsis Diagnosis and Treatment
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Synergizing gradient boosting and patient similarity: A hybrid interpretable framework for ICU length-of-stay prediction

Mohammad Tanhaei
Machine Learning with Applications
Sepsis Diagnosis and Treatment
article

Synergizing gradient boosting and patient similarity: A hybrid interpretable framework for ICU length-of-stay prediction

Mohammad Tanhaei
article en

Abstract

Accurate forecasting of intensive care unit (ICU) length of stay (LOS) is important for resource planning, yet it often requires a compromise between predictive performance and interpretability. We present Gradient-Boosted Patient Similarity (GB-PS), a hybrid framework in which XGBoost-derived global feature importance is used to guide a covariance-aware k -Nearest Neighbors regressor for ICU LOS prediction. Using a retrospective cohort of 10,000 adult ICU admissions from MIMIC-IV, we compared GB-PS with a pure XGBoost baseline. GB-PS achieved a mean absolute error (MAE) of 1.65 days (RMSE 2.79, R 2 0.60), compared with 2.02 days (RMSE 2.78, R 2 0.65) for pure XGBoost. Thus, the hybrid model reduced typical patient-level absolute error while explaining slightly less overall variance. Because ICU bed planning is often driven by day-level prediction error, we treated MAE as the primary operational metric and interpreted the lower R 2 as a trade-off rather than a contradiction. Subgroup analyses showed stable performance in elderly and high-severity patients. The interpretability offered by GB-PS is example-based: predictions can be traced to clinically similar historical cases rather than to a fully explicit global rule set. Overall, GB-PS provides a pragmatic hybrid alternative that modestly improves MAE while preserving a clinically interpretable case-based reasoning pathway. • GB-PS framework synergizes gradient boosting with patient similarity. • XGBoost feature weights guide a covariance-aware k-NN distance metric. • Model achieves 1.65 days MAE on MIMIC-IV, outperforming pure XGBoost. • Case-based reasoning offers transparent explanations for clinical use. • The method is robust in high-severity patients and efficient on CPUs.

Machine Learning with ApplicationsVol. 24
Ilam University (IR)
Decent work and economic growth
Openalex Percentile: Top 7%
Sepsis Diagnosis and Treatment
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.