Federated Learning Enables Big Data for Rare Cancer Boundary Detection

Abstract

Although machine learning (ML) has shown promise in numerous domains, there are concerns about generalizability to out-of-sample data. This is currently addressed by centrally sharing ample, and importantly diverse, data from multiple sites. However, such centralization is challenging to scale (or even not feasible) due to various limitations. Federated ML (FL) provides an alternative to train accurate and generalizable ML models, by only sharing numerical model updates. Here we present findings from the largest FL study to-date, involving data from 71 healthcare institutions across 6 continents, to generate an automatic tumor boundary detector for the rare disease of glioblastoma, utilizing the largest dataset of such patients ever used in the literature (25,256 MRI scans from 6,314 patients). We demonstrate a 33% improvement over a publicly trained model to delineate the surgically targetable tumor, and 23% improvement over the tumor's entire extent. We anticipate our study to: 1) enable more studies in healthcare informed by large and diverse data, ensuring meaningful results for rare diseases and underrepresented populations, 2) facilitate further quantitative analyses for glioblastoma via performance optimization of our consensus model for eventual public release, and 3) demonstrate the effectiveness of FL at such scale and task complexity as a paradigm shift for multi-site collaborations, alleviating the need for data sharing.

federated learning, deep learning, convolutional neural network, segmentation, brain tumor, glioma, glioblastoma, FeTS, BraTS

References in corpus (2)

• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• Generalised Dice overlap as a deep learning loss function for highly unbalanced segmentations

Cited by in corpus (15)

• Multimodal Data Integration for Oncology in the Era of Deep Neural Networks: A Review
• Advanced Deep Learning and Large Language Models: Comprehensive Insights for Cancer Detection
• Federated brain tumor segmentation: an extensive benchmark
• AI in radiological imaging of soft-tissue and bone tumours: a systematic review evaluating against CLAIM and FUTURE-AI guidelines
• Federated learning, ethics, and the double black box problem in medical AI
• From Classical Machine Learning to Emerging Foundation Models: Review on Multimodal Data Integration for Cancer Research
• Fighting the scanner effect in brain MRI segmentation with a progressive level-of-detail network trained on multi-site data
• Evolving Horizons in Radiotherapy Auto-Contouring: Distilling Insights, Embracing Data-Centric Frameworks, and Moving Beyond Geometric Quantification
• Privacy-preserving patient clustering for personalized federated learning
• Decentralized Personalization for Federated Medical Image Segmentation via Gossip Contrastive Mutual Learning
• Federated Modality-specific Encoders and Partially Personalized Fusion Decoder for Multimodal Brain Tumor Segmentation
• Optimizing Federated Learning Configurations for MRI Prostate Segmentation and Cancer Detection: A Simulation Study
• FedECA: Federated External Control Arms for Causal Inference with Time-To-Event Data in Distributed Settings
• Kalman Filter Aided Federated Koopman Learning
• A Robust Framework for Secure Cardiovascular Risk Prediction: An Architectural Case Study of Differentially Private Federated Learning