A Causal Framework for Evaluating ICU Discharge Strategies
Authors: Sagar Nagaraj Simha, Juliette Ortholand, Dave Dongelmans, Jessica D. Workum, Olivier W. M. Thijssens, Ameen Abu-Hanna, Giovanni Cinà
Organizations: Department of Medical Informatics, Amsterdam UMC, University of Amsterdam, NL · Department of Intensive Care Medicine, Amsterdam UMC, University of Amsterdam, NL · Department of Intensive Care, Elisabeth-TweeSteden Hospital, Tilburg, NL · Department of Intensive Care, Erasmus MC, Rotterdam, NL · Pacmed, Amsterdam, NL · Institute of Logic, Language and Computation, University of Amsterdam, NL
In this applied paper, we address the difficult open problem of when to discharge patients from the Intensive Care Unit. This can be conceived as an optimal stopping scenario with three added challenges: 1) the evaluation of a stopping strategy from observational data is itself a complex causal inference problem, 2) the composite objective is to minimize the length of intervention and maximize the outcome, but the two cannot be collapsed to a single dimension, and 3) the recording of variables stops when the intervention is discontinued. Our contributions are two-fold. First, we generalize the implementation of the g-formula Python package, providing a framework to evaluate stopping strategies for problems with the aforementioned structure, including positivity and coverage checks. Second, with a fully open-source pipeline, we apply this approach to MIMIC-IV, a public ICU dataset, demonstrating the potential for strategies that improve upon current care.
Figures & tables
Figure 1: The ICU discharge trade-off represented in the space of potential outcomes for our multi-objective stopping problem. Given a fixed dataset, each strategy returns an average 90-day mortality on the y-axis as well as an average length of ICU stay on the x-axis. The red box defines the set of strategies that constitute an improvement over current clinical practice, i.e. strategies leading to lower mortality and lower utilization of ICU beds. The blue dashed area defines the set of strategies we are actually able to evaluate given the available data.
Component
Clinical decision rule (12-hour grid)
Respiratory failure
Invasive controlled ventilation; OR SpO 2 <85% ; OR O 2 flow >85 ; OR (ventilation mode unknown AND [SpO 2 <85% OR O 2 flow >85 OR PaCO 2 >80 OR respiratory rate >45 /min]).
Hemodynamic instability
Mean arterial pressure <50 mmHg; OR lactate >6 mmol/L; OR heart rate >160 /min.
Neurologic failure
Glasgow Coma Scale <6 .
Severe metabolic derangement
Arterial bicarbonate <10 mmol/L.
Table 1: Dynamic discharge strategy (DS1) based on physiologic need for organ support: ICU care is continued if any of the red flags above are present, otherwise patients are discharged.
Figure 2: Positivity and coverage diagnostics over time for all evaluated dynamic strategies separated between patients kept and discharged. Top: coverage rate (fraction of history-compatible observations at epoch t for which the clinician’s action matches the target strategy’s prescription). Bottom: absolute match count (fraction of history-compatible observations at epoch t for which the clinician’s action matches the target strategy’s prescription and the decision at time t ).
All-cause mortality (%)
Difference to
Mortality split (%)
Length of ICU stay (days)
Intervention
Observed
G-formula estimate
natural course (%)
In ICU
After discharge
Median (IQR)
Mean
Natural course
20.08
19.1
(18.4, 21.5)
–
–
6.2
12.9
2.00
(1.00–4.00)
3.44
Discharge 3rd day
–
17.4
(16.7, 20.1)
-1.8
(-3.3, 0.5)
5.2
12.2
3
(3–3)
3
Strategy DS1
–
18.0
(15.8, 18.9)
-1.2
(-4.4, -0.6)
3.6
14.4
0.5
(0.5–1.0)
1.76
Strategy [ Knight, 2003 ]
–
86.0
(83.4, 89.4)
+66.9
(63.5, 69.5)
85.5
0.5
35.0
(15.5–62.5)
40.56
Table 2: Estimated 90-day all-cause mortality (in-ICU or post-discharge death) under dynamic and static discharge strategies using the parametric g-formula (NICE estimator). Mean and 95% confidence intervals are shown. Follow-up corresponds to 180 half-day intervals (90 days).
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
Variable name
Summary over last 12h
Distribution
Ventilation mode
last status
categorical
Hours since last recorded vent mode
last value
zero-inflated normal
Arterial pco2
mean
truncated normal
Arterial po2
mean
bounded normal
Oxygen flow
last value
bounded normal
O2 saturation
mean
bounded normal
Appendix
Table 3: Clinical variables with their 12-hour aggregation type and distribution
Test name
Variable
Test condition
Respiratory: airway
airway
airway patent
Respiratory: FiO 2
fio2
fio2 ≤ 0.6
Respiratory: blood oxygen
spo2
spo2 ≥ 95 (%)
Respiratory: bicarbonate
hco3
hco3 ≥ 19 (mmol/L)
Respiratory: rate
resp (rate)
10 ≤ resp ≤ 30 (bpm)
Cardiovascular: blood pressure
bp (systolic)
bp ≥ 100 (mm Hg)
Appendix
Table 4: Codified discharge criteria applied to electronic health record data. All criteria must be satisfied prior to ICU discharge.
Figure 3: Positivity diagnostics over time for all evaluated strategies. Top: coverage rate ρt (fraction of history-compatible observations at epoch t for which the clinician’s action matches the target strategy’s prescription). Middle: effective sample size ratio ESSt/Nt among matching observations (IS weight evenness; 1 = uniform, 1/Nt = fully concentrated). Bottom: absolute match count Nt . The red dashed line in the top and middle panels marks the 0.3 warning threshold.
Figure 4: PCA positivity scatter plots for the three static threshold strategy at 3 days, across decision epochs t∈{12,24,…,120} h (columns). Each panel projects the first two PCA components of the covariate space. Rows 1 & 3 : blue dots indicate patients for whom the strategy prescribes keep and the clinician kept; grey dots indicate patients for whom the strategy prescribes keep but the clinician discharged (positivity violation). Rows 2 & 4 : orange dots indicate agreement on discharge; grey dots indicate patients the strategy would discharge but clinicians retained (positivity violation). Panel titles report cohort size n and coverage rate ρt .
Figure 5: PCA positivity scatter plots for the dynamic strategies (Natural course, Knight strategy, and DS1 strategy; top to bottom), across decision epochs t∈{12,24,…,120} h (columns). Layout and colour coding are identical to Figure 4 .
Figure 6: Time-indexed standardized mean differences comparing observed and simulated covariate evolution under the natural course.
Figure 7: Covariate distribution comparison under the natural course and MIMIC-IV data for In-ICU death. The close overlap across physiologic variables demonstrates that the natural course simulation reproduces the empirical covariate distributions at In-ICU death.
Figure 8: Covariate distribution comparison under the natural course and MIMIC-IV data. The close overlap across physiologic variables demonstrates that the natural course simulation reproduces the empirical covariate distributions at discharge.
Hospital quality improvement (QI) programs routinely face multiple candidate interventions to optimize hospital flow, but existing methods struggle to estimate and rank the causal effects of such interventions. This work focuses on one of the most standard hospital metrics, the average length of stay (LOS), and its causal estimand, the average time saved. To characterize this causal effect, qualitative approaches rely on expert judgment to map patient trajectories, making them susceptible to cognitive biases; quantitative approaches rely on data-driven models, which fail when interventions are hypothetical with no historical data or have complex causal mechanisms that require clinical reasoning rather than data alone. We propose expert-guided g-computation, or egg-computation, which combines the complementary strengths of both approaches by connecting the Gantt charts commonly used to map patient trajectories with the causal DAG literature. We introduce a causal model over Gantt charts and establish identification using a variant of g-computation that seeks expert input only for components unidentifiable from data. To make egg-computation practical, we develop an LLM-assisted pipeline that reliably scales up expert reasoning. In simulations, egg-computation outperforms conventional causal inference methods when patients have diverse causal structures and intervention mechanisms. In a study of eleven candidate QI interventions at an urban safety-net hospital, the LLM pipeline generated graphs and time-saving estimates highly concordant with those of human experts. Beyond healthcare, egg-computation is a broadly applicable framework for estimating the average time saved for candidate interventions whose causal mechanisms can be represented using Gantt charts.
Patrick Vossler, Jialin Ouyang, F. Richard Guo +5
University of California, San Francisco · University of Michigan, Ann Arbor · University of Washington, Seattle
Intensive care units (ICU) generate long, dense and evolving streams of clinical information, where physicians must repeatedly reassess patient states under time pressure, underscoring a clear need for reliable AI decision support. Existing ICU benchmarks typically treat historical clinician actions as ground truth. However, these actions are made under incomplete information and limited temporal context of the underlying patient state, and may therefore be suboptimal, making it difficult to assess the true reasoning capabilities of AI systems. We introduce RealICU, a hindsight-annotated benchmark for evaluating large language models (LLMs) under realistic ICU conditions, where labels are created after senior physicians review the full patient trajectory. We formulate four physician-motivated tasks: assess Patient Status, Acute Problems, Recommended Actions, and Red Flag actions that risk unsafe outcomes. We partition each trajectory with 30-min windows and release two datasets: RealICU-Gold with 930-window annotations from 94 MIMIC-IV patients, and RealICU-Scale with 11,862 windows extended by Oracle, a physician-validated LLM hindsight labeler. Existing LLMs including memory-augmented ones performed poorly on RealICU, exposing two failure modes: a recall-safety tradeoff for clinical recommendations, and an anchoring bias to early interpretations of the patient. We further introduce ICU-Evo to study structured-memory agents that improves long-horizon reasoning but does not fully eliminate safety failures. Together, RealICU provides a clinically grounded testbed for measuring and improving AI sequential decision-support in high-stakes care. Project page: https://chengzhi-leo.github.io/RealICU-Bench/
Chengzhi Shen, Weixiang Shen, Tobias Susetzky +8
Cherise · Technical University of Munich (TUM) · TUM University Hospital +8
Many time-series forecasts depend not only on prior observations but also on actions specified during the forecast period. In intensive care units (ICUs), future vital signs and laboratory values are influenced by treatments such as vasopressors. However, models that predict the full future sequence all at once make little use of these treatments, whereas autoregressive models can accumulate errors. We introduce DRIFT, a hybrid framework in which a direct model produces the primary forecast and a recursive, action-conditioned model contributes constrained corrections. We evaluate DRIFT on 6,046 admissions from MIMIC-IV and 8,345 admissions from eICU-CRD. Averaged across the 8-, 24-, and 48-hour forecast endpoints, DRIFT reduces mean absolute error for mean arterial pressure (MAP) by 0.673% relative to an action-conditioned Temporal Fusion Transformer (TFT-action) on MIMIC-IV and achieves the lowest corresponding error among the compared models on eICU-CRD. Although the overall accuracy improvement is modest, a MIMIC-IV audit restricted to windows in which the supplied treatment sequence was altered showed that DRIFT achieved lower observed-target MAP error than TFT-action at 8 and 24 hours. Treatment-sequence alteration increased DRIFT's MAP error by 0.21-0.26 mmHg more than it increased TFT-action's error, with prediction changes occurring primarily after the supplied paths diverged. In a separate robustness experiment, the MAP advantage persisted under three shared checkpoint-selection rules emphasizing overall endpoint error, MAP error, or both equally.
Weixin Liu, Juming Xiong, Congning Ni +4
Vanderbilt University, Nashville, Tennessee, USA · Vanderbilt University Medical Center, Nashville, Tennessee, USA