论文标题
使用对比度学习的标签有效的多任务分段
Label-Efficient Multi-Task Segmentation using Contrastive Learning
论文作者
论文摘要
尽管其对自动化分段任务的重要性，但获得3D医学图像的注释既昂贵又耗时。尽管多任务学习被认为是使用少量注释数据训练分割模型的有效方法，但仍缺乏对各种子任务的系统理解。在这项研究中，我们提出了一个具有基于对比学习的子任务的多任务分割模型，并将其与其他多任务模型进行比较，从而改变了用于培训的标记数据的数量。我们进一步扩展了模型，以便它可以通过半监督的方式通过正则化分支来利用未标记的数据。我们通过实验表明，当注释数据的量受到限制时，我们提出的方法优于其他多任务方法，包括最先进的完全监督模型。
Obtaining annotations for 3D medical images is expensive and time-consuming, despite its importance for automating segmentation tasks. Although multi-task learning is considered an effective method for training segmentation models using small amounts of annotated data, a systematic understanding of various subtasks is still lacking. In this study, we propose a multi-task segmentation model with a contrastive learning based subtask and compare its performance with other multi-task models, varying the number of labeled data for training. We further extend our model so that it can utilize unlabeled data through the regularization branch in a semi-supervised manner. We experimentally show that our proposed method outperforms other multi-task methods including the state-of-the-art fully supervised model when the amount of annotated data is limited.
