arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.05902cs.CVcs.AIphysics.med-ph
收藏

ClickSAM:使用点击提示微调Segment Anything模型用于超声图像分割

ClickSAM: Fine-tuning Segment Anything Model using click prompts for ultrasound image segmentation

The Hockaday School(霍克黛学校);Liberty High School(自由高中);The University of Texas at Dallas(德克萨斯大学达拉斯分校);The University of Texas Southwestern Medical Center at Dallas(德克萨斯大学达拉斯西南医学中心)

Aimee Guo, Grace Fei, Hemanth Pasupuleti, Jing Wang

AI总结:

ClickSAM通过两阶段点击提示微调SAM,利用真值掩码对比生成正负点击并结合Centroidal Voronoi Tessellation算法,显著提升超声图像分割性能。

摘要:

新发布的Segment Anything Model(SAM)因其卓越的分割精度、多样的输入提示、训练能力以及高效的模型设计而成为图像处理中的热门工具。然而,其当前模型是在一个并非针对医学图像(尤其是超声图像)定制的多样化数据集上训练的。超声图像往往包含大量噪声,使得分割重要结构变得困难。在本项目中,我们开发了ClickSAM,它使用点击提示对Segment Anything模型进行微调,以用于超声图像。ClickSAM包含两个训练阶段:第一阶段在位于真值轮廓中心的单次点击提示上进行训练,第二阶段则侧重于通过额外的正点击和负点击提示来提升模型性能。通过将第一阶段的预测与真值掩码进行比较,计算出真阳性、假阳性和假阴性区域。正点击利用真阳性和假阴性区域生成,负点击利用假阳性区域生成。随后采用Centroidal Voronoi Tessellation算法在每个区域中收集正点击和负点击提示,用于在第二阶段训练中增强模型性能。借助点击训练方法,ClickSAM在超声图像分割中展现出优于其他现有模型的性能。

英文摘要:

The newly released Segment Anything Model (SAM) is a popular tool used in image processing due to its superior segmentation accuracy, variety of input prompts, training capabilities, and efficient model design. However, its current model is trained on a diverse dataset not tailored to medical images, particularly ultrasound images. Ultrasound images tend to have a lot of noise, making it difficult to segment out important structures. In this project, we developed ClickSAM, which fine-tunes the Segment Anything Model using click prompts for ultrasound images. ClickSAM has two stages of training: the first stage is trained on single-click prompts centered in the ground-truth contours, and the second stage focuses on improving the model performance through additional positive and negative click prompts. By comparing the first stage predictions to the ground-truth masks, true positive, false positive, and false negative segments are calculated. Positive clicks are generated using the true positive and false negative segments, and negative clicks are generated using the false positive segments. The Centroidal Voronoi Tessellation algorithm is then employed to collect positive and negative click prompts in each segment that are used to enhance the model performance during the second stage of training. With click-train methods, ClickSAM exhibits superior performance compared to other existing models for ultrasound image segmentation.

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

↑