Federated survival analysis via data augmentation using multi-task variational autoencoder
摘要
Survival models find vast applications in biomedical studies. However, survival data used to train these models are usually distributed, censored and facing a growing concern for data privacy. In addition to these challenges, the inherently skewed nature of survival time distributions has received comparatively limited attention. To address these issues, this study introduces a novel federated learning framework that leverages a multi-task variational autoencoder (MVAE) for data augmentation, specifically designed to mitigate both censoring and data skewness problems in survival analysis. Experimental results from extensive simulated and real world survival datasets have demonstrated the effectiveness of the proposed methodology with possible deployments at the server or the clients.