Federated Survival Forests

Abstract

Survival analysis is a subfield of statistics concerned with modeling the occurrence time of a particular event of interest for a population. Survival analysis found widespread applications in healthcare, engineering, and social sciences. However, real-world applications involve survival datasets that are distributed, incomplete, censored, and confidential. In this context, federated learning can tremendously improve the performance of survival analysis applications. Federated learning provides a set of privacy-preserving techniques to jointly train machine learning models on multiple datasets without compromising user privacy, leading to a better generalization performance. However, despite the widespread development of federated learning in recent AI research, few studies focus on federated survival analysis. In this work, we present a novel federated algorithm for survival analysis based on one of the most successful survival models, the random survival forest. We call the proposed method Federated Survival Forest (FedSurF). With a single communication round, FedSurF obtains a discriminative power comparable to deep-learning-based federated models trained over hundreds of federated iterations. Moreover, FedSurF retains all the advantages of random forests, namely low computational cost and natural handling of missing values and incomplete datasets. These advantages are especially desirable in real-world federated environments with multiple small datasets stored on devices with low computational capabilities. Numerical experiments compare FedSurF with state-of-the-art survival models in federated networks, showing how FedSurF outperforms deep-learning-based federated algorithms in realistic environments with non-identically distributed data.

References in corpus (15)

• Federated Learning: Challenges, Methods, and Future Directions
• The Future of Digital Health with Federated Learning
• Random survival forests
• DeepSurv: Personalized Treatment Recommender System Using A Cox Proportional Hazards Deep Neural Network
• SCAFFOLD: Stochastic Controlled Averaging for Federated Learning
• Measuring the Effects of Non-Identical Data Distribution for Federated Visual Classification
• A Scalable Discrete-Time Survival Model for Neural Networks
• Time-to-Event Prediction with Neural Networks and Cox Regression
• A Field Guide to Federated Optimization
• Deep Neural Networks for Survival Analysis Based on a Multi-Task Framework
• FLamby: Datasets and Benchmarks for Cross-Silo Federated Learning in Realistic Healthcare Settings
• A General Machine Learning Framework for Survival Analysis
• Practical Challenges in Differentially-Private Federated Survival Analysis of Medical Data
• FedPseudo: Pseudo value-based Deep Learning Models for Federated Survival Analysis
• SecureFedYJ: a safe feature Gaussianization protocol for Federated Learning

Cited by in corpus (2)

• Scaling Survival Analysis in Healthcare with Federated Survival Forests: A Comparative Study on Heart Failure and Breast Cancer Genomics
• FedECA: Federated External Control Arms for Causal Inference with Time-To-Event Data in Distributed Settings