FIRST-ICU: forecasting interventions and risk stratification in the ICU using graph neural network autoencoders
Abstract
Critically ill patients frequently require multiple concurrent interventions with complex interdependencies, yet existing prediction models treat these as independent events. We developed FIRST-ICU (Forecasting Interventions and Risk Stratification in the ICU), a unified deep learning framework integrating a graph neural network encoder, LSTM, and a novel Intervention Interaction Attention Module (IIAM) for joint prediction of seven ICU interventions. FIRST-ICU was developed using MIMIC-IV (n = 23,926) and externally validated on AmsterdamUMCdb (n = 12,603) without retraining. The Temporal Decoder achieved AUC-ROC > 0.98 for all interventions with Brier scores 0.90 across intervention categories despite substantial differences in prescribing practices. A Generative Topographic Mapping layer identified six clinically distinct phenotypes, enabling interpretable risk stratification. FIRST-ICU advances multi-intervention prediction through joint modelling of treatment co-occurrence patterns, validated physiological learning, cross-continental generalisability, and interpretable risk stratification. It provides a framework for multi-intervention ICU decision support.
Fields of Study & Subjects
Sepsis Diagnosis and Treatment
Machine Learning in Healthcare
Artificial Intelligence in Healthcare and Education
Publication details
- Journal / source
- npj Digital Medicine
- Publisher
- Springer Science and Business Media LLC
- Volume
- 9
- Issue
- 1
- Publication date
- 11 June 2026
- Language
- EN