中文

ASLseg:在循环中适配SAM用于半监督肝脏肿瘤分割

计算机视觉与模式识别 2024-05-21 v2

摘要

肝脏肿瘤分割对于计算机辅助诊断、手术规划和预后评估至关重要。然而,获取并维护一个带有密集标注的大规模数据集极具挑战性。半监督学习(SSL)是应对这些挑战的常用技术。最近,Segment Anything Model(SAM)在一些医学图像分割任务中展现出良好的性能,但在肝脏肿瘤分割方面表现不佳。在本文中,我们提出了一种新颖的半监督框架,名为ASLseg,它可以有效地将SAM适配到SSL设置中,并结合肝脏肿瘤的领域特定知识和通用知识。具体来说,通过特定SSL范式训练的分割模型提供生成的伪标签作为提示,输入到微调后的SAM。然后,使用一个适配网络来细化SAM的预测结果并生成更高质量的伪标签。最后,选择可靠的伪标签来扩展已标注集,以进行迭代训练。在LiTS数据集上的大量实验证明了我们的ASLseg具有压倒性的性能。

关键词

引用

@article{arxiv.2312.07969,
  title  = {ASLseg: Adapting SAM in the Loop for Semi-supervised Liver Tumor Segmentation},
  author = {Shiyun Chen and Li Lin and Pujin Cheng and Xiaoying Tang},
  journal= {arXiv preprint arXiv:2312.07969},
  year   = {2024}
}