让语言模型的世界观和行动策略一起进化,提升长期决策能力
COMAP: Co-Evolving World Models and Agent Policies for LLM Agents

- 通过闭环交互同步优化世界模型与策略
- 在多个任务上比基线提升16.75%(Qwen3-4B)
- 适合需要长期规划的智能体应用
赋予语言智能体世界模型可使其预判环境变化并评估动作可行性。然而现有文本世界模型训练后固定不变,难以适应智能体演进带来的状态-动作分布变化。同时,现有策略改进方法依赖外部奖励或验证器,在真实交互环境中适用性受限。本文提出COMAP框架,通过闭环交互协同演化文本世界模型与智能体策略。每个决策步骤中,世界模型预测候选动作的未来状态反馈,智能体基于反馈可靠性进行前瞻性反思并优化动作。生成的在线轨迹用于世界模型自蒸馏更新,使其更匹配智能体的动态交互分布。在具身任务规划、网络导航和工具使用基准测试中,COMAP持续优于基线,如在Qwen3-4B上实现16.75%的相对提升。分析表明,协同进化循环随时间提高世界模型预测精度,促进更有效的长程决策。代码已开源:https://github.com/loyiv/CoMAP。
原文摘要 · Abstract (English)
Equipping language agents with world models enables them to anticipate environment dynamics and evaluate candidate actions before execution. However, existing textual world models are typically fixed after training, preventing them from adapting to the on-policy state-action distributions induced by an evolving agent. Meanwhile, agent-improvement methods often rely on external rewards or verifiers, limiting their applicability in realistic interactive environments. In this paper, we propose COMAP, a novel framework that co-evolves textual world models and agent policies through closed-loop interaction. At each decision step, the world model predicts future state feedback for candidate actions, and the agent performs future-aware reflection by estimating the reliability of this feedback and refining its action accordingly. The resulting on-policy trajectories are then used to update the world model via self-distillation, allowing it to better match the agent's evolving interaction distribution. Across embodied task planning, Web navigation, and tool-use benchmarks, COMAP consistently outperforms competitive baselines, e.g., +16.75% relative improvement with Qwen3-4B. Further analyses show that the co-evolutionary loop improves the world model's prediction accuracy over time and leads to more effective long-horizon decision-making. Our code is available at: https://github.com/loyiv/CoMAP.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。