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

Polyp-SAM:迁移SAM用于息肉分割

Polyp-SAM: Transfer SAM for Polyp Segmentation

Emory University(埃默里大学);Winship Cancer Institute(温希普癌症研究所);School of Medicine, Emory University(埃默里大学医学院)

Yuheng Li, Mingzhe Hu, Xiaofeng Yang

AI总结:

针对结肠镜数据有限导致息肉分割网络训练困难的问题,本研究提出微调的Polyp-SAM模型,对比不同迁移策略,在五个公开数据集上均取得优异表现,验证了SAM适配医学图像分割的潜力。

摘要:

结肠息肉被认为是结直肠癌的重要前驱病变。结肠息肉的自动分割可显著降低结肠癌的误诊率,并提高医生的标注效率。尽管已有许多息肉分割方法被提出,但利用有限的结肠镜数据训练大规模分割网络仍是一项挑战。近期,Segment Anything Model(SAM,分割一切模型)在自然图像和医学图像分割领域均受到广泛关注。SAM在多个图像基准测试中表现出优异性能,因此在医学图像分割方面展现出巨大潜力。本研究提出了Polyp-SAM——一种用于息肉分割的微调SAM模型,并将其性能与多种先进的息肉分割模型进行对比。我们还比较了SAM的两种迁移学习策略:微调编码器与不微调编码器。在五个公开数据集上的评估结果显示,我们的Polyp-SAM在两个数据集上达到了最优性能,在另外三个数据集上也表现出色,所有数据集的dice系数均高于88%。本研究证明了将SAM适配到医学图像分割任务的巨大潜力。我们计划在以下地址发布本文的代码和模型权重:https://github.com/ricklisz/Polyp-SAM。

英文摘要:

Colon polyps are considered important precursors for colorectal cancer. Automatic segmentation of colon polyps can significantly reduce the misdiagnosis of colon cancer and improve physician annotation efficiency. While many methods have been proposed for polyp segmentation, training large-scale segmentation networks with limited colonoscopy data remains a challenge. Recently, the Segment Anything Model (SAM) has recently gained much attention in both natural and medical image segmentation. SAM demonstrates superior performance in several image benchmarks and therefore shows great potential for medical image segmentation. In this study, we propose Poly-SAM, a finetuned SAM model for polyp segmentation, and compare its performance to several state-of-the-art polyp segmentation models. We also compare two transfer learning strategies of SAM with and without finetuning its encoders. Evaluated on five public datasets, our Polyp-SAM achieves state-of-the-art performance on two datasets and impressive performance on three datasets, with dice scores all above 88%. This study demonstrates the great potential of adapting SAM to medical image segmentation tasks. We plan to release the code and model weights for this paper at: https://github.com/ricklisz/Polyp-SAM.

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

↑