arXiv:2505.22019cs.CLcs.AI2025-05NeurIPS被引 71

用强化学习让视觉大模型更懂图文信息,自主规划搜索与推理路径。

VRAG-RL: Empower Vision-Perception-Based RAG for Visually Rich Information Understanding via Iterative Reasoning with Reinforcement Learning

  • 设计视觉感知动作空间,支持裁剪缩放,实现从粗到精的视觉信息获取
  • 通过奖励机制融合查询重写与检索效果,提升搜索相关性
  • 适合需要理解复杂图文内容的研究者和开发者

有效检索、推理与理解视觉丰富信息仍是RAG方法的挑战。传统文本方法无法处理视觉信息,现有视觉RAG常受限于固定流程,且因模型能力激活不足而难以有效推理。我们提出VRAG-RL,一种针对视觉丰富信息的强化学习框架。该框架使视觉语言模型(VLM)能与搜索引擎交互,借助视觉感知标记自主采样单轮或多轮推理轨迹,并基于样本持续优化。针对两大关键问题:(i) 多模态RAG通常仅将图像嵌入上下文,导致推理令牌分配不足且忽略视觉特异性;(ii) 模型与检索器交互时,因无法准确表达需求而难以获取相关结果。为此,我们设计了专用于视觉输入的动作空间,包含裁剪与缩放等操作,实现从粗到精的信息采集。同时,引入融合查询重写、检索性能与模型自评的奖励机制,弥合用户原始提问与检索器之间的差距。实验表明,该框架显著提升复杂图文理解任务表现,代码已开源。

原文摘要 · Abstract (English)

Effectively retrieving, reasoning and understanding visually rich information remains a challenge for RAG methods. Traditional text-based methods cannot handle visual-related information. On the other hand, current vision-based RAG approaches are often limited by fixed pipelines and frequently struggle to reason effectively due to the insufficient activation of the fundamental capabilities of models. As RL has been proven to be beneficial for model reasoning, we introduce VRAG-RL, a novel RL framework tailored for complex reasoning across visually rich information. With this framework, VLMs interact with search engines, autonomously sampling single-turn or multi-turn reasoning trajectories with the help of visual perception tokens and undergoing continual optimization based on these samples. Our approach highlights key limitations of RL in RAG domains: (i) Prior Multi-modal RAG approaches tend to merely incorporate images into the context, leading to insufficient reasoning token allocation and neglecting visual-specific perception; and (ii) When models interact with search engines, their queries often fail to retrieve relevant information due to the inability to articulate requirements, thereby leading to suboptimal performance. To address these challenges, we define an action space tailored for visually rich inputs, with actions including cropping and scaling, allowing the model to gather information from a coarse-to-fine perspective. Furthermore, to bridge the gap between users' original inquiries and the retriever, we employ a simple yet effective reward that integrates query rewriting and retrieval performance with a model-based reward. Our VRAG-RL optimizes VLMs for RAG tasks using specially designed RL strategies, aligning the model with real-world applications. The code is available at https://github.com/Alibaba-NLP/VRAG.

视觉RAG强化学习多模态推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。