arXiv:2510.16907cs.AIcs.CL2025-10中稿 · NeurIPS被引 61

让视觉语言模型通过显式推理构建世界模型,显著提升多轮任务表现。

VAGEN: Reinforcing World Model Reasoning for Multi-Turn VLM Agents

  • 将视觉状态推理拆分为状态估计与变化预测,增强世界建模能力。
  • 30亿参数模型在五项基准上达0.82分,是未训练模型的3倍以上。
  • 适合研究视觉推理、多轮智能体或强化学习的开发者使用。

相较于语言模型代理,视觉语言模型(VLM)代理面临从文本状态到复杂视觉观察的转变,带来部分可观测性问题,需依赖强大的世界建模能力。本文探讨:VLM代理能否通过显式视觉状态推理构建内部世界模型?为此,我们采用强化学习(RL)架构化地强制并奖励代理的推理过程,将其形式化为部分可观测马尔可夫决策过程(POMDP)。实验表明,将推理分解为状态估计(当前状态是什么?)和过渡建模(接下来会发生什么?)对成功至关重要,通过五种推理策略验证。研究发现,内部信念的最优表示方式取决于任务:自然语言在一般任务中更优,结构化格式则在精确操控任务中不可或缺。基于此,我们设计了密集的逐回合世界建模奖励,实现精准状态预测,并引入双层广义优势估计(Bi-Level GAE),实现逐回合信用分配。在该视觉状态推理框架下,一个30亿参数模型在五个多样化代理基准上达到0.82分,相较未训练模型(0.21)提升3倍,超越GPT-5(0.75)、Gemini 2.5 Pro(0.67)和Claude 4.5(0.62)。所有实验均在可扩展的VAGEN框架中完成,支持多轮VLM代理在多样化视觉环境中的训练与分析。代码与数据已公开于https://vagen-ai.github.io。

原文摘要 · Abstract (English)

A key challenge in training Vision-Language Model (VLM) agents, compared to Language Model (LLM) agents, lies in the shift from textual states to complex visual observations. This transition introduces partial observability and demands robust world modeling. We ask: Can VLM agents construct internal world models through explicit visual state reasoning? To address this question, we architecturally enforce and reward the agent's reasoning process via reinforcement learning (RL), formulating it as a Partially Observable Markov Decision Process (POMDP). We find that decomposing the agent's reasoning into State Estimation ("what is the current state?") and Transition Modeling ("what comes next?") is critical for success, as demonstrated through five reasoning strategies. Our investigation into how agents represent internal beliefs reveals that the optimal representation is task-dependent: Natural Language excels at capturing semantic relationships in general tasks, while Structured formats are indispensable for precise manipulation and control. Building on these insights, we design a World Modeling Reward that provides dense, turn-level supervision for accurate state prediction, and introduce Bi-Level General Advantage Estimation (Bi-Level GAE) for turn-aware credit assignment. Through this form of visual state reasoning, a 3B-parameter model achieves a score of 0.82 across five diverse agent benchmarks, representing a 3$\times$ improvement over its untrained counterpart (0.21) and outperforming proprietary reasoning models such as GPT-5 (0.75), Gemini 2.5 Pro (0.67) and Claude 4.5 (0.62). All experiments are conducted within our VAGEN framework, a scalable system for training and analyzing multi-turn VLM agents in diverse visual environments. Code and data are publicly available at https://vagen-ai.github.io.

视觉推理世界模型多轮智能体强化学习

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