Generalizable Transfer Learning of Automated Tumor Segmentation from Cervical Cancers Toward a Universal Model for Uterine Malignancies in Diffusion-Weighted MRI

Yu-Chun Lin,Yenpo Lin,Yen-Ling Huang,Chih-Yi Ho,Hsin-Ju Chiang,Hsin-Ying Lu,Chun-Chieh Wang,Jiun-Jie Wang,Shu-Hang Ng,Chyong-Huey Lai,Gigin Lin
DOI: https://doi.org/10.1186/s13244-022-01356-8
IF: 5.0363
2023-01-01
Insights into Imaging
Abstract:PurposeTo investigate the generalizability of transfer learning (TL) of automated tumor segmentation from cervical cancers toward a universal model for cervical and uterine malignancies in diffusion-weighted magnetic resonance imaging (DWI).MethodsIn this retrospective multicenter study, we analyzed pelvic DWI data from 169 and 320 patients with cervical and uterine malignancies and divided them into the training (144 and 256) and testing (25 and 64) datasets, respectively. A pretrained model was established using DeepLab V3 + from the cervical cancer dataset, followed by TL experiments adjusting the training data sizes and fine-tuning layers. The model performance was evaluated using the dice similarity coefficient (DSC).ResultsIn predicting tumor segmentation for all cervical and uterine malignancies, TL models improved the DSCs from the pretrained cervical model (DSC 0.43) when adding 5, 13, 26, and 51 uterine cases for training (DSC improved from 0.57, 0.62, 0.68, 0.70, p < 0.001). Following the crossover at adding 128 cases (DSC 0.71), the model trained by combining data from adding all the 256 patients exhibited the highest DSCs for the combined cervical and uterine datasets (DSC 0.81) and cervical only dataset (DSC 0.91).ConclusionsTL may improve the generalizability of automated tumor segmentation of DWI from a specific cancer type toward multiple types of uterine malignancies especially in limited case numbers.
What problem does this paper attempt to address?