豆包大模型团队发布全新Detail Image Caption评估基准，提升VLM Caption评测可靠性

2024-07-13 04:07

AIxiv专栏是机器之心发布学术、技术内容的栏目。过去数年，机器之心AIxiv专栏接收报道了2000多篇内容，覆盖全球各大高校与企业的顶级实验室，有效促进了学术交流与传播。如果您有优秀的工作想要分享，欢迎投稿或者联系报道。投稿邮箱：[email protected]；[email protected]

当前的视觉语言模型（VLM）主要通过 QA 问答形式进行性能评测，而缺乏对模型基础理解能力的评测，例如 detail image caption 性能的可靠评测手段。

针对这一问题，中科院、北大和字节豆包大模型团队发布了 DetailCaps-4870 数据集，并提出了一种有效的评估指标 CAPTURE，取得了开源评估指标中最高的专家评价一致性，并低成本实现了与 GPT-Eval 可比的效果。

论文：https://arxiv.org/abs/2405.19092
数据集：https://huggingface.co/datasets/foundation-multimodal-models/DetailCaps-4870
代码：https://github.com/foundation-multimodal-models/CAPTURE

简介

当前的 LVLM（large vision-language model）评测存在以下问题：

现有的 LVLM 评测方案主要采用 VQA 形式，很大程度受到指令遵循（instruction following）能力的影响，且 QA prompt 的设计容易引入人类的偏见（bias）。
Image caption 任务可以有效评估模型理解能力，但现有的 caption benchmark 多使用短 caption 作为 ground truth，这在 lvlm 时代完全过时。
同时，现有的 image caption 评测指标与人类、GPT 等专家评价结果的一致性较差，常用的 bleu、rouge 等指标抽取 n-gram 进行匹配，对关键信息的准确性不够敏感。而 GPT-Eval 虽然和专家评价较为一致，但是会带来高昂的评测成本。

针对这些问题，该研究提出了新的 Detail image caption benchmark 和 evaluation metric，以在较低的成本下实现了对 LVLM 图像理解能力的准确评估。

在所提出的评测数据集 & 指标的指导下，该研究还探索了发掘 LVLM 自身能力进行 detail image caption 的数据构造方法，有效提升了 detail caption 数据质量。

图一：左侧为 CAPTURE metric 实例展示，右侧为 detail caption construction 方法展示

数据集

相比类似大小的评测数据集，该研究提出的 Detail image caption benchmark 文本长度更长，无重复的 2-gram 数量明显更多，包含更丰富的视觉信息：

表一：DetailCaps benchmark 统计信息

评估指标

该研究所提出的 CAPTURE (CAPtion evaluation by exTracting and coUpling coRE information) 指标通过 4 步进行 caption 质量评估。如下图所示，首先使用 Factual praser [1] 抽取 detail caption 中的 object, attribute, relation 元素，紧接着过滤掉没有实际意义的 objects。之后，通过三阶段匹配（精确匹配、近义词匹配、embedding 匹配）分别计算 obj、attr、rel 元素的匹配得分（F1 分数），加权作为最终结果。

图二：CAPTURE metric 详细计算步骤

Detail caption 数据合成

在 DetailCaps benchmark 和 CAPTURE metric 的指导下，该研究提出了一种基于 divide-and-conquer 思想发掘 LVLM 潜力进行数据合成的方法，有效提升了 detail caption 数据质量。该方案首先使用 LVLM 进行全图 caption 生成，之后使用分割模型（SAM [2]）和聚类等过滤方法找到图中的关键位置并将其裁剪出来进行局部 caption 生成。文章使用一种 word-level 的过滤方法来减少 caption 中的幻觉，该方法首先解析（parse）出描述图中视觉元素的词和短语，再通过目标检测模型（Owlv2 [3]）过滤掉得分低的元素。最后，将过滤后的全图 caption、局部 caption 送入 LLM（LLaMA2 [4]）融合为最终的图像描述。

实验

CAPTURE 指标

（1）CAPTURE vs 其他 caption 指标

该研究在 DetailCaps-100（人工标注 reference caption，人工评估模型生成 caption 得分）和 DetailCaps-4870（GPT 标注 reference caption，GPT-4 评估模型生成 caption 得分）上，对 LLaVA-1.5 [5]、CogVLM [6] 和 ShareCaptioner [7] 三个模型生成的 caption 进行了专家打分，并计算各个评估指标与专家评价的一致性：

表二：CAPTURE 与其他 detail image caption 评估指标的专家评价一致性对比

各 caption 评估指标和专家评价的一致性通过 pearson correlation（线性相关性）、 R^2（绝对值大小）、 Kendall's tau（排序偏序对一致性）和 (Per-) Sample (Kendall's) tau（每个样本单独计算取平均）指标进行衡量。

结果显示，CAPTURE 在各种指标上都取得了最优的专家评价一致性。在这些指标中，Sample tau 与实际 detail image caption 评估的计算方式最为接近，CAPTURE 也是唯一在这个指标上取得与 GPT4-Eval 接近的方法，在评估的准确率、成本上取得了良好的平衡。

（2）消融分析

研究者还对 CAPTURE 中的各个模块进行了消融分析，并验证其有效性：

表三：CAPTURE 各模块的消融分析

