用规则学习让大模型对齐真实环境,提升智能体决策效率。
WALL-E: World Alignment by Rule Learning Improves World Model-based LLM Agents
- 通过无梯度规则学习,让大模型与环境动态对齐。
- 在Minecraft和ALFWorld中成功率提升15%-30%,推理耗时减少20%以上。
- 适合需要高效推理与低成本重规划的自主智能体研究者。
大语言模型能否直接作为基于模型智能体的世界模型?尽管大模型先验知识与具体环境动态之间存在差距,但我们的研究表明,通过将大模型与部署环境对齐,可有效弥合这一差距,且这种“世界对齐”可通过在大模型上进行规则学习高效实现。得益于大模型丰富的先验知识,仅需少量额外规则即可使大模型预测与环境动态一致。为此,我们提出一种神经符号方法,通过对比智能体探索轨迹与世界模型预测,无梯度地诱导、更新和剪枝规则。最终构建的世界模型由大模型与学习到的规则组成。基于此,我们构建了名为WALL-E的具身大模型智能体,采用模型预测控制(MPC)优化前瞻动作。相比现有方法,WALL-E的推理仅需少数关键规则,无需输入冗长轨迹。在Minecraft和ALFWorld的开放世界挑战中,其成功率显著超越基线:Minecraft中成功率提升15%-30%,重规划轮次减少8-20次,推理令牌消耗仅为60%-80%;ALFWorld中成功率达95%仅需6次迭代,创历史新高。
原文摘要 · Abstract (English)
Can large language models (LLMs) directly serve as powerful world models for model-based agents? While the gaps between the prior knowledge of LLMs and the specified environment's dynamics do exist, our study reveals that the gaps can be bridged by aligning an LLM with its deployed environment and such "world alignment" can be efficiently achieved by rule learning on LLMs. Given the rich prior knowledge of LLMs, only a few additional rules suffice to align LLM predictions with the specified environment dynamics. To this end, we propose a neurosymbolic approach to learn these rules gradient-free through LLMs, by inducing, updating, and pruning rules based on comparisons of agent-explored trajectories and world model predictions. The resulting world model is composed of the LLM and the learned rules. Our embodied LLM agent "WALL-E" is built upon model-predictive control (MPC). By optimizing look-ahead actions based on the precise world model, MPC significantly improves exploration and learning efficiency. Compared to existing LLM agents, WALL-E's reasoning only requires a few principal rules rather than verbose buffered trajectories being included in the LLM input. On open-world challenges in Minecraft and ALFWorld, WALL-E achieves higher success rates than existing methods, with lower costs on replanning time and the number of tokens used for reasoning. In Minecraft, WALL-E exceeds baselines by 15-30% in success rate while costing 8-20 fewer replanning rounds and only 60-80% of tokens. In ALFWorld, its success rate surges to a new record high of 95% only after 6 iterations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。