FIRST-ICU: forecasting interventions and risk stratification in the ICU using graph neural network autoencoders

Read full text
Authors: Nosa Aikodon, Ivan Olier, Brian W. Johnston, Ingeborg D. Welters, Gregory Y. H. Lip, Sandra Ortega-Martorell
Cited by: 1
References: 87

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