NECA: Neighbor Event Curve Augmentation for Multi-Horizon ICU Deterioration Prediction
Organizations: Department of Statistics and Data Science, National University of Singapore, Singapore · School of Computing, National University of Singapore, Singapore · Institute for Infocomm Research (I2R), A*STAR, Singapore
Abstract
Early prediction of ICU organ dysfunction events supports timely clinical monitoring and intervention, but estimating short-term organ-specific risk from early ICU records remains challenging due to irregular measurements, heterogeneous clinical variables, and the need to predict risk across multiple future horizons. Existing supervised models usually treat each patient as an isolated sample and do not explicitly use the observed event timing of similar historical patients. We propose Neighbor Event-Curve Augmentation (NECA), a model-agnostic method that retrieves training-set patients with similar early-window profiles and converts their subsequent organ dysfunction event onset times into cumulative event-curve features over 6h, 12h, 24h, and 42h. These features are concatenated with the original patient representation and used by standard downstream predictors. On PhysioNet 2012 and a 4,000-stay MIMIC-IV subset, NECA improves average PR-AUC and ROC-AUC across four organ dysfunction event tasks and seven supervised learners. The largest gain appears for neurological dysfunction events, where task-level average PR-AUC increases from 0.461 to 0.557 on PhysioNet 2012. Ablations show that clinically meaningful retrieval and horizon-specific neighbor risk drive most of the improvement, while the full cumulative curve adds task-dependent temporal context. These results support similar-patient event timing as a structured temporal risk signal for ICU organ dysfunction event prediction. The source code will be publicly released upon acceptance of this paper.
Figures & tables
| Endpoint | Recorded event rule | Early-window variables |
|---|---|---|
| Renal | Maximum Cr mg/dL or BUN mg/dL | Age, gender, minimum MAP, maximum Cr and BUN, last urine output, maximum lactate |
| Neurological | Last GCS | Age, gender, minimum and last GCS, minimum MAP, maximum HR |
| Electrolyte / metabolic | Minimum pH , minimum HCO 3 , or maximum lactate | Age, gender, maximum lactate, minimum pH and HCO 3 , maximum Cr and BUN |
| Inflammatory | Maximum WBC or temperature C | Age, gender, maximum WBC, maximum temperature, minimum platelets, maximum lactate |
| PR-AUC | ROC-AUC | ||||
|---|---|---|---|---|---|
| Dataset | Endpoint | Raw | Curve-NECA | Raw | Curve-NECA |
| PN12 | Neurological | 0.461 | 0.557 | 0.689 | 0.766 |
| Renal | 0.353 | 0.387 | 0.711 | 0.740 | |
| Electrolyte/metabolic | 0.357 | 0.378 | 0.671 | 0.692 | |
| Inflammatory | 0.514 | 0.538 | 0.692 | 0.727 | |
| MIMIC | Neurological | 0.419 | 0.520 | 0.749 | 0.809 |
Clinical Outcome Prediction
Clinical outcome prediction uses patient records, imaging, signals, and biomarkers to estimate risks, disease trajectories, treatment response, and survival.
See all 268 papers