FLARE: Federated Learning and Reinforcement Optimization for Cancer Survival Models
Document Type
Conference Paper
Publication Date
2025
Publication Title
2025 IEEE CONFERENCE ON ARTIFICIAL INTELLIGENCE, CAI
Abstract
Breast cancer remains a leading cause of mortality among women, with accurate survival prediction posing a major clinical challenge. While computational advances and abundant healthcare data enable data-driven prognostic models, data privacy remains a key barrier to multi-institutional collaboration. This study proposes FLARE, a novel framework integrating Federated Learning (FL) and Reinforcement Learning (RL) to predict breast cancer survival. Our approach jointly trains Random Survival Forests (RSF), Cox Proportional Hazards (CoxPH), and DeepHit across distributed datasets while preserving privacy. RL dynamically optimizes model parameters to enhance predictive accuracy and generalization. Experiments on the METABRIC dataset show that our FL-RL-enhanced RSF model achieves a C-index of 0.88, outperforming traditional survival models. This work underscores the potential of FL and RL in advancing clinical prognostics while safeguarding patient data.
DOI
10.1109/CAI64502.2025.00104
Recommended Citation
Farahani, Fataneh Tabarteh and Kumar, Sathish, "FLARE: Federated Learning and Reinforcement Optimization for Cancer Survival Models" (2025). Computer Science Faculty Publications. 13.
https://engagedscholarship.csuohio.edu/encs_facpub/13
Comments
Presented at the 2025 Conference on Artificial Intelligence-CAI-Annual, Santa Clara, CA, MAY 05-07, 2025.
This work was supported by the Department of Energy Grant No.DE-SC0024686.