用语言模型先验从少量观测中快速构建环境模型。
Learning POMDP World Models from Observations with Language-Model Priors

- 用大模型生成候选环境模型并迭代优化。
- 仅用少量交互数据达到顶尖方法的性能与效率。
- 适合追求低样本学习的智能体研发者。
无论导航建筑、操作机器人还是玩游戏,智能体要有效行动,必须先学习环境内部模型。部分可观测马尔可夫决策过程(POMDP)为这类内部世界模型提供了灵活建模方式,但仅从观测-动作轨迹中学习仍具挑战性,通常需大量环境交互。本文探讨能否通过语言模型先验减少高成本交互,并提出 extit{Pinductor}(POMDP-inductor):大语言模型从少量观测-动作轨迹中生成候选POMDP模型,并通过信念似然得分迭代优化。尽管信息量更少, extit{Pinductor} 在性能和样本效率上均匹配依赖隐藏状态特权访问的基于LLM的POMDP学习方法,显著超越传统表格型POMDP基线。进一步实验显示,性能随语言模型能力提升而增长,且在语义信息缺失时表现平稳下降。这些结果表明,语言模型先验是部分可观测环境下高效世界模型学习的实用工具,也是迈向真实世界通用智能体的重要一步。代码已开源:https://github.com/atomresearch/pinductor。
原文摘要 · Abstract (English)
Whether navigating a building, operating a robot, or playing a game, an agent that acts effectively in an environment must first learn an internal model of how that environment works. Partially-observable Markov decision processes (POMDPs) provide a flexible modeling class for such internal world models, but learning them from observation-action trajectories alone is challenging and typically requires extensive environment interaction. We ask whether language-model priors can reduce costly interaction by leveraging prior knowledge, and introduce \emph{Pinductor} (POMDP-inductor): an LLM proposes candidate POMDP models from a few observation-action trajectories and iteratively refines them to optimize a belief-based likelihood score. Despite using strictly less information, \emph{Pinductor} matches the performance and sample efficiency of LLM-based POMDP learning methods that assume privileged access to the hidden state, while significantly surpassing the sample efficiency of tabular POMDP baselines. Further results show that performance scales with LLM capability and degrades gracefully as semantic information about the environment is withheld. Together, these results position language-model priors as a practical tool for sample-efficient world-model learning under partial observability, and a step toward generalist agents in real-world environments. Code is available at https://github.com/atomresearch/pinductor.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。