让AI通过编辑图像逐步聚焦,提升表格图表理解能力
ReFocus: Visual Editing as a Chain of Thought for Structured Image Understanding
- 用代码控制图像编辑,引导模型分步聚焦关键区域
- 表格任务平均提升11.0%,图表任务提升6.8%
- 适合需要精细视觉推理的多跳理解场景
结构化图像理解(如读取表格和图表)需要在图像中跨多个结构与文本进行策略性注意力转移,形成推理链以得出最终答案。然而当前多模态大模型缺乏这种多跳选择性注意力能力。本文提出ReFocus,一种简单有效的框架,使多模态大模型能通过代码对输入图像进行视觉编辑,生成“视觉思维”过程,实现注意力的动态调整。具体而言,ReFocus让模型生成Python代码调用工具,依次绘制框选、高亮区域、遮蔽无关部分,从而优化视觉推理。我们在包含表格和图表的多种结构化图像理解任务上进行了实验,结果表明,相比无视觉编辑的GPT-4o,ReFocus在所有任务上均有显著提升,表格任务平均增益11.0%,图表任务增益6.8%。我们深入分析了不同视觉编辑的效果,并解释为何不引入额外信息也能提升性能。此外,我们基于ReFocus构建了14k训练数据集,证明带有中间信息的视觉思维链监督比标准VQA数据更优,使模型平均提升8.0%,优于仅用问答对训练的模型2.6%。
原文摘要 · Abstract (English)
Structured image understanding, such as interpreting tables and charts, requires strategically refocusing across various structures and texts within an image, forming a reasoning sequence to arrive at the final answer. However, current multimodal large language models (LLMs) lack this multihop selective attention capability. In this work, we introduce ReFocus, a simple yet effective framework that equips multimodal LLMs with the ability to generate "visual thoughts" by performing visual editing on the input image through code, shifting and refining their visual focuses. Specifically, ReFocus enables multimodal LLMs to generate Python codes to call tools and modify the input image, sequentially drawing boxes, highlighting sections, and masking out areas, thereby enhancing the visual reasoning process. We experiment upon a wide range of structured image understanding tasks involving tables and charts. ReFocus largely improves performance on all tasks over GPT-4o without visual editing, yielding an average gain of 11.0% on table tasks and 6.8% on chart tasks. We present an in-depth analysis of the effects of different visual edits, and reasons why ReFocus can improve the performance without introducing additional information. Further, we collect a 14k training set using ReFocus, and prove that such visual chain-of-thought with intermediate information offers a better supervision than standard VQA data, reaching a 8.0% average gain over the same model trained with QA pairs and 2.6% over CoT.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。