Federated prediction for scalable and privacy-preserved knowledge-based planning in radiotherapy

Although aggregating data from different institutions could alleviate this problem, data sharing is a practical challenge due to concerns about patient data privacy and other technical obstacles. Purpose: This work aims to address this dilemma by developing FedKBP+, a comprehensive federated learning (FL) platform for predictive tasks in real-world applications in radiotherapy treatment planning. Methods: We implemented a unified communication stack based on Google Remote Procedure Call (gRPC) to support communication between participants whether located on the same workstation or distributed across multiple workstations. In addition to supporting the centralized FL strategies commonly available in existing open-source frameworks, FedKBP+ also provides a fully decentralized FL model where participants directly exchange model weights to each other through Peer-to-Peer communication. We evaluated FedKBP+ on three predictive tasks using scale-attention network (SA-Net) as the predictive model. Results: Using 340 cases (training: 200; validation: 40; testing: 100) from the OpenKBP Challenge, a 3D dose prediction model trained with FedAvg algorithm outperformed the model trained on the local data, and achieved predictive accuracy comparable to that of a centrally trained model using pooled data in both independent and identically distributed (IID) and non-IID settings. We further evaluated the performance of FedKBP+ against NVFlare on the task of brain tumor segmentation using 227 cases from eight sites in the 2021 BraTS challenge dataset (training: 152; validation: 27; testing: 48). FedKBP+ surpassed the NVFlare framework in both accuracy and training efficiency.