English

Adaptive Aggregation Weights for Federated Segmentation of Pancreas MRI

Image and Video Processing 2025-05-08 v3 Computer Vision and Pattern Recognition Distributed, Parallel, and Cluster Computing

Abstract

Federated learning (FL) enables collaborative model training across institutions without sharing sensitive data, making it an attractive solution for medical imaging tasks. However, traditional FL methods, such as Federated Averaging (FedAvg), face difficulties in generalizing across domains due to variations in imaging protocols and patient demographics across institutions. This challenge is particularly evident in pancreas MRI segmentation, where anatomical variability and imaging artifacts significantly impact performance. In this paper, we conduct a comprehensive evaluation of FL algorithms for pancreas MRI segmentation and introduce a novel approach that incorporates adaptive aggregation weights. By dynamically adjusting the contribution of each client during model aggregation, our method accounts for domain-specific differences and improves generalization across heterogeneous datasets. Experimental results demonstrate that our approach enhances segmentation accuracy and reduces the impact of domain shift compared to conventional FL methods while maintaining privacy-preserving capabilities. Significant performance improvements are observed across multiple hospitals (centers).

Keywords

Cite

@article{arxiv.2410.22530,
  title  = {Adaptive Aggregation Weights for Federated Segmentation of Pancreas MRI},
  author = {Hongyi Pan and Gorkem Durak and Zheyuan Zhang and Yavuz Taktak and Elif Keles and Halil Ertugrul Aktas and Alpay Medetalibeyoglu and Yury Velichko and Concetto Spampinato and Ivo Schoots and Marco J. Bruno and Rajesh N. Keswani and Pallavi Tiwari and Candice Bolan and Tamas Gonda and Michael G. Goggins and Michael B. Wallace and Ziyue Xu and Ulas Bagci},
  journal= {arXiv preprint arXiv:2410.22530},
  year   = {2025}
}

Comments

This paper has been accepted to ISBI 2025