Federated brain tumor segmentation: An extensive benchmark

被引:0
作者
Manthe, Matthis [1 ,2 ]
Duffner, Stefan [2 ]
Lartizien, Carole [1 ]
机构
[1] Univ Claude Bernard Lyon 1, INSA Lyon, CREATIS, CNRS,Inserm,U1294,UMR 5220, F-69621 Lyon, France
[2] Univ Lumiere Lyon 2, Univ Claude Bernard Lyon 1, INSA Lyon, Cent Lyon,CNRS,UMR5205,LIRIS, F-69621 Villeurbanne, France
关键词
Deep learning; Federated learning; Medical image segmentation; BraTS; Personalized federated learning; Clustered federated learning;
D O I
10.1016/j.media.2024.103270
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Recently, federated learning has raised increasing interest in the medical image analysis field due to its ability to aggregate multi-center data with privacy-preserving properties. A large amount of federated training schemes have been published, which we categorize into global (one final model), personalized (one model per institution) or hybrid (one model per cluster of institutions) methods. However, their applicability on the recently published Federated Brain Tumor Segmentation 2022 dataset has not been explored yet. We propose an extensive benchmark of federated learning algorithms from all three classes on this task. While standard FedAvg already performs very well, we show that some methods from each category can bring a slight performance improvement and potentially limit the final model(s) bias toward the predominant data distribution of the federation. Moreover, we provide a deeper understanding of the behavior of federated learning on this task through alternative ways of distributing the pooled dataset among institutions, namely an Independent and Identical Distributed (IID) setup, and a limited data setup. Our code is available at (https: //github.com/MatthisManthe/Benchmark_FeTS2022).
引用
收藏
页数:19
相关论文
共 89 条
  • [81] Gradient Leakage Attacks in Federated Learning: Research Frontiers, Taxonomy, and Future Directions
    Yang, Haomiao
    Ge, Mengyu
    Xue, Dongyun
    Xiang, Kunlan
    Li, Hongwei
    Lu, Rongxing
    [J]. IEEE NETWORK, 2024, 38 (02): : 247 - 254
  • [82] Fed2: Feature-Aligned Federated Learning
    Yu, Fuxun
    Zhang, Weishan
    Qin, Zhuwei
    Xu, Zirui
    Wang, Di
    Liu, Chenchen
    Tian, Zhi
    Chen, Xiang
    [J]. KDD '21: PROCEEDINGS OF THE 27TH ACM SIGKDD CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING, 2021, : 2066 - 2074
  • [83] Yuan H., 2022, What Do We Mean by Generalization in Federated Learning?
  • [84] Yuan Honglin, 2022, INT C LEARN REPR
  • [85] Zeineldin RA, 2022, arXiv, DOI DOI 10.48550/ARXIV.2212.09310
  • [86] Zhang MC, 2021, Arxiv, DOI [arXiv:2012.08565, 10.48550/ARXIV.2012.08565]
  • [87] Zhao Y, 2022, Arxiv, DOI [arXiv:1806.00582, DOI 10.1016/J.NEUCOM.2021.07.098]
  • [88] Zhu L., 2019, ADV NEURAL INFORM PR, V32
  • [89] Federated Medical Image Analysis with Virtual Sample Synthesis
    Zhu, Wei
    Luo, Jiebo
    [J]. MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2022, PT III, 2022, 13433 : 728 - 738