arXiv:2605.12289cs.LGcs.AI2026-05被引 1

用大模型先验提升强化学习决策,解决知识与环境动态不匹配问题

PriorZero: Bridging Language Priors and World Models for Decision Making

论文配图:PriorZero: Bridging Language Priors and World Models for Decision Making
图 1 · 摘自论文原文
  • 在MCTS根节点注入大模型语义先验,聚焦有意义动作
  • 世界模型与大模型分离训练,提升探索效率和最终性能
  • 适合需要长程规划的复杂任务,如文字冒险游戏和指令跟随

利用大型语言模型(LLMs)丰富的世界知识来增强强化学习(RL)智能体,是迈向通用智能的有前景路径。然而,现有方法受限于先验-动态不匹配:静态的LLM知识无法适应长期任务中的复杂转移动态。将LLM先验作为固定策略会限制探索多样性,而端到端微调则面临优化不稳定和信用分配难题。为此,我们提出PriorZero,一种统一框架,通过解耦的滚动-训练设计,将LLM衍生的概念先验融入基于世界模型的规划中。在滚动阶段,新颖的根先验注入机制仅在蒙特卡洛树搜索(MCTS)根节点引入LLM先验,聚焦语义上有望的动作,同时保留世界模型的深度前瞻能力。在训练阶段,PriorZero将世界模型学习与LLM适配解耦:世界模型持续基于交互数据优化,以联合改进其动态、策略和价值预测;其价值估计随后用于提供细粒度信用信号,通过交替优化实现稳定的大模型微调。在多种基准测试中,包括Jericho中的文本冒险游戏和BabyAI中的指令跟随网格世界任务,实验表明PriorZero始终提升探索效率和最终性能,为大模型赋能的决策提供了可行框架。代码已开源。

原文摘要 · Abstract (English)

Leveraging the rich world knowledge of Large Language Models (LLMs) to enhance Reinforcement Learning (RL) agents offers a promising path toward general intelligence. However, a fundamental prior-dynamics mismatch hinders existing approaches: static LLM knowledge cannot directly adapt to the complex transition dynamics of long-horizon tasks. Using LLM priors as fixed policies limits exploration diversity, as the prior is blind to environment-specific dynamics; while end-to-end fine-tuning suffers from optimization instability and credit assignment issues. To bridge this gap, we propose PriorZero, a unified framework that integrates LLM-derived conceptual priors into world-model-based planning through a decoupled rollout-training design. During rollout, a novel root-prior injection mechanism incorporates LLM priors exclusively at the root node of Monte Carlo Tree Search (MCTS), focusing search on semantically promising actions while preserving the world model's deep lookahead capability. During training, PriorZero decouples world-model learning from LLM adaptation: the world model is continuously refined on interaction data to jointly improve its dynamics, policy, and value predictions, its value estimates are then leveraged to provide fine-grained credit assignment signals for stable LLM fine-tuning via alternating optimization. Experiments across diverse benchmarks, including text-based adventure games in Jericho and instruction-following gridworld tasks in BabyAI, demonstrate that PriorZero consistently improves both exploration efficiency and asymptotic performance, establishing a promising framework for LLM-empowered decision-making. Our code is available at https://github.com/opendilab/LightZero.

强化学习大模型决策规划

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