DeepAI
Log In Sign Up

Federated Survival Analysis with Discrete-Time Cox Models

06/16/2020
by   Mathieu Andreux, et al.
OWKIN, INC.
0

Building machine learning models from decentralized datasets located in different centers with federated learning (FL) is a promising approach to circumvent local data scarcity while preserving privacy. However, the prominent Cox proportional hazards (PH) model, used for survival analysis, does not fit the FL framework, as its loss function is non-separable with respect to the samples. The naïve method to bypass this non-separability consists in calculating the losses per center, and minimizing their sum as an approximation of the true loss. We show that the resulting model may suffer from important performance loss in some adverse settings. Instead, we leverage the discrete-time extension of the Cox PH model to formulate survival analysis as a classification problem with a separable loss function. Using this approach, we train survival models using standard FL techniques on synthetic data, as well as real-world datasets from The Cancer Genome Atlas (TCGA), showing similar performance to a Cox PH model trained on aggregated data. Compared to previous works, the proposed method is more communication-efficient, more generic, and more amenable to using privacy-preserving techniques.

READ FULL TEXT VIEW PDF

page 1

page 2

page 3

page 4

07/12/2022

FedPseudo: Pseudo value-based Deep Learning Models for Federated Survival Analysis

Survival analysis, time-to-event analysis, is an important problem in he...
10/14/2020

Improving Accuracy of Federated Learning in Non-IID Settings

Federated Learning (FL) is a decentralized machine learning protocol tha...
04/30/2022

Privacy Sensitive Speech Analysis Using Federated Learning to Assess Depression

Recent studies have used speech signals to assess depression. However, s...
07/24/2022

BPFISH: Blockchain and Privacy-preserving FL Inspired Smart Healthcare

This paper proposes Federated Learning (FL) based smart healthcare syste...
07/11/2022

A Federated Cox Model with Non-Proportional Hazards

Recent research has shown the potential for neural networks to improve u...
05/02/2018

A Simple Discrete-Time Survival Model for Neural Networks

There is currently great interest in applying neural networks to predict...
08/10/2022

KL-divergence Based Deep Learning for Discrete Time Model

Neural Network (Deep Learning) is a modern model in Artificial Intellige...

References