Distributed deep learning networks among institutions for medical imaging

被引:223
作者
Chang, Ken [1 ]
Balachandar, Niranjan [2 ]
Lam, Carson [2 ]
Yi, Darvin [2 ]
Brown, James [1 ]
Beers, Andrew [1 ]
Rosen, Bruce [1 ]
Rubin, Daniel L. [2 ]
Kalpathy-Cramer, Jayashree [1 ,3 ]
机构
[1] Massachusetts Gen Hosp, Dept Radiol, Athinoula A Martinos Ctr Biomed Imaging, Charlestown, MA 02129 USA
[2] Stanford Univ, Dept Radiol & Biomed Data Sci, Palo Alto, CA 94305 USA
[3] Massachusetts Gen Hosp, MGH & BWH Ctr Clin Data Sci, Boston, MA 02114 USA
基金
美国国家卫生研究院;
关键词
deep learning; neural networks; distributed learning; medical imaging; DIABETIC-RETINOPATHY; NEURAL-NETWORK;
D O I
10.1093/jamia/ocy017
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Objective: Deep learning has become a promising approach for automated support for clinical diagnosis. When medical data samples are limited, collaboration among multiple institutions is necessary to achieve high algorithm performance. However, sharing patient data often has limitations due to technical, legal, or ethical concerns. In this study, we propose methods of distributing deep learning models as an attractive alternative to sharing patient data. Methods: We simulate the distribution of deep learning models across 4 institutions using various training heuristics and compare the results with a deep learning model trained on centrally hosted patient data. The training heuristics investigated include ensembling single institution models, single weight transfer, and cyclical weight transfer. We evaluated these approaches for image classification in 3 independent image collections (retinal fundus photos, mammography, and ImageNet). Results: We find that cyclical weight transfer resulted in a performance that was comparable to that of centrally hosted patient data. We also found that there is an improvement in the performance of cyclical weight transfer heuristic with a high frequency of weight transfer. Conclusions: We show that distributing deep learning models is an effective alternative to sharing patient data. This finding has implications for any collaborative deep learning study.
引用
收藏
页码:945 / 954
页数:10
相关论文
共 24 条
[1]  
[Anonymous], ARXIVABS160502688
[2]  
[Anonymous], 2015, Diabetic retinopathy detection
[3]   Residual Convolutional Neural Network for the Determination of IDH Status in Low- and High-Grade Gliomas from MR Imaging [J].
Chang, Ken ;
Bai, Harrison X. ;
Zhou, Hao ;
Su, Chang ;
Bi, Wenya Linda ;
Agbodza, Ena ;
Kavouridis, Vasileios K. ;
Senders, Joeky T. ;
Boaro, Alessandro ;
Beers, Andrew ;
Zhang, Biqi ;
Capellini, Alexandra ;
Liao, Weihua ;
Shen, Qin ;
Li, Xuejun ;
Xiao, Bo ;
Cryan, Jane ;
Ramkissoon, Shakti ;
Ramkissoon, Lori ;
Ligon, Keith ;
Wen, Patrick Y. ;
Bindra, Ranjit S. ;
Woo, John ;
Arnaout, Omar ;
Gerstner, Elizabeth R. ;
Zhang, Paul J. ;
Rosen, Bruce R. ;
Yang, Li ;
Huang, Raymond Y. ;
Kalpathy-Cramer, Jayashree .
CLINICAL CANCER RESEARCH, 2018, 24 (05) :1073-1081
[4]  
Chen H., 2015, ARXIV150701239
[5]  
Collobert R., 2008, P 25 ICML, P160, DOI [DOI 10.1145/1390156.1390177, 10.1145/1390156.1390177]
[6]  
Dean J., 2012, P ADV NEUR INF PROC, V1, P1223
[7]   Multi-center machine learning in imaging psychiatry: A meta-model approach [J].
Dluhos, Petr ;
Schwarz, Daniel ;
Cahn, Wiepke ;
van Haren, Neeltje ;
Kahn, Rene ;
Spaniel, Filip ;
Horacek, Jiri ;
Kasparek, Tomas ;
Schnack, Hugo .
NEUROIMAGE, 2017, 155 :10-24
[8]   Dermatologist-level classification of skin cancer with deep neural networks [J].
Esteva, Andre ;
Kuprel, Brett ;
Novoa, Roberto A. ;
Ko, Justin ;
Swetter, Susan M. ;
Blau, Helen M. ;
Thrun, Sebastian .
NATURE, 2017, 542 (7639) :115-+
[9]  
Glorot X, 2010, P 13 INT C ART INT S, P249, DOI DOI 10.1109/LGRS.2016.2565705
[10]   Development and Validation of a Deep Learning Algorithm for Detection of Diabetic Retinopathy in Retinal Fundus Photographs [J].
Gulshan, Varun ;
Peng, Lily ;
Coram, Marc ;
Stumpe, Martin C. ;
Wu, Derek ;
Narayanaswamy, Arunachalam ;
Venugopalan, Subhashini ;
Widner, Kasumi ;
Madams, Tom ;
Cuadros, Jorge ;
Kim, Ramasamy ;
Raman, Rajiv ;
Nelson, Philip C. ;
Mega, Jessica L. ;
Webster, R. .
JAMA-JOURNAL OF THE AMERICAN MEDICAL ASSOCIATION, 2016, 316 (22) :2402-2410