
2024年最大的图像分割数据集是怎么来的?——图像分割SAM模型
2024年最大的图像分割数据集是怎么来的?
Segment Anything Model (SAM):Meta AI 的一种新的 AI 模型,只需单击一下即可“剪切”任何图像中的任何对象。它已经在 1100 万张图像和 11 亿个掩码的数据集上进行了训练,在各种分割任务中具有很强的zero shot性能。
目录
一、引言
自然语言处理领域,基础模型通过零样本迁移学习快速使用到其他任务。
ChatGPT就是使用提示法进行交互的模型。几乎所有的文本都可以用来序列预测,因为序列预测不需要标签。
为什么计算机视觉没有泛用性强的零样本迁移学习模型?
图片资源没有标注分界框和分割掩码,创建基础模型是一个挑战。
SAM模型在数据收集循环中使用,建立庞大的分割数据集,在1100万张特许和尊重隐私的图像上有超过10亿个掩码。该模型的设计和训练是可提示的,因此可以zero-shot的方式转移到新的图像分布和任务。
SAM的目标是创建一个类似NLP领域的foundationmodel(例如GPT),它是一种使用prompt engineering机制的预训练模型并可以灵活地解决下游任务。
其中提示Prompt的类型:
• 一组前景、背景 标记点• 粗略的 标记框 或者 掩码• 自由格式的 文字
无论哪种情况,模型都应该足够灵活,输出分割掩码!
二、SAM模型技术原理
1.模型特色
(1)多返回对象
可提示分割任务是在给定任何提示的情况下返回有效的分割掩码。有效掩码的要求意味着即使提示不明确,并且可能涉及多个对象,输出也应该是其中至少一个对象的合理掩码。
每一列都是由SAM模型产生的3种有效输出
(2)预训练
可提示分割任务提出了一种自然的预训练算法,该算法模拟每个训练样本的prompt序列(例如,点、框、掩码),并将模型的掩码预测与GroundTruth进行比较。
与前人的交互式分割工作不同:
交互式分割的目的是在 足够的用户输入 ( 多次 与用户交互并 不断优化 )后最终预测有效的掩码可提示分割的目的是始终预测 任何提示 的 有效掩码 ,即使提示不明确(与用户交互后产生 多个 预测结果)
(3)零样本学习Zero-shot transfer
预训练任务赋予了模型在推理时对任何提示做出适当响应的能力,因此下游任务可以通过设计适当的提示来解决。
例如,如果有一个目标检测任务识别出猫的边界框,则可以通过我们的模型使用框作为输入来解决猫实例分割任务
2. Segment Anything Model框架
这里对SAM的各个模块进行了一些中文解释,建议详情看图。
3.掩码质量评估
为了评估掩码质量,研究者随机抽取了500张图像(大约5万个掩码),在模型预测掩码的基础上对其进行专业矫正。
数据引擎:为了对庞大数据的图像进行掩码标注,作者开发了数据引擎。
模型标注数据:标注好的数据用来优化模型。以此循环,迭代优化模型以及数据质量。
数据开发分为以下阶段,
1、模型辅助手动标注:标注员细化掩码标签;
2、半自动标注阶段:为使标注者专注于不太突出的对象,首先SAM自动分割高置信度的掩码,然后向标注者展示预填充这些掩码的图像,并要求他们标注任何其他未标注的对象;
3、全自动阶段:已收集足够多的掩码来改进模型;已开发模糊感知模型,允许在有歧义的情况下预测有效的掩码。
4.应用分析
研究者表示SAM具有协助处理许多真实世界场景的能力。
对SAM在多种真实世界分割场景(例如,自然图像、农业、制造业、遥感和医疗健康场景)中的性能进行了评估。在像自然图像这样的常见场景中,它具有优秀的泛化能力;
而在低对比度的场景中,它的效果较差,而且在复杂场景中需要强大的知识。
三、SAM模型发展应用
SAM在图像处理领域的应用——软件场景
软件场景需要对图像编辑和修复进行操作,例如移除对象、填充对象和替换对象。
现有的修复工作,如利用DDPM模型做图像修复的repaint模型,需要对每个掩码进行精细的注释以达到良好的性能,这是一项劳动密集型的工作。
SAM可以通过简单的提示如点或框来生成准确的掩码,可以帮助辅助图像编辑场景。
Inpaint Anything算法:用户可以通过点击图像中的任何对象来选择它。对于对象移除,该流程由 SAM 和最先进的修复器组成;对于对象的填充和替换,第二步使用像 Stable Diffusion (SD) 这样的 AI 生成的内容模型。
参考文献:
《A Comprehensive Survey on Segment Anything Model for Vision and Beyond》
链接:
https://arxiv.org/pdf/2305.08196.pdf
SAM在图像处理领域的应用——视觉相关
随着 ViT 的出现,基于ViT的医学图像分割方法,在医学图像分割中取得了超越以往的性能。然而,这类网络面向的是特定的任务,缺乏对其他任务的泛化能力。
在医学图像分割任务中,SAM 的表现并不理想,主要原因在于训练数据的匮乏。尽管 SAM 在训练过程中构建了一个复杂而高效的数据引擎,但所收集到的病例数据量有限,且很少具备实际医疗应用价值。
一种解决方法是针对下游任务进行微调,自适应策略称为医疗 SAM 适配器采用一种称Adaption 的参数高效微调( parameter-efficient fine-tuning,PEFT)技术,即将几个参数高效Adapter 模块插入到原始的基本模型中,然后只调整 Adapter 参数,同时冻结所有预先训练的参数。
医疗图像 SAM 概览,包括了CT图像、MRI图像、结肠镜检查图像、多格式图像、H&E 染色组织切片图像等
最后放上一些参考链接

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐
所有评论(0)