FedMed-GAN: Federated Domain Translation on Unsupervised Cross-Modality Brain Image Synthesis

TL;DR

FedMed-GAN introduces a federated learning framework for unsupervised cross-modality brain image synthesis, effectively handling privacy constraints and data dispersion across institutions, achieving state-of-the-art results.

Contribution

The paper presents FedMed-GAN, a novel federated GAN model for unsupervised brain image translation that mitigates mode collapse and adapts to unpaired and paired data without central data collection.

Findings

Outperforms existing centralized methods in brain image synthesis tasks.

Effectively handles unpaired and paired data with variation adaptation.

Maintains high generator performance while mitigating mode collapse.

Abstract

Utilizing multi-modal neuroimaging data has been proved to be effective to investigate human cognitive activities and certain pathologies. However, it is not practical to obtain the full set of paired neuroimaging data centrally since the collection faces several constraints, e.g., high examination cost, long acquisition time, and image corruption. In addition, these data are dispersed into different medical institutions and thus cannot be aggregated for centralized training considering the privacy issues. There is a clear need to launch a federated learning and facilitate the integration of the dispersed data from different institutions. In this paper, we propose a new benchmark for federated domain translation on unsupervised brain image synthesis (termed as FedMed-GAN) to bridge the gap between federated learning and medical GAN. FedMed-GAN mitigates the mode collapse without sacrificing the performance of generators, and is widely applied to different proportions of unpaired and paired data with variation adaptation property. We treat the gradient penalties by federally averaging algorithm and then leveraging differential privacy gradient descent to regularize the training dynamics. A comprehensive evaluation is provided for comparing FedMed-GAN and other centralized methods, which shows the new state-of-the-art performance by our FedMed-GAN. Our code has been released on the website: https://github.com/M-3LAB/FedMed-GAN