arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2312.07969cs.CV

ASLseg:适配循环中SAM的半监督肝肿瘤分割方法

ASLseg: Adapting SAM in the Loop for Semi-supervised Liver Tumor Segmentation

Southern University of Science and Technology(南方科技大学);The University of Hong Kong(香港大学);Jiaxing Research Institute, Southern University of Science and Technology(南方科技大学嘉兴研究院)

Shiyun Chen, Li Lin, Pujin Cheng, Xiaoying Tang

AI总结:

针对半监督肝肿瘤分割中SAM表现不佳、标注数据稀缺的问题,提出ASLseg框架,通过伪标签提示微调SAM、适配网络优化结果、筛选可靠伪标签迭代训练,在LiTS数据集上取得优异性能。

摘要:

肝肿瘤分割对计算机辅助诊断、手术规划和预后评估至关重要,但获取并维护带有密集标注的大规模数据集难度较大,半监督学习(SSL)是应对这类挑战的常用技术。近期,Segment Anything Model(SAM,分割一切模型)在部分医学图像分割任务中展现出可观性能,但在肝肿瘤分割任务上表现不佳。本文提出一种新型半监督框架ASLseg,可有效将SAM适配至半监督学习场景,融合肝肿瘤的领域特定知识与通用知识。具体而言,经特定半监督范式训练的分割模型会生成伪标签,将其作为提示输入至微调后的SAM;随后通过适配网络优化SAM的预测结果,生成质量更高的伪标签;最后筛选出可靠的伪标签扩充标注集,用于迭代训练。在LiTS数据集上开展的大量实验表明,ASLseg的性能具有显著优势。

英文摘要:

Liver tumor segmentation is essential for computer-aided diagnosis, surgical planning, and prognosis evaluation. However, obtaining and maintaining a large-scale dataset with dense annotations is challenging. Semi-Supervised Learning (SSL) is a common technique to address these challenges. Recently, Segment Anything Model (SAM) has shown promising performance in some medical image segmentation tasks, but it performs poorly for liver tumor segmentation. In this paper, we propose a novel semi-supervised framework, named ASLseg, which can effectively adapt the SAM to the SSL setting and combine both domain-specific and general knowledge of liver tumors. Specifically, the segmentation model trained with a specific SSL paradigm provides the generated pseudo-labels as prompts to the fine-tuned SAM. An adaptation network is then used to refine the SAM-predictions and generate higher-quality pseudo-labels. Finally, the reliable pseudo-labels are selected to expand the labeled set for iterative training. Extensive experiments on the LiTS dataset demonstrate overwhelming performance of our ASLseg.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