Investigating transfer learning to improve the deep-learning-based segmentation of organs at risk among different medical centers for nasopharyngeal carcinoma
机构:[1]Chinese Acad Med Sci & Peking Union Med Coll, Canc Hosp, Natl Canc Ctr, Beijing, Peoples R China[2]Chinese Acad Med Sci, Hebei Canc Hosp, Natl Canc Ctr, Natl Clin Res Ctr Canc, Langfang, Peoples R China[3]Univ Chinese Acad Sci, Zhejiang Canc Hosp, Hangzhou, Peoples R China浙江省肿瘤医院[4]Sun Yat Sen Univ, Sun Yat Sen Univ Canc Ctr, Chinese Acad Sci,Ion Med Technol Co Ltd, State Key Lab Oncol South China,United Lab Frontie, Guangzhou, Peoples R China
Purpose: Convolutional neural networks (CNNs) offer a promising approach to automating organ segmentation in radiotherapy. However, variations of segmentation protocols made by different medical centers may induce a well-trained CNN model in one center and may not perform well in other centers. In this study, we proposed a transfer learning method to improve the performance of deep-learning based segmentation models among different medical centers using nasopharyngeal cancer (NPC) data.Methods: The NPC data included 300 cases (S_Train) from one institution (the source center) and 60 cases from another (the target center), divided into a training set of 50 cases (T_Train) and a test set of 10 target cases (T_Test). A ResNet CNN architecture was developed with 103 layers. We first trained Model_S and Model_T from scratch with the datasets S_Train and T_train, respectively. Transfer learning was then used to train Model_ST by fine-tuning the last 10 layers of Model_S with images from T_Train. We also investigated the effect of the numbers of re-trained layers on the performance. The performance of each model was evaluated using the dice similarity coefficient, and it was used as the evaluation metrics. We compared the dice similarity coefficient value using the three different models (Model_S, Model_T, and Model_ST).Results: When Model_S, Model_T, and Model_ST were applied to the T_Test dataset, the transfer learning (Model_ST) had the best performance. Compared with Model_S, the p-values of all organs at risk were less than 0.05. Compared with Model_T, the p-values of most organs at risk were less than 0.05, but there was no significant statistical difference in Model_ST for the brain stem (p = 0.071), mandible (p = 0.177), left temporal lobes (p = 0.084), and right temporal lobes (p = 0.068). Although there was no statistical difference for these organs, the mean accuracy of Model_ST was higher than that of Model_T. The proposed transfer learning can reduce the training time by up to 33%.Conclusion: Transfer learning can improve organ segmentation for NPC by adapting a previously trained CNN model to a new image domain, reducing the training time and saving physicians from labeling a large number of contours.
基金:
Beijing Natural Science Foundation [7222149]; CAMS Innovation Fund for Medical Sciences [2021-I2M-C, T-A-016]; National Natural Science Foundation of China [12005302, 11875320]; Beijing Nova Program [Z201100006820058]; Beijing Hope Run Special Fund of Cancer Foundation of China [LC2021A15]
第一作者机构:[1]Chinese Acad Med Sci & Peking Union Med Coll, Canc Hosp, Natl Canc Ctr, Beijing, Peoples R China[2]Chinese Acad Med Sci, Hebei Canc Hosp, Natl Canc Ctr, Natl Clin Res Ctr Canc, Langfang, Peoples R China
通讯作者:
通讯机构:[3]Univ Chinese Acad Sci, Zhejiang Canc Hosp, Hangzhou, Peoples R China[4]Sun Yat Sen Univ, Sun Yat Sen Univ Canc Ctr, Chinese Acad Sci,Ion Med Technol Co Ltd, State Key Lab Oncol South China,United Lab Frontie, Guangzhou, Peoples R China
推荐引用方式(GB/T 7714):
Chen Xinyuan,Zhu Ji,Yang Yiwei,et al.Investigating transfer learning to improve the deep-learning-based segmentation of organs at risk among different medical centers for nasopharyngeal carcinoma[J].FRONTIERS IN PHYSICS.2023,11:doi:10.3389/fphy.2023.1147900.
APA:
Chen, Xinyuan,Zhu, Ji,Yang, Yiwei,Zhang, Jie,Men, Kuo...&Dai, Jianrong.(2023).Investigating transfer learning to improve the deep-learning-based segmentation of organs at risk among different medical centers for nasopharyngeal carcinoma.FRONTIERS IN PHYSICS,11,
MLA:
Chen, Xinyuan,et al."Investigating transfer learning to improve the deep-learning-based segmentation of organs at risk among different medical centers for nasopharyngeal carcinoma".FRONTIERS IN PHYSICS 11.(2023)