Early prediction of the risk of ICU mortality with Deep Federated Learning

12/01/2022
by   Korbinian Randl, et al.
0

Intensive Care Units usually carry patients with a serious risk of mortality. Recent research has shown the ability of Machine Learning to indicate the patients' mortality risk and point physicians toward individuals with a heightened need for care. Nevertheless, healthcare data is often subject to privacy regulations and can therefore not be easily shared in order to build Centralized Machine Learning models that use the combined data of multiple hospitals. Federated Learning is a Machine Learning framework designed for data privacy that can be used to circumvent this problem. In this study, we evaluate the ability of deep Federated Learning to predict the risk of Intensive Care Unit mortality at an early stage. We compare the predictive performance of Federated, Centralized, and Local Machine Learning in terms of AUPRC, F1-score, and AUROC. Our results show that Federated Learning performs equally well as the centralized approach and is substantially better than the local approach, thus providing a viable solution for early Intensive Care Unit mortality prediction. In addition, we show that the prediction performance is higher when the patient history window is closer to discharge or death. Finally, we show that using the F1-score as an early stopping metric can stabilize and increase the performance of our approach for the task at hand.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
05/30/2022

FLICU: A Federated Learning Workflow for Intensive Care Unit Mortality Prediction

Although Machine Learning (ML) can be seen as a promising tool to improv...
research
02/06/2021

FedMood:Federated Learning on Mobile Health Data for Mood Detection

Depression is one of the most common mental illness problems, and the sy...
research
02/18/2023

A Federated Approach for Hate Speech Detection

Hate speech detection has been the subject of high research attention, d...
research
11/12/2019

Harmonic Mean Point Processes: Proportional Rate Error Minimization for Obtundation Prediction

In healthcare, the highest risk individuals for morbidity and mortality ...
research
07/28/2023

Toward Transparent Sequence Models with Model-Based Tree Markov Model

In this study, we address the interpretability issue in complex, black-b...
research
10/18/2019

Privacy-preserving Federated Bayesian Learning of a Generative Model for Imbalanced Classification of Clinical Data

In clinical research, the lack of events of interest often necessitates ...

Please sign up or login with your details

Forgot password? Click here to reset