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 条
  • [51] Paszke A, 2019, Arxiv, DOI [arXiv:1912.01703, 10.48550/arXiv.1912.01703, DOI 10.48550/ARXIV.1912.01703]
  • [52] Federated learning enables big data for rare cancer boundary detection
    Pati, Sarthak
    Baid, Ujjwal
    Edwards, Brandon
    Sheller, Micah
    Wang, Shih-Han
    Reina, G. Anthony
    Foley, Patrick
    Gruzdev, Alexey
    Karkada, Deepthi
    Davatzikos, Christos
    Sako, Chiharu
    Ghodasara, Satyam
    Bilello, Michel
    Mohan, Suyash
    Vollmuth, Philipp
    Brugnara, Gianluca
    Preetha, Chandrakanth J.
    Sahm, Felix
    Maier-Hein, Klaus
    Zenk, Maximilian
    Bendszus, Martin
    Wick, Wolfgang
    Calabrese, Evan
    Rudie, Jeffrey
    Villanueva-Meyer, Javier
    Cha, Soonmee
    Ingalhalikar, Madhura
    Jadhav, Manali
    Pandey, Umang
    Saini, Jitender
    Garrett, John
    Larson, Matthew
    Jeraj, Robert
    Currie, Stuart
    Frood, Russell
    Fatania, Kavi
    Huang, Raymond Y.
    Chang, Ken
    Balana, Carmen
    Capellades, Jaume
    Puig, Josep
    Trenkler, Johannes
    Pichler, Josef
    Necker, Georg
    Haunschmidt, Andreas
    Meckel, Stephan
    Shukla, Gaurav
    Liem, Spencer
    Alexander, Gregory S.
    Lombardo, Joseph
    [J]. NATURE COMMUNICATIONS, 2022, 13 (01)
  • [53] Pati S, 2021, Arxiv, DOI arXiv:2105.05874
  • [54] Pillutla K, 2022, Arxiv, DOI [arXiv:2204.03809, 10.48550/arXiv.2204.03809]
  • [55] Liang PP, 2020, Arxiv, DOI arXiv:2001.01523
  • [56] Qu LQ, 2022, PROC CVPR IEEE, P10051, DOI [10.1109/CVPR52688.2022.00982, 10.1109/cvpr52688.2022.00982]
  • [57] Handling data heterogeneity with generative replay in collaborative learning for medical imaging
    Qu, Liangqiong
    Balachandar, Niranjan
    Zhang, Miao
    Rubin, Daniel
    [J]. MEDICAL IMAGE ANALYSIS, 2022, 78
  • [58] Rawat A., 2022, Robust learning protocol for federated tumor segmentation challenge
  • [59] Reddi S.J., 2022, Adaptive Federated Optimization
  • [60] Survey on federated learning threats: Concepts, taxonomy on attacks and defences, experimental study and challenges
    Rodriguez-Barroso, Nuria
    Jimenez-Lopez, Daniel
    Victoria Luzon, M.
    Herrera, Francisco
    Martinez-Camara, Eugenio
    [J]. INFORMATION FUSION, 2023, 90 : 148 - 173