让AI智能体学会在行动前模拟未来,提升复杂任务表现
Dyna-Mind: Learning to Simulate from Experience for Better AI Agents
- 用真实交互数据构建搜索树,训练模型生成带模拟的推理过程
- 在合成与真实环境中显著提升长周期任务成功率,如解谜和安卓操作
- 适合研究长期规划、具身智能与人类认知启发的AI系统
推理模型在数学和编程领域取得了显著进展,但在长周期、交互式任务(如网页导航、手机/电脑操作)中表现不佳。受人类认知研究启发,我们提出当前AI代理需要‘虚拟试错’能力——即在行动前进行未来状态的内在模拟。为此,我们提出两阶段训练框架Dyna-Mind:第一阶段引入ReSim,通过环境交互收集的真实经验构建扩展搜索树,训练模型生成结构化推理轨迹,使其推理基于真实的动态世界;第二阶段提出Dyna-GRPO,一种在线强化学习方法,利用真实回放中的结果奖励与中间状态反馈,进一步优化模拟与决策能力。在两个合成基准(Sokoban、ALFWorld)和一个真实基准(AndroidWorld)上的实验表明:(1) ReSim有效赋予模型模拟能力;(2) Dyna-GRPO通过结果与过程双重信号,学习到更优的长周期规划策略。结果凸显模拟在提升AI代理推理、规划与执行能力中的核心作用。
原文摘要 · Abstract (English)
Reasoning models have recently shown remarkable progress in domains such as math and coding. However, their expert-level abilities in math and coding contrast sharply with their performance in long-horizon, interactive tasks such as web navigation and computer/phone-use. Inspired by literature on human cognition, we argue that current AI agents need ''vicarious trial and error'' - the capacity to mentally simulate alternative futures before acting - in order to enhance their understanding and performance in complex interactive environments. We introduce Dyna-Mind, a two-stage training framework that explicitly teaches (V)LM agents to integrate such simulation into their reasoning. In stage 1, we introduce Reasoning with Simulations (ReSim), which trains the agent to generate structured reasoning traces from expanded search trees built from real experience gathered through environment interactions. ReSim thus grounds the agent's reasoning in faithful world dynamics and equips it with the ability to anticipate future states in its reasoning. In stage 2, we propose Dyna-GRPO, an online reinforcement learning method to further strengthen the agent's simulation and decision-making ability by using both outcome rewards and intermediate states as feedback from real rollouts. Experiments on two synthetic benchmarks (Sokoban and ALFWorld) and one realistic benchmark (AndroidWorld) demonstrate that (1) ReSim effectively infuses simulation ability into AI agents, and (2) Dyna-GRPO leverages outcome and interaction-level signals to learn better policies for long-horizon, planning-intensive tasks. Together, these results highlight the central role of simulation in enabling AI agents to reason, plan, and act more effectively in the ever more challenging environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。