让大模型学会模拟动作后果,提升智能体的环境适应能力。
Reinforcement World Model Learning for LLM-based Agents
- 用自监督方式训练大模型预测动作后的状态变化。
- 在两个基准上显著优于基础模型,接近专家数据训练效果。
- 适合想提升大模型决策能力的研究者和开发者。
大语言模型在语言任务中表现强劲,但在智能体场景下常无法预判动作后果并适应环境变化,亟需具备世界建模能力。我们提出强化世界模型学习(RWML),一种基于模拟与真实状态对齐的自监督方法,在文本状态空间中学习动作条件的世界模型。该方法通过模拟下一状态与真实观测状态的一致性来优化模型,避免了仅追求词语精确复现导致的模型坍塌问题。相比基于大模型评分的奖励机制,本方法更抗奖励欺骗。在ALFWorld和τ² Bench上的实验表明,即使完全自监督,性能也显著超越基线;结合任务成功奖励后,在两个基准上分别领先直接强化学习6.9和5.7分,且达到专家数据训练水平。
原文摘要 · Abstract (English)
Large language models (LLMs) have achieved strong performance in language-centric tasks. However, in agentic settings, LLMs often struggle to anticipate action consequences and adapt to environment dynamics, highlighting the need for world-modeling capabilities in LLM-based agents. We propose Reinforcement World Model Learning (RWML), a self-supervised method that learns action-conditioned world models for LLM-based agents on textual states using sim-to-real gap rewards. Our method aligns simulated next states produced by the model with realized next states observed from the environment, encouraging consistency between internal world simulations and actual environment dynamics in a pre-trained embedding space. Unlike next-state token prediction, which prioritizes token-level fidelity (i.e., reproducing exact wording) over semantic equivalence and can lead to model collapse, our method provides a more robust training signal and is empirically less susceptible to reward hacking than LLM-as-a-judge. We evaluate our method on ALFWorld and $τ^2$ Bench and observe significant gains over the base model, despite being entirely self-supervised. When combined with task-success rewards, our method outperforms direct task-success reward RL by 6.9 and 5.7 points on ALFWorld and $τ^2$ Bench respectively, while matching the performance of expert-data training.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。