实验结果显示，Stop words 有效提升了 Sample tau，说明了该模块的有效性。但是 stop words filtering 会对不同样例的 detail caption 造成不同的影响，从而导致 pcc、kendall tau 微降。Soft matching 也提升了 sample tau，同时在 1-R2 分数上有明显增益作用，将 CAPTURE 预测分数和专家打分的绝对分值对齐。在加权计算最终得分时，obj:attr:rel 比例取 default 的 5:5:2 最好，提升或降低各元素的比重都会造成性能下降。

（3）开源 LVLM 的 detail caption 性能

表四：开源 LVLM 的 detail caption 性能对比

总体来看，InternVL-1.5 是当前表现最好的开源 LVLM。从 LLaVA、MiniGemini 结果可以看出，LLM 参数量的提升对模型 detail caption 能力有一致的提升作用。同时，分辨率较高、使用高质量 detail caption 进行训练的模型，性能会更好。

Detail caption 数据构造

基于 detail caption 评估数据集和评测指标，研究者验证了所提出的 detail caption 数据合成方案的有效性。

（1）Detail caption 合成方法在不同 LVLM 上的有效性

如下表所示，该研究提出的 detail caption 合成方法在 LLaVA-1.5-7B、LLaVA-1.5-13B、LLaVA-NEXT-7B 和 Mini-Gemini-7B-HD 上取得了一致的 detail caption 质量提升：

表五：不同 LVLM 使用该研究提出的 detail caption 合成方法效果

（2）通过 Self-loop 进一步提升 detail caption 性能

研究者还尝试通过数据打标 -> 模型训练 -> 重新打标的训练流程进行 Self-loop 来进一步提升 LVLM detail caption 性能，在四轮 loop 中都取得了正面效果。同时，对比开源方案 [8] 与本文所提出的 word-level 幻觉过滤方案，证明了其设计的有效性：

表六：Detail caption 合成方案的 self-looping 效果和消融分析

（3）LVLM 自身打标的 detail caption 可提升其整体性能

该研究使用 LLaVA-1.5 按照给出的 detail caption 构造方案对 sharegpt4v-100k 数据进行了重新打标，并将打标数据用于 LLaVA-1.5 的 SFT 训练，在多个 benchmark 上取得了一致的性能提升：

表七：合成 detail caption 数据在 LLaVA-1.5-7B 模型训练中的作用

参考文献

[1] Zhuang Li, Yuyang Chai, Terry Zhuo Yue, Lizhen Qu, Gholamreza Haffari, Fei Li, Donghong Ji, and Quan Hung Tran. Factual: A benchmark for faithful and consistent textual scene graph parsing. arXiv:2305.17497, 2023

[2] Alexander Kirillov, Eric Mintun, Nikhila Ravi, Hanzi Mao, Chloe Rolland, Laura Gustafson, Tete Xiao, Spencer Whitehead, Alexander C Berg, Wan-Yen Lo, et al. Segment anything. ICCV 2023

[3] Matthias Minderer, Alexey Gritsenko, and Neil Houlsby. Scaling open-vocabulary object detection. NIPS 2024

[4] Hugo Touvron, Louis Martin, Kevin Stone, Peter Albert, Amjad Almahairi, Yasmine Babaei, Nikolay Bashlykov, Soumya Batra, Prajjwal Bhargava, Shruti Bhosale, et al. Llama 2: Open foundation and fine-tuned chat models. arXiv:2307.09288, 2023

[5] Haotian Liu, Chunyuan Li, Yuheng Li, and Yong Jae Lee. Improved baselines with visual instruction tuning. NeurIPS 2023 Workshop on Instruction Tuning and Instruction Following, 2023

[6] Weihan Wang, Qingsong Lv, Wenmeng Yu, Wenyi Hong, Ji Qi, Yan Wang, Junhui Ji, Zhuoyi Yang, Lei Zhao, Xixuan Song, Jiazheng Xu, Bin Xu, Juanzi Li, Yuxiao Dong, Ming Ding, and Jie Tang. Cogvlm: Visual expert for pretrained language models. arXiv:2311.03079, 2023

[7] Lin Chen, Jisong Li, Xiaoyi Dong, Pan Zhang, Conghui He, Jiaqi Wang, Feng Zhao, and Dahua Lin. Sharegpt4v: Improving large multi-modal models with better captions. arXiv:2311.12793, 2023

[8] Zhang Li, Biao Yang, Qiang Liu, Zhiyin Ma, Shuo Zhang, Jingxu Yang, Yabo Sun, Yuliang Liu, and Xiang Bai. Monkey: Image resolution and text label are important things for large multi-modal models. arXiv:2311.06607, 2023

豆包大模型团队

字节跳动豆包大模型团队成立于 2023 年，致力于开发业界最先进的 AI 大模型技术，成为世界一流的研究团队，为科技和社会发展作出贡献。

豆包大模型团队在 AI 领域拥有长期愿景与决心，研究方向涵盖 NLP、CV、语音等，在中国、新加坡、美国等地设有实验室和研究岗位。团队依托平台充足的数据、计算等资源，在相关领域持续投入，已推出自研通用大模型，提供多模态能力，下游支持豆包、扣子、即梦等 50 + 业务，并通过火山引擎开放给企业客户。目前，豆包 APP 已成为中国市场用户量最大的 AIGC 应用。欢迎加入字节跳动豆包大模型团队。

https://mp.weixin.qq.com/s/ZjQ-v6reZXhBP6G27cbmlQ

转载请联系本公众号获得授权

投稿或寻求报道：[email protected]

微信扫码关注该文公众号作者

戳这里提交新闻线索和高质量文章给我们。

来源: qq

点击查看作者最近其他文章