Federated Learning Using Variable Local Training for Brain Tumor Segmentation

被引:1
作者
Tuladhar, Anup [1 ,2 ]
Tyagi, Lakshay [3 ]
Souza, Raissa [1 ,2 ,4 ]
Forkert, Nils D. [1 ,2 ,5 ,6 ]
机构
[1] Univ Calgary, Dept Radiol, Cumming Sch Med, Calgary, AB, Canada
[2] Univ Calgary, Hotchkiss Brain Inst, Calgary, AB, Canada
[3] Indian Inst Technol Kanpur, Dept Chem Engn, Kanpur, Uttar Pradesh, India
[4] Univ Calgary, Biomed Engn Program, Calgary, AB, Canada
[5] Univ Calgary, Cumming Sch Med, Dept Clin Neurosci, Calgary, AB, Canada
[6] Univ Calgary, Alberta Childrens Hosp Res Inst, Calgary, AB, Canada
来源
BRAINLESION: GLIOMA, MULTIPLE SCLEROSIS, STROKE AND TRAUMATIC BRAIN INJURIES, BRAINLES 2021, PT II | 2022年 / 12963卷
基金
加拿大健康研究院;
关键词
Distributed learning; Federated learning; Convolutional neural network; Medical image analysis; Brain; Tumor segmentation; MICCAI; FeTS; BraTS;
D O I
10.1007/978-3-031-09002-8_35
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The potential for deep learning to improve medical image analysis is often stymied by the difficulty in acquiring and collecting sufficient data to train models. One major barrier to data acquisition is the private and sensitive nature of the data in question, as concerns about patient privacy, among others, make data sharing between institutions difficult. Distributed learning avoids the need to share data centrally by training models locally. One approach to distributed learning is federated learning, where models are trained in parallel at local institutions and aggregated together into a global model. The 2021 Federated Tumor Segmentation (FeTS) challenge focuses on federated learning for brain tumor segmentation usingmagnetic resonance imaging scans collected from a real-world federation of collaborating institutions. We developed a federated training algorithm that uses a combination of variable local epochs in each federated round, a decaying learning rate, and an ensemble weight aggregation function. When testing on unseen validation data our model trained with federated learning achieves very similar performance (average DSC score of 0.674) to a central model trained on pooled data (average DSC score 0.685). When our federated learning algorithm was evaluated on unseen training and testing data, it achieved similar performances on the FeTS challenge leaderboards 1 and 2 (average DSC scores of 0.623 and 0.608, respectively). This federated learning algorithm offers an approach to training deep learning learning models without the need to share private and sensitive patient data.
引用
收藏
页码:392 / 404
页数:13
相关论文
共 14 条
[1]  
Bakas S., 2017, Segmentation Labels for the Pre-operative Scans of the TCGA-LGG collection, V286
[2]   Data Descriptor: Advancing The Cancer Genome Atlas glioma MRI collections with expert segmentation labels and radiomic features [J].
Bakas, Spyridon ;
Akbari, Hamed ;
Sotiras, Aristeidis ;
Bilello, Michel ;
Rozycki, Martin ;
Kirby, Justin S. ;
Freymann, John B. ;
Farahani, Keyvan ;
Davatzikos, Christos .
SCIENTIFIC DATA, 2017, 4
[3]   Distributed deep learning networks among institutions for medical imaging [J].
Chang, Ken ;
Balachandar, Niranjan ;
Lam, Carson ;
Yi, Darvin ;
Brown, James ;
Beers, Andrew ;
Rosen, Bruce ;
Rubin, Daniel L. ;
Kalpathy-Cramer, Jayashree .
JOURNAL OF THE AMERICAN MEDICAL INFORMATICS ASSOCIATION, 2018, 25 (08) :945-954
[4]   Deep Learning-A Technology With the Potential to Transform Health Care [J].
Hinton, Geoffrey .
JAMA-JOURNAL OF THE AMERICAN MEDICAL ASSOCIATION, 2018, 320 (11) :1101-1102
[5]   Secure, privacy-preserving and federated machine learning in medical imaging [J].
Kaissis, Georgios A. ;
Makowski, Marcus R. ;
Ruckert, Daniel ;
Braren, Rickmer F. .
NATURE MACHINE INTELLIGENCE, 2020, 2 (06) :305-311
[6]   Supervised machine learning tools: a tutorial for clinicians [J].
Lo Vercio, Lucas ;
Amador, Kimberly ;
Bannister, Jordan J. ;
Crites, Sebastian ;
Gutierrez, Alejandro ;
MacDonald, M. Ethan ;
Moore, Jasmine ;
Mouches, Pauline ;
Rajashekar, Deepthi ;
Schimert, Serena ;
Subbanna, Nagesh ;
Tuladhar, Anup ;
Wang, Nanjia ;
Wilms, Matthias ;
Winder, Anthony ;
Forkert, Nils D. .
JOURNAL OF NEURAL ENGINEERING, 2020, 17 (06)
[7]   Machine learning for precision medicine [J].
MacEachern, Sarah J. ;
Forkert, Nils D. .
GENOME, 2021, 64 (04) :416-425
[8]  
McMahan H.B., 2016, ARXIV
[9]  
Pati S, 2021, Arxiv, DOI arXiv:2105.05874
[10]  
Reina G.A., 2021, arXiv, DOI DOI 10.1088/1361-6560/AC97D9