在视觉提示中加入「标记」，微软等让GPT-4V看的更准、分的更细

　　本文来自于微信公众号 机器之心(ID:almosthuman2014)，作者:机器之心。

　　最近一段时间，我们见证了大型语言模型(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 在细粒度视觉任务上的问题。

　　如图1(右)所示，SoM 采用交互式分割模型(例如 SAM)将图像划分为不同粒度级别的区域，并在这些区域上添加一组标记(mark)，例如字母数字、掩码(mask)、框(box)。使用添加标记的图像作为输入，以解决上述问题。

　　我们先来看下效果，左为 GPT-4V，右为 GPT-4V+SoM，很明显后者分类更细致、准确。

　　下图示例依然如此，GPT-4V+SoM 效果更明显。

　　此外，对于这项研究，有人问道:「SoM 是手动(人工输入)还是自动的?」

　　论文一作 Jianwei Yang 表示，SoM 是自动或半自动的。他们编译了很多自己构建自己的分割工具，比如 SEEM、Semantic-SAM 和 SAM，用来帮助用户自动为图像分割区域。同时用户也可以自己选择区域。

　　使用 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.8J&F)。

　　消融研究

　　研究者探讨了标记类型如何影响 Flickr30k 数据集上短语关联任务的最终性能，并比较了两种类型的标记。第一种是数字和掩码，第二种是数字、掩码和框。

　　结果如下表3所示，添加额外的框可以显著提升性能。

　　此外研究者探究了当生成带有真值注释的标记时，GPT-4V 如何表现。他们选择在 RefCOCOg 验证集中用真值掩码替换预测到的分割掩码。这意味着 GPT-4V 只需要从注释短语区域选择一个即可。如预期一样，参考分割的性能可以得到进一步提升，尤其是当分割模型有一些缺失的区域。

　　结果如下表4所示，在 SoM 中使用真值掩码可以将 RefCOCOg 上的性能提升14.5%(mIoU)。