Transfer Learning for KiTS21 Challenge

Xi Yang,Jianpeng Zhang,Jing Zhang,Yong Xia
DOI: https://doi.org/10.1007/978-3-030-98385-7_21
2022-01-01
Abstract:Transfer learning has witnessed a recent surge of interest after proving successful in multiple applications. However, it highly relies 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 bench-mark datasets were collected for the segmentation of only one type of organ and/or tumor, 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. Our method achieves 0.890 Dice score, 0.805 SurfaceDice, and 0.822 Tumor Dice in the KiTS21 challenge.
What problem does this paper attempt to address?