Transfer learning for KiTS21 Challenge

Xi Yang, Zhang Jianpeng, Xia Yong

Keywords: Transfer learning, Limited annotation, Kidney tumor segmentation

Abstract: Transfer learning has witnessed a recent surge of interest after proving successful in multiple applications. However, it highly re- lies on the quantity of annotated data. Constrained by the labor cost and expertise, it is hard to annotate sufficient organs and tumors at the voxel level for medical image segmentation. Consequently, most benchmark datasets were collected for the segmentation of only one type of organ and/or tumors, and all task-irrelevant organs and tumors were annotated as the background. We aim to make use of these partially but plentifully labeled datasets to boost the segmentation performance of the annotation-limited KiTS21 segmentation task. To this end, we first construct a general medical image segmentation model that learns to segment these partially labeled organs or tumors. Then we transfer its pretrained weights to a specific downstream task, i.e., KiTS21. The primary experiments demonstrate the effectiveness of the proposed transfer learning strategy.