Fair Federated Survival Analysis
Md Mahmudur Rahman, Sanjay Purushotham
Abstract
Federated Survival Analysis (FSA) is an emerging Federated Learning (FL) paradigm that enables training survival models on decentralized data while preserving privacy. However, existing FSA approaches largely overlook the potential risk of bias in predictions arising from demographic and censoring disparities across clients' datasets, which impacts the fairness and performance of federated survival models, especially for underrepresented groups. To address this gap, we introduce FairFSA, a novel FSA framework that adapts existing fair survival models to the federated setting. FairFSA jointly trains survival models using distributionally robust optimization, penalizing worst-case errors across subpopulations that exceed a specified probability threshold. Partially observed survival outcomes in clients are reconstructed with federated pseudo values (FPV) before model training to address censoring. Furthermore, we design a weight aggregation strategy by enhancing the FedAvg algorithm with a fairness-aware concordance index-based aggregation method to foster equitable performance distribution across clients. To the best of our knowledge, this is the first work to study and integrate fairness into Federated Survival Analysis. Comprehensive experiments on distributed non-IID datasets demonstrate FairFSA's superiority in fairness and accuracy over state-of-the-art FSA methods, establishing it as a robust FSA approach capable of handling censoring while providing equitable and accurate survival predictions for all subjects.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 58f9df5a-6871-4d0d-9ba7-9720b1a3f6eeBuilds on5
- Distributionally Robust Federated AveragingYuyang Deng, Mohammad Mahdi Kamani, Mehrdad MahdaviNeurIPS 2020 · 176 citations
- An Effective Meaningful Way to Evaluate Survival ModelsShiang Qi, Neeraj Kumar, Mahtab Farrokh, Weijie Sun et al.ICML 2023 · 28 citations
- DeepPseudo: Pseudo Value Based Deep Learning Models for Competing Risk AnalysisMd. Mahmudur Rahman, Koji Matsuo, Shinya Matsuzaki, Sanjay PurushothamAAAI 2021 · 22 citations
- Fair and Interpretable Models for Survival AnalysisMd. Mahmudur Rahman, Sanjay PurushothamKDD 2022 · 11 citations
- FedPseudo: Privacy-Preserving Pseudo Value-Based Deep Learning Models for Federated Survival AnalysisMd. Mahmudur Rahman, Sanjay PurushothamKDD 2023 · 8 citations
Related papers
- Fair Federated Learning Under Domain Skew with Local Consistency and Domain DiversityYuhang Chen, Wenke Huang, Mang YeCVPR 2024
- FairFed: Enabling Group Fairness in Federated LearningYahya H. Ezzeldin, Shen Yan, Chaoyang He, Emilio Ferrara et al.AAAI 2023 · 310 citations
- FedAA: A Reinforcement Learning Perspective on Adaptive Aggregation for Fair and Robust Federated LearningJialuo He, Wei Chen, Xiaojin ZhangAAAI 2025 · 12 citations
- Re-weighting Based Group Fairness Regularization via Classwise Robust OptimizationSangwon Jung, Taeeon Park, Sanghyuk Chun, Taesup MoonICLR 2023 · 5 citations
- FedVal: Different good or different bad in federated learningViktor Valadi, Xinchi Qiu, Pedro Porto Buarque de Gusmão, Nicholas D. Lane et al.USENIX Security 2023
