FedPseudo: Pseudo value-based Deep Learning Models for Federated
Survival Analysis
- URL: http://arxiv.org/abs/2207.05247v1
- Date: Tue, 12 Jul 2022 01:10:36 GMT
- Title: FedPseudo: Pseudo value-based Deep Learning Models for Federated
Survival Analysis
- Authors: Md Mahmudur Rahman, Sanjay Purushotham
- Abstract summary: We propose a first-of-its-kind, pseudo value-based deep learning model for federated survival analysis called FedPseudo.
Our proposed FL framework achieves similar performance as the best centrally trained deep survival analysis model.
- Score: 9.659041001051415
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Survival analysis, time-to-event analysis, is an important problem in
healthcare since it has a wide-ranging impact on patients and palliative care.
Many survival analysis methods have assumed that the survival data is centrally
available either from one medical center or by data sharing from multi-centers.
However, the sensitivity of the patient attributes and the strict privacy laws
have increasingly forbidden sharing of healthcare data. To address this
challenge, the research community has looked at the solution of decentralized
training and sharing of model parameters using the Federated Learning (FL)
paradigm. In this paper, we study the utilization of FL for performing survival
analysis on distributed healthcare datasets. Recently, the popular Cox
proportional hazard (CPH) models have been adapted for FL settings; however,
due to its linearity and proportional hazards assumptions, CPH models result in
suboptimal performance, especially for non-linear, non-iid, and heavily
censored survival datasets. To overcome the challenges of existing federated
survival analysis methods, we leverage the predictive accuracy of the deep
learning models and the power of pseudo values to propose a first-of-its-kind,
pseudo value-based deep learning model for federated survival analysis (FSA)
called FedPseudo. Furthermore, we introduce a novel approach of deriving pseudo
values for survival probability in the FL settings that speeds up the
computation of pseudo values. Extensive experiments on synthetic and real-world
datasets show that our pseudo valued-based FL framework achieves similar
performance as the best centrally trained deep survival analysis model.
Moreover, our proposed FL approach obtains the best results for various
censoring settings.
Related papers
- Addressing Data Heterogeneity in Federated Learning of Cox Proportional Hazards Models [8.798959872821962]
This paper outlines an approach in the domain of federated survival analysis, specifically the Cox Proportional Hazards (CoxPH) model.
We present an FL approach that employs feature-based clustering to enhance model accuracy across synthetic datasets and real-world applications.
arXiv Detail & Related papers (2024-07-20T18:34:20Z) - Survival modeling using deep learning, machine learning and statistical methods: A comparative analysis for predicting mortality after hospital admission [9.719996519981333]
We conducted a comparative study of several survival analysis methods, including Cox proportional hazards (CoxPH), stepwise CoxPH, elastic net penalized Cox model, and GBM learning.
As a case study, we performed a retrospective analysis of patients admitted through the emergency department of a tertiary hospital from 2017 to 2019.
The results of the C-index indicate that deep learning achieved comparable performance, with DeepSurv producing the best discrimination.
arXiv Detail & Related papers (2024-03-04T10:46:02Z) - SAVAE: Leveraging the variational Bayes autoencoder for survival
analysis [10.0060346233449]
We introduce SAVAE (Survival Analysis Variational Autoencoder), a novel approach based on Variational Autoencoders.
Savoe contributes significantly to the field by introducing a tailored ELBO formulation for survival analysis.
It offers a general method that consistently performs well on various metrics, demonstrating robustness and stability through different experiments.
arXiv Detail & Related papers (2023-12-22T12:36:50Z) - Interpretable Survival Analysis for Heart Failure Risk Prediction [50.64739292687567]
We propose a novel survival analysis pipeline that is both interpretable and competitive with state-of-the-art survival models.
Our pipeline achieves state-of-the-art performance and provides interesting and novel insights about risk factors for heart failure.
arXiv Detail & Related papers (2023-10-24T02:56:05Z) - MedDiffusion: Boosting Health Risk Prediction via Diffusion-based Data
Augmentation [58.93221876843639]
This paper introduces a novel, end-to-end diffusion-based risk prediction model, named MedDiffusion.
It enhances risk prediction performance by creating synthetic patient data during training to enlarge sample space.
It discerns hidden relationships between patient visits using a step-wise attention mechanism, enabling the model to automatically retain the most vital information for generating high-quality data.
arXiv Detail & Related papers (2023-10-04T01:36:30Z) - Scaling Survival Analysis in Healthcare with Federated Survival Forests:
A Comparative Study on Heart Failure and Breast Cancer Genomics [7.967995669387532]
In real-world applications, survival data are often incomplete, censored, distributed, and confidential.
We propose an extension of the Federated Survival Forest algorithm, called FedSurF++.
arXiv Detail & Related papers (2023-08-04T15:25:56Z) - Practical Challenges in Differentially-Private Federated Survival
Analysis of Medical Data [57.19441629270029]
In this paper, we take advantage of the inherent properties of neural networks to federate the process of training of survival analysis models.
In the realistic setting of small medical datasets and only a few data centers, this noise makes it harder for the models to converge.
We propose DPFed-post which adds a post-processing stage to the private federated learning scheme.
arXiv Detail & Related papers (2022-02-08T10:03:24Z) - Bootstrapping Your Own Positive Sample: Contrastive Learning With
Electronic Health Record Data [62.29031007761901]
This paper proposes a novel contrastive regularized clinical classification model.
We introduce two unique positive sampling strategies specifically tailored for EHR data.
Our framework yields highly competitive experimental results in predicting the mortality risk on real-world COVID-19 EHR data.
arXiv Detail & Related papers (2021-04-07T06:02:04Z) - FLOP: Federated Learning on Medical Datasets using Partial Networks [84.54663831520853]
COVID-19 Disease due to the novel coronavirus has caused a shortage of medical resources.
Different data-driven deep learning models have been developed to mitigate the diagnosis of COVID-19.
The data itself is still scarce due to patient privacy concerns.
We propose a simple yet effective algorithm, named textbfFederated textbfL textbfon Medical datasets using textbfPartial Networks (FLOP)
arXiv Detail & Related papers (2021-02-10T01:56:58Z) - WRSE -- a non-parametric weighted-resolution ensemble for predicting
individual survival distributions in the ICU [0.251657752676152]
Dynamic assessment of mortality risk in the intensive care unit (ICU) can be used to stratify patients, inform about treatment effectiveness or serve as part of an early-warning system.
We show competitive results with state-of-the-art probabilistic models, while greatly reducing training time by factors of 2-9x.
arXiv Detail & Related papers (2020-11-02T10:13:59Z) - UNITE: Uncertainty-based Health Risk Prediction Leveraging Multi-sourced
Data [81.00385374948125]
We present UNcertaInTy-based hEalth risk prediction (UNITE) model.
UNITE provides accurate disease risk prediction and uncertainty estimation leveraging multi-sourced health data.
We evaluate UNITE on real-world disease risk prediction tasks: nonalcoholic fatty liver disease (NASH) and Alzheimer's disease (AD)
UNITE achieves up to 0.841 in F1 score for AD detection, up to 0.609 in PR-AUC for NASH detection, and outperforms various state-of-the-art baselines by up to $19%$ over the best baseline.
arXiv Detail & Related papers (2020-10-22T02:28:11Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.