Incremental Learning Meets Transfer Learning: Application to Multi-site Prostate MRI Segmentation.
Incremental Learning Meets Transfer Learning: Application to Multi-site Prostate MRI Segmentation.
复制标题
渐进学习与迁移学习的结合:在多部位前列腺 MRI 分割中的应用。
DOI:
10.1007/978-3-031-18523-6_1
复制
发表时间:
2022
期刊:
影响因子:
--
通讯作者:
Duncan,JamesS
中科院分区:
文献类型:
--
作者:
You,Chenyu;Xiang,Jinlin;Su,Kun;Zhang,Xiaoran;Dong,Siyuan;Onofrey,John;Staib,Lawrence;Duncan,JamesS
Many medical datasets have recently been created for medical image segmentation tasks, and it is natural to question whether we can use them to sequentially train a single model that (1) performs better on all these datasets, and (2) generalizes well and transfers better to the unknown target site domain. Prior works have achieved this goal by jointly training one model on multi-site datasets, which achieve competitive performance on average but such methods rely on the assumption about the availability of all training data, thus limiting its effectiveness in practical deployment. In this paper, we propose a novel multi-site segmentation framework calledincremental-transfer learning (ITL), which learns a model from multi-site datasets in an end-to-end sequential fashion. Specifically, “incremental” refers to training sequentially constructed datasets, and “transfer” is achieved by leveraging useful information from the linear combination of embedding features on each dataset. In addition, we introduce our ITL framework, where we train the network including a site-agnostic encoder with pretrained weights and at most two segmentation decoder heads. We also design a novel site-level incremental loss in order to generalize well on the target domain. Second, we show for the first time that leveraging our ITL training scheme is able to alleviate challenging catastrophic forgetting problems in incremental learning. We conduct experiments using five challenging benchmark datasets to validate the effectiveness of our incremental-transfer learning approach. Our approach makes minimal assumptions on computation resources and domain-specific expertise, and hence constitutes a strong starting point in multi-site medical image segmentation.
登录
查看更多内容
影响因子:
--
作者:
W. Miall;P. D. Oldham
通讯作者:
P. D. Oldham
影响因子:
168.9
作者:
K. Matthews;C. J. Rakaczky
通讯作者:
C. J. Rakaczky
影响因子:
2.8
作者:
W. Harlan;R. K. Osborne;Ashton Graybiel
通讯作者:
Ashton Graybiel
影响因子:
4.2
作者:
Glass David C.;Lake C. Raymond;Contrada Richard J.;Kehoe Kathleen;Erlanger Laura R.
通讯作者:
Erlanger Laura R.
DOI:
10.1037//0278-6133.4.5.413
发表时间:
1985
期刊:
Health psychology : official journal of the Division of Health Psychology, American Psychological Association
影响因子:
--
作者:
Carmelli,D;Chesney,MA;Ward,MM;Rosenman,RH
通讯作者:
Rosenman,RH