Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Federated Learning for Cross-Modality Medical Image Segmentation via Augmentation-Driven Generalization

About

Artificial intelligence has emerged as a transformative tool in medical image analysis, yet developing robust and generalizable segmentation models remains difficult due to fragmented, privacy-constrained imaging data siloed across institutions. While federated learning (FL) enables collaborative model training without centralizing data, cross-modality domain shifts pose a critical challenge, particularly when models trained on one modality fail to generalize to another. Many existing solutions require paired multimodal data per patient or rely on complex architectures, both of which are impractical in real clinical settings. In this work, we consider a realistic FL scenario where each client holds single-modality data (CT or MRI), and systematically investigate augmentation strategies for cross-modality generalization. Using abdominal organ segmentation and whole-heart segmentation as representative multi-class and binary segmentation benchmarks, we evaluate convolution-based spatial augmentation, frequency-domain manipulation, domain-specific normalization, and global intensity nonlinear (GIN) augmentation. Our results show that GIN consistently outperforms alternatives in both centralized and federated settings by simulating cross-modality appearance variations while preserving anatomical structure. For the pancreas, Dice score improved from 0.073 to 0.437, a 498% gain. Our federated approach achieves 93-98% of centralized training accuracy, demonstrating strong cross-modality generalization without compromising data privacy, pointing toward feasible federated AI deployment across diverse healthcare systems.

Sachin Dudda Nagaraju, Ashkan Moradi, Bendik Skarre Abrahamsen, Mattijs Elschot• 2026

Related benchmarks

TaskDatasetResultRank
Abdominal Organ SegmentationAMOS MRI 2020 (test)
Liver DSC0.9187
16
Abdominal Organ SegmentationAMOS CT 2020 (test)
Liver Dice Score93.4
16
Whole Heart SegmentationClient 4 MRI (test)
Dice (Myo)0.7088
7
Showing 3 of 3 rows

Other info

Follow for update