arXiv:2506.06725cs.AIcs.LG2025-06被引 9

让大模型主动探索并自我修正理论,提升对具体环境的预测能力。

WorldLLM: Improving LLMs' world modeling using curiosity-driven theory-making

  • 用自然语言假设引导模型预测,通过贝叶斯框架迭代优化。
  • 在文本游戏环境中,预测准确率显著提升,且生成可解释的环境理论。
  • 适合需要理解复杂动态系统的研究者或开发人员使用。

大型语言模型(LLMs)具备广泛的世界知识,但在模拟等结构化、领域特定的场景中常难以生成精确预测。这源于其无法将泛化的、非结构化的理解与具体环境相融合。为此,我们提出WorldLLM框架,通过结合贝叶斯推断、自主主动探索与强化学习,增强基于LLM的世界建模能力。该框架利用LLM的上下文学习能力,以自然语言假设作为提示,指导基于LLM的世界模型进行预测。这些假设通过贝叶斯推断框架不断优化,其中第二层LLM作为提议分布,依据收集到的证据进行更新。证据由一种好奇心驱动的强化学习策略获取,该策略探索环境,寻找在当前假设下对数似然较低的转移状态。通过不断交替优化假设与收集新证据,框架实现预测性能的持续提升。实验表明,WorldLLM在要求操作和组合物体的文本游戏环境中有效提升了预测准确性,并生成了人类可理解的环境动态理论。

原文摘要 · Abstract (English)

Large Language Models (LLMs) possess general world knowledge but often struggle to generate precise predictions in structured, domain-specific contexts such as simulations. These limitations arise from their inability to ground their broad, unstructured understanding in specific environments. To address this, we present WorldLLM, a framework that enhances LLM-based world modeling by combining Bayesian inference and autonomous active exploration with reinforcement learning. WorldLLM leverages the in-context learning abilities of LLMs to guide an LLM-based world model's predictions using natural language hypotheses given in its prompt. These hypotheses are iteratively refined through a Bayesian inference framework that leverages a second LLM as the proposal distribution given collected evidence. This evidence is collected using a curiosity-driven reinforcement learning policy that explores the environment to find transitions with a low log-likelihood under our LLM-based predictive model using the current hypotheses. By alternating between refining hypotheses and collecting new evidence, our framework autonomously drives continual improvement of the predictions. Our experiments demonstrate the effectiveness of WorldLLM in a textual game environment that requires agents to manipulate and combine objects. The framework not only enhances predictive accuracy, but also generates human-interpretable theories of environment dynamics.

世界模型强化学习贝叶斯推断可解释性

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