让AI agent像人一样思考、行动并模拟世界,提升长期任务表现。
Dyna-Think: Synergizing Reasoning, Acting, and World Model Simulation in AI Agents
- 融合内部世界模型的思维框架,通过模拟预测行动后果。
- 性能接近R1但平均用词量减半,支持域内与域外任务泛化。
- 强调世界建模能力对智能体表现的关键作用,适合长程任务研究者。
近期基于大语言模型(如DeepSeek-R1)的推理进展展示了在数学与编程等领域的出色能力,体现验证、目标分解与自我反思等认知行为。然而,长周期智能体任务中哪些行为有效、哪些缺失仍不明确。本文提出Dyna-Think框架,将规划、内部世界模型模拟、推理与行动协同整合,以增强智能体性能。为此,我们设计了两种训练方法:DIT(Dyna-Think Imitation Learning)通过重构R1的思维过程,聚焦于与计划动作相关的世界模型模拟,并用此数据训练策略;DDT(Dyna-Think Dyna Training)采用两阶段训练:先通过状态预测或批判生成等目标提升世界建模能力,再优化行动策略。在OSWorld和WindowsAgentArena上的评估表明,Dyna-Think显著提升智能体在域内与域外任务的表现,达到与R1相当的最佳样本性能,但平均生成令牌数减少2倍。实证研究揭示:1)使用批判生成训练世界模型能有效提升策略表现;2)表现更优的智能体具备更强的世界建模能力。结果表明,将世界模型模拟融入智能体是提升其推理、规划与执行能力的有前景方向。
原文摘要 · Abstract (English)
Recent progress in reasoning with large language models (LLMs), such as DeepSeek-R1, demonstrates impressive capabilities in domains like mathematics and coding, by exhibiting complex cognitive behaviors such as verification, goal decomposition, and self-reflection. However, it is unclear what behavior is effective and what behavior is missing for long-horizon AI agents tasks. In this work, we propose Dyna-Think, a thinking framework that integrates planning with an internal world model with reasoning and acting to enhance AI agent performance. To enable Dyna-Think, we propose Dyna-Think Imitation Learning (DIT) and Dyna-Think Dyna Training (DDT). To initialize a policy with Dyna-Think, DIT reconstructs the thinking process of R1 to focus on performing world model simulation relevant to the proposed (and planned) action, and trains the policy using this reconstructed data. To enhance Dyna-Think, DDT uses a two-stage training process to first improve the agent's world modeling ability via objectives such as state prediction or critique generation, and then improve the agent's action via policy training. We evaluate our methods on OSWorld and WindowsAgentArena, and demonstrate that Dyna-Think improves the agent's in-domain and out-of-domain performance, achieving similar best-of-n performance compared to R1 while generating 2x less tokens on average. Our extensive empirical studies reveal that 1) using critique generation for world model training is effective to improve policy performance; and 2) AI agents with better performance correlate with better world modeling abilities. We believe our results suggest a promising research direction to integrate world model simulation into AI agents to enhance their reasoning, planning, and acting capabilities.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。