FedMed-GAN: Federated domain translation on unsupervised cross- modality brain image synthesis

被引:24
作者
Wang, Jinbao [1 ]
Xie, Guoyang [1 ,2 ]
Huang, Yawen [3 ]
Lyu, Jiayi [4 ]
Zheng, Feng [1 ,5 ]
Zheng, Yefeng [3 ]
Jin, Yaochu [2 ,6 ]
机构
[1] Southern Univ Sci & Technol, Dept Comp Sci & Engn, Shenzhen 518055, Peoples R China
[2] Univ Surrey, Dept Comp Sci, Guildford GU2 7XH, Surrey, England
[3] Tencent Jarvis Lab, Shenzhen 518040, Peoples R China
[4] Univ Chinese Acad Sci, Sch Engn Sci, Beijing, Peoples R China
[5] Peng Cheng Lab, Shenzhen 518000, Peoples R China
[6] Bielefeld Univ, Fac Technol, D-33619 Bielefeld, Germany
基金
国家重点研发计划; 中国国家自然科学基金;
关键词
Federated learning; Unsupervised learning; Cross-modality synthesis; Brain image; Deep learning;
D O I
10.1016/j.neucom.2023.126282
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Utilizing multi-modal neuroimaging data is proven to be effective in investigating human cognitive activ-ities 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 federated learning and facilitate the integration of dispersed data from different institutions. In this paper, we propose a new benchmark for federated domain translation on unsupervised brain image synthesis (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 properties. We treat the gradient penalties using the federated averaging algorithm and then leverage the differential privacy gra-dient descent to regularize the training dynamics. A comprehensive evaluation is provided for comparing FedMed-GAN and other centralized methods, demonstrating that the proposed algorithm outperforms the state-of-the-art. Our code is available at: https://github.com/M-3LAB/FedMed-GAN.(c) 2023 Elsevier B.V. All rights reserved.
引用
收藏
页数:10
相关论文
共 37 条
[1]   Deep Learning with Differential Privacy [J].
Abadi, Martin ;
Chu, Andy ;
Goodfellow, Ian ;
McMahan, H. Brendan ;
Mironov, Ilya ;
Talwar, Kunal ;
Zhang, Li .
CCS'16: PROCEEDINGS OF THE 2016 ACM SIGSAC CONFERENCE ON COMPUTER AND COMMUNICATIONS SECURITY, 2016, :308-318
[2]   A Combined Manifold Learning Analysis of Shape and Appearance to Characterize Neonatal Brain Development [J].
Aljabar, P. ;
Wolz, R. ;
Srinivasan, L. ;
Counsell, S. J. ;
Rutherford, M. A. ;
Edwards, A. D. ;
Hajnal, J. V. ;
Rueckert, D. .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2011, 30 (12) :2072-2086
[3]  
Augenstein S., 2020, ICLR 2020 8 INT C LE
[4]  
Bakas S., 2017, Lecture Notes in Computer Science
[5]  
Chen D., 2020, NEURIPS
[6]  
Chen Dingfan, 2020, Advances in Neural Information Processing Systems, V33
[7]   Image Synthesis in Multi-Contrast MRI With Conditional Generative Adversarial Networks [J].
Dar, Salman U. H. ;
Yurt, Mahmut ;
Karacan, Levent ;
Erdem, Aykut ;
Erdem, Erkut ;
Cukur, Tolga .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2019, 38 (10) :2375-2388
[8]   The Algorithmic Foundations of Differential Privacy [J].
Dwork, Cynthia ;
Roth, Aaron .
FOUNDATIONS AND TRENDS IN THEORETICAL COMPUTER SCIENCE, 2013, 9 (3-4) :211-406
[9]   Generative Adversarial Networks [J].
Goodfellow, Ian ;
Pouget-Abadie, Jean ;
Mirza, Mehdi ;
Xu, Bing ;
Warde-Farley, David ;
Ozair, Sherjil ;
Courville, Aaron ;
Bengio, Yoshua .
COMMUNICATIONS OF THE ACM, 2020, 63 (11) :139-144
[10]   Anatomic and Molecular MR Image Synthesis Using Confidence Guided CNNs [J].
Guo, Pengfei ;
Wang, Puyang ;
Yasarla, Rajeev ;
Zhou, Jinyuan ;
Patel, Vishal M. ;
Jiang, Shanshan .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2021, 40 (10) :2832-2844