TRACE: Tracking Multimodal Clinical Trajectories for Inpatient Risk Prediction
Organizations: Yeji X
Abstract
Accurate inpatient risk prediction requires modeling heterogeneous clinical evidence that accumulates irregularly over time. This challenge spans in-hospital mortality, prolonged length of stay, and readmission, each with task-specific observation windows. Existing multimodal approaches combine structured health record data, chest x-rays, and clinical text, but primarily focus on modality fusion or missing-modality compensation, with limited attention to temporal trajectories over an admission. We propose TRACE (TRAjectory modeling of multimodal Clinical Events), a daily-scale framework organizing clinical events into day-level trajectories, separating day-invariant global context from day-specific local events drawn from EHR data, chest x-rays, radiology reports, and discharge summaries. The trajectory is encoded using temporal attention, allowing unavailable modalities to emit no tokens. A phenotype-guided contrastive objective further organizes admissions under heterogeneous disease distributions. Across cohorts from MIMIC-IV, eICU, and MIMIC-III comprising nearly 180,000 total stays, we evaluate TRACE on mortality, length of stay, and readmission prediction. TRACE achieves the highest AUROC across all nine dataset-task settings and the highest ACC in eight, outperforming baselines. Ablation studies validate our attention masks and contrastive learning, while trajectory-level attribution localizes risk-associated hospital days and modalities.
Figures & tables
| Dataset / Input Configuration | Mortality | Long LOS | Readmission | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| ACC | AUROC | AUPRC | mF1 | ACC | AUROC | AUPRC | mF1 | ACC | AUROC | AUPRC | mF1 | |
| MIMIC-IV: Unimodal | ||||||||||||
| C | 0.9099 | 0.8734 | 0.5536 | 0.7544 | 0.6489 | 0.6831 | 0.7602 | 0.5085 | 0.8179 | 0.7667 | 0.4736 | 0.6911 |
| E | 0.9542 | 0.8742 | 0.7698 | 0.7871 | 0.6421 | 0.6531 | 0.8105 | 0.5489 | 0.8696 | 0.8109 | 0.6414 | 0.7166 |
| R | 0.8681 | 0.8009 | 0.3920 | 0.6541 | 0.7164 | 0.6555 | 0.8209 | 0.4174 | 0.8225 | 0.7116 | 0.4513 | 0.6001 |
| D | 0.8964 | 0.6947 | 0.1758 | 0.4727 | – | – | – | – | 0.8048 | 0.6009 | 0.2473 | 0.4459 |
| Task | Metric | TGL | All-Causal | Full-Attention |
|---|---|---|---|---|
| Mortality | ACC | 0.9528 | 0.9533 | 0.9369 |
| AUROC | 0.9804 | 0.9778 | 0.9576 | |
| AUPRC | 0.8605 | 0.8500 | 0.7911 | |
| mF1 | 0.8886 | 0.8864 | 0.8275 | |
| Long LOS | ACC | 0.7409 | 0.7305 | 0.7193 |
| AUROC | 0.7675 | 0.7583 | 0.7462 |
| Dataset | Phenotype | Mortality | Long LOS | Readmission | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC | |||||||||
| MIMIC-IV | Circulatory | 764 | .9708 | +.0061 | .8272 | +.0131 | .7899 | +.0042 | .8331 | +.0048 | .8164 | .0015 | .6596 | .0044 |
| Trauma & Poisoning | 112 | .9374 | +.0120 | .7273 | +.0821 | .7490 | +.0127 | .7948 | +.0143 | .8920 | +.0088 | .6345 | .0070 | |
| Endocrine & Metabolic | 206 | .9805 | +.0073 | .8428 | +.0688 | .7517 | +.0189 | .8686 | +.0130 | .7788 | +.0040 | .6280 | .0113 | |
| Respiratory | 152 | .9582 | +.0117 | .8273 | +.0517 | .7492 | +.0254 | .8570 | +.0239 | .7868 | +.0047 | .6901 | .0170 | |
Clinical Outcome Prediction
Clinical outcome prediction uses patient records, imaging, signals, and biomarkers to estimate risks, disease trajectories, treatment response, and survival.
See all 265 papers