4 个月前

AutoSAM:通过重载提示编码器将SAM适应于医学图像

AutoSAM:通过重载提示编码器将SAM适应于医学图像

摘要

最近推出的Segment Anything模型(SAM)结合了巧妙的架构和大量的训练数据,从而获得了显著的图像分割能力。然而,该模型在处理分布外(OOD)领域的图像时,如医学图像,无法重现相同的效果。此外,尽管SAM可以通过掩码或一组点进行条件化,但有时可能需要一个完全自动化的解决方案。在这项工作中,我们用一个在同一输入图像上操作的编码器替换了SAM的条件化机制。通过添加这一编码器并在不进一步微调SAM的情况下,我们在多个医学图像和视频基准测试中取得了最先进的结果。这个新的编码器是通过冻结的SAM提供的梯度进行训练的。为了检查其内部知识并提供一种轻量级的分割解决方案,我们还学习了通过浅层反卷积网络将其解码为掩码的方法。

基准测试

基准方法指标
video-polyp-segmentation-on-sun-seg-easyAutoSAM
Dice: 0.753
S measure: 0.815
Sensitivity: 0.672
mean E-measure: 0.855
mean F-measure: 0.774
weighted F-measure: 0.716
video-polyp-segmentation-on-sun-seg-hardAutoSAM
Dice: 0.759
S-Measure: 0.822
Sensitivity: 0.726
mean E-measure: 0.866
mean F-measure: 0.764
weighted F-measure: 0.714

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
AutoSAM:通过重载提示编码器将SAM适应于医学图像 | 论文 | HyperAI超神经