Federated Learning for Brain Tumor Segmentation

Sammanfattning

This thesis investigates the potential of federated learning (FL) in medical image analysis, addressing the challenges posed by data privacy regulations in accessing medical datasets. The motivation stems from the increasing interest in artificial intelligence (AI)research, particularly in medical imaging for tumor detection using magnetic resonance imaging (MRI) and computer tomography (CT) scans. However, data accessibility remains a significant hurdle due to privacy regulations like the General Data Protection Regulation (GDPR). FL emerges as a solution by focusing on sharing network parameters instead of raw medical data, thus ensuring patient confidentiality. The aims of the study are to understand the requirements for FL models to perform comparably to centrally trained models, explore the impact of different aggregation functions, assess dataset heterogeneity, and evaluate the generalization of FL models. To achieve these goals, this thesis uses the BraTS 2021 dataset, which contains 1251 cases of brain tumor volumes from 23 distinct sites, with different distributions of the data across 3-8 nodes in a federation. The federation is set up to perform brain tumor segmentation, using different forms of aggregationfunctions (FedAvg. FedOpt, and FedProx) to finalize a global model. The final FL models demonstrate similar performance to that of centralized and local models, with minor variations. However, FL models’ performance varies depending on the dataset distribution and aggregation method used. Additionally, this study explores the impact of privacy-preserving techniques, such as differential privacy (DP), on FL model performance. While DP methods generally result in lower performance compared to non-DP methods, their effectiveness varies across different data distributions, and aggregation functions.