全新视觉提示方法 SoM(Set-of-Mark),让 OpenAI 多模态大模型 GPT-4V 在视觉内容理解方面有了质的提升。在视觉提示中加入「标记」,微软等让GPT-4V看的更准、分的更细
最近一段时间,我们见证了大型语言模型(LLM)的显著进步。特别是,生成式预训练 Transformer 或 GPT 的发布引领了业界和学术界的多项突破。自 GPT-4 发布以来,大型多模态模型 (LMM) 引起了研究界越来越多的兴趣,许多工作致力于构建多模态 GPT-4。
近日,GPT-4V (ision) 由于出色的多模态感知和推理能力得到了大家格外的关注。然而,尽管 GPT-4V 具有前所未有的视觉语言理解能力,但其细粒度 visual grounding(输入是图片和对应的物体描述,输出是描述物体的 box)能力相对较弱,或者尚未发挥出来。
举例来说,当用户询问下图中「放置在右边笔记本电脑的左边是什么物体?」GPT-4V 给出了马克杯这个错误的答案。当用户接着询问,「想找一个靠窗的座位,我可以坐在哪里?」GPT-4V 同样回答不正确。
在意识到上述问题后,来自微软、香港科技大学等机构的研究者提出了一种新的视觉 prompt 方法 Set-of-Mark(SoM),来解决 GPT-4V 在细粒度视觉任务上的问题。
- 论文地址:https://arxiv.org/pdf/2310.11441.pdf
- 论文主页:https://som-gpt4v.github.io/
如图 1(右)所示,SoM 采用交互式分割模型(例如 SAM)将图像划分为不同粒度级别的区域,并在这些区域上添加一组标记(mark),例如字母数字、掩码(mask)、框(box)。使用添加标记的图像作为输入,以解决上述问题。
我们先来看下效果,左为 GPT-4V,右为 GPT-4V+SoM,很明显后者分类更细致、准确。
下图示例依然如此,GPT-4V+SoM 效果更明显。
此外,对于这项研究,有人问道:「SoM 是手动(人工输入)还是自动的?」
论文一作 Jianwei Yang 表示,SoM 是自动或半自动的。他们编译了很多自己构建自己的分割工具,比如 SEEM、Semantic-SAM 和 SAM,用来帮助用户自动为图像分割区域。同时用户也可以自己选择区域。
用于视觉的 SoM prompt
使用 SoM prompt GPT-4V 的独特优点是它可以产生文本之外的输出。由于每个标记都与掩码表征的图像区域特定关联,因此可以追溯文本输出中任何提到的标记的掩码。
生成成对文本和掩码的能力使 SoM 能够 prompt GPT-4V 来生成视觉关联的文本,更重要的是支持各种细粒度视觉任务,这对普通的 GPT-4V 模型来说是一个挑战。
通过简单的 prompt 工程,SoM 可以让 GPT-4V 广泛地用于多种视觉任务,例如:
- 开放词汇图像分割:该研究要求 GPT-4V 详尽地给出所有标记区域的类别以及从预定池中选择的类别。
- 参考分割:给定一个参考表达式,GPT-4V 的任务是从图像分区工具箱生成的候选区域中选择最匹配的区域。
- 短语关联(Phrase Grounding):与参考分割略有不同,短语关联使用由多个名词短语组成的完整句子。该研究要求 GPT-4V 为所有标记的短语分配相应的区域。
- 视频对象分割:以两个图像作为输入。第一个图像是查询图像,其中包含第二个图像中需要识别的一些对象。鉴于 GPT-4V 支持多个图像作为输入,因此 SoM 也可以应用于视频中跨帧的关联视觉对象。
实验及结果
研究者使用「分而治之」(divide-and-conquer)的策略来运行实验和评估。对于每个实例,他们使用新的聊天窗口,这样一来,评估期间就不会出现上下文泄露了。
具体来讲,研究者从每个数据集中选择了小规模的验证数据子集。对于数据集中的每个图像,他们在使用图像分割工具箱提取的区域上覆盖了一组标记。同时基于具体的任务,研究者利用不同的分割工具来提出区域。
下表 1 列出了每个任务的设置细节。
研究者将其方法与以下模型进行比较:
- 预测坐标的 GPT-4V 基线模型
- SOTA 专用模型
- 开源 LMM
定量结果
详细的实验结果如下表 2 所示。
首先是图像分割任务。研究者将 GPT-4V + SoM 与 COCO Panoptic 分割数据集上的强大分割模型 MaskDINO、ADE20K Panoptic 分割数据集上的模型 OpenSeeD 进行了比较。
结果显示,GPT-4V + SoM 的零样本性能接近微调后的 MaskDINO,并大幅优于 OpenSeeD。GPT-4V 在 COCO 和 ADE20K 上的相似性能表现出其对广泛视觉和语义域任务的强大泛化能力。
然后是参考(referrring)任务,研究者评估了 RefCOCOg 数据集上的模型 RES 和 REC。他们使用 MaskDINO 来提出掩码,并在图像上覆盖上掩码和数字。同时使用 mIoU 作为评估指标,并与 SOTA 专用模型 PolyFormer 和 SEEM 进行比较。
结果显示,GPT-4V+SoM 击败了 Grounding DINO、Polyformer 等专用模型以及 Shikra、LLaVA-1.5、MiniGPT-v2 和 Ferret 等最近的开源 LMM。
接着是 Flickr30K 上的短语关联任务,研究者使用 Grounding DINO 为每个图像生成框建议。GPT-4V+SoM 实现了比 GLIPv2 和 Grounding DINO 更强的零样本性能。
最后研究者在 DAVIS2017 数据集上评估了视频分割任务。GPT-4V+SoM 实现了优于其他专用视觉模型的最佳追踪性能(78.8 J&F)。
消融研究
研究者探讨了标记类型如何影响 Flickr30k 数据集上短语关联任务的最终性能,并比较了两种类型的标记。第一种是数字和掩码,第二种是数字、掩码和框。
结果如下表 3 所示,添加额外的框可以显著提升性能。
此外研究者探究了当生成带有真值注释的标记时,GPT-4V 如何表现。他们选择在 RefCOCOg 验证集中用真值掩码替换预测到的分割掩码。这意味着 GPT-4V 只需要从注释短语区域选择一个即可。如预期一样,参考分割的性能可以得到进一步提升,尤其是当分割模型有一些缺失的区域。
结果如下表 4 所示,在 SoM 中使用真值掩码可以将 RefCOCOg 上的性能提升 14.5%(mIoU)。