arXiv:2604.13579cs.CL2026-04ACL被引 7

用多轮强化学习让智能体更懂长文档图文问答。

MM-Doc-R1: Training Agents for Long Document Visual Question Answering through Multi-turn Reinforcement Learning

论文配图:MM-Doc-R1: Training Agents for Long Document Visual Question Answering through Multi-turn Reinforcement Learning
图 1 · 摘自论文原文
  • 设计多轮迭代的视觉感知智能体,逐步挖掘长文档信息。
  • 在MMLongbench-Doc上比基线高10.4%,用Qwen3模型提升超5%。
  • 新算法SPO稳定训练,适合复杂图文问答研究者。

传统检索增强生成系统在长文档多跳查询上表现不佳,因其仅单次检索。我们提出MM-Doc-R1框架,采用具代理性质、视觉感知的工作流,通过迭代式信息发现与融合解决长文档图文问答问题。为激励智能体的信息搜索能力,提出相似性策略优化(SPO),解决现有多轮强化学习算法(如GRPO)中基线估计偏差问题。核心洞察是:多轮强化学习中,轨迹间语义越相似,其共享基线估计越准确。SPO通过轨迹间奖励的相似性加权平均计算更精确基线,优于GRPO将初始状态基线应用于所有中间状态的做法。这为智能体提供更稳定、准确的学习信号,显著提升训练性能。在MMLongbench-Doc基准测试中,MM-Doc-R1相较以往基线提升10.4%;SPO相较GRPO,在Qwen3-8B上提升5.0%,Qwen3-4B上提升6.1%。结果验证了该集成框架与新型训练算法在复杂长文档图文问答任务中的先进性。

原文摘要 · Abstract (English)

Conventional Retrieval-Augmented Generation (RAG) systems often struggle with complex multi-hop queries over long documents due to their single-pass retrieval. We introduce MM-Doc-R1, a novel framework that employs an agentic, vision-aware workflow to address long document visual question answering through iterative information discovery and synthesis. To incentivize the information seeking capabilities of our agents, we propose Similarity-based Policy Optimization (SPO), addressing baseline estimation bias in existing multi-turn reinforcement learning (RL) algorithms like GRPO. Our core insight is that in multi-turn RL, the more semantically similar two trajectories are, the more accurate their shared baseline estimation becomes. Leveraging this, SPO calculates a more precise baseline by similarity-weighted averaging of rewards across multiple trajectories, unlike GRPO which inappropriately applies the initial state's baseline to all intermediate states. This provides a more stable and accurate learning signal for our agents, leading to superior training performance that surpasses GRPO. Our experiments on the MMLongbench-Doc benchmark show that MM-Doc-R1 outperforms previous baselines by 10.4%. Furthermore, SPO demonstrates superior performance over GRPO, boosting results by 5.0% with Qwen3-8B and 6.1% with Qwen3-4B. These results highlight the effectiveness of our integrated framework and novel training algorithm in advancing the state-of-the-art for complex, long-document visual question answering.

图文问答强化学习长文档

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