Federated Learning for ICU Mortality Prediction: Balancing Accuracy and Privacy in a Multi-Hospital Setting
Benhammou, Y.; Kalyan, S.; Kumar, S.
Show abstract
Intensive care units (ICUs) manage critically ill patients whose clinical outcomes rely on timely and accurate decision-making. Predictive modeling using electronic health records (EHRs) has shown promise in forecasting adverse events such as in-hospital mortality. However, constructing robust and generalizable models typically requires large-scale, multi-institutional datasets. Privacy regulations such as HIPAA and GDPR make centralized data aggregation across hospitals challenging, posing a barrier to collaborative healthcare research. In this study, we explore a privacy-preserving approach for ICU mortality prediction using Federated Learning (FL), which enables distributed model training without sharing sensitive patient data. Leveraging a curated subset of the eICU Collaborative Research Database--a multi-center dataset of over 200,000 ICU admissions--we simulate a real-world federated learning scenario using data from three distinct hospitals comprising 5,000 ICU stays. We design a federated logistic regression framework and evaluate its performance against independently trained local models. Our findings indicate that the federated model consistently outperforms local baselines, achieving improved accuracy and AUC while maintaining strict data privacy. Our work delivers a reproducible and scalable framework for privacy-preserving healthcare AI and demonstrates the practical feasibility of deploying federated models for critical outcome prediction in ICU settings.
Matching journals
The top 9 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- From theoretical models to practical deployment: A perspective and case study of opportunities and challenges in AI-driven healthcare research for low-income settings 94%
- Modular Clinical Decision Support Networks (MoDN)—Updatable, Interpretable, and Portable Predictions for Evolving Clinical Environments 94%
- Uncovering the effects of model initialization on deep model generalization: A study with adult and pediatric chest X-ray images 94%
Similar papers in this journal
- FedNolowe: A Normalized Loss-Based Weighted Aggregation Strategy for Robust Federated Learning in Heterogeneous Environments 94%
- Compressive Big Data Analytics: An Ensemble Meta-Algorithm for High-dimensional Multisource Datasets 94%
- Identification of high-risk COVID-19 patients using machine learning 93%
Similar papers in this journal
- Mitigating Machine Learning Bias Between High Income and Low-Middle Income Countries for Enhanced Model Fairness and Generalizability 95%
- Developing Machine Learning Models for Predicting Intensive Care Unit Resource Use During the COVID-19 Pandemic 94%
- Emergency department admissions during COVID-19: explainable machine learning to characterise data drift and detect emergent health risks 94%
Similar papers in this journal
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.