arXiv:2601.08955cs.CLcs.AI2026-01被引 4

让智能体通过自适应预演规划,提升复杂任务决策能力。

Imagine-then-Plan: Agent Learning from Adaptive Lookahead with World Models

  • 用可变预测时长的预演机制生成多步想象轨迹
  • 在多个基准上性能显著优于现有方法
  • 适合需要长期推理的复杂任务研究者

世界模型能建模环境状态的未来动态,使智能体无需真实交互即可推理与行动。当前方法多采用单步或固定时长的模拟,未能充分挖掘复杂任务规划潜力。本文提出Imagine-then-Plan(ITP)框架,通过智能体策略与世界模型交互生成多步‘想象’轨迹。由于不同任务和阶段所需的预演长度不同,我们引入一种自适应前瞻机制,在最终目标与任务进展间权衡。生成的想象轨迹提供丰富的未来信息,如任务进展与潜在冲突,融合当前观测后形成部分可观测且可想象的马尔可夫决策过程,指导策略学习。我们实现了无需训练和强化学习训练两种版本的ITP。在多个典型智能体基准上的实验表明,ITP显著优于对比基线。进一步分析验证了自适应前瞻极大增强了智能体的推理能力,为解决更广泛复杂任务提供了洞见。代码与数据将公开于https://github.com/loyiv/ITP。

原文摘要 · Abstract (English)

Recent advances in world models have shown promise for modeling future dynamics of environmental states, enabling agents to reason and act without accessing real environments. Current methods mainly perform single-step or fixed-horizon rollouts, leaving their potential for complex task planning under-exploited. We propose Imagine-then-Plan (\texttt{ITP}), a unified framework for agent learning via lookahead imagination, where an agent's policy model interacts with the learned world model, yielding multi-step ``imagined'' trajectories. Since the imagination horizon may vary by tasks and stages, we introduce a novel adaptive lookahead mechanism by trading off the ultimate goal and task progress. The resulting imagined trajectories provide rich signals about future consequences, such as achieved progress and potential conflicts, which are fused with current observations, formulating a partially \textit{observable} and \textit{imaginable} Markov decision process to guide policy learning. We instantiate \texttt{ITP} with both training-free and reinforcement-trained variants. Extensive experiments across representative agent benchmarks demonstrate that \texttt{ITP} significantly outperforms competitive baselines. Further analyses validate that our adaptive lookahead largely enhances agents' reasoning capability, providing valuable insights into addressing broader, complex tasks. Our code and data will be publicly available at https://github.com/loyiv/ITP.

智能体学习世界模型规划推理

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