让大模型学会因果推理,提升长期规划能力
Language Agents Meet Causality -- Bridging LLMs and Causal World Models
- 用因果表示学习构建可解释的世界模型,连接语言与因果变量
- 在复杂环境和长周期规划中,性能显著优于纯大模型方法
- 适合需要可靠推理的AI系统开发者或研究者参考
大型语言模型(LLMs)在规划与推理任务中展现出巨大潜力,但这些任务需要对环境具备因果理解。尽管LLMs能从预训练数据中获取常识性因果知识,但常不完整、错误或不适用于特定环境。相比之下,因果表示学习(CRL)专注于挖掘给定环境中的潜在因果结构。本文提出一种框架,将CRL与LLMs结合,实现因果感知的推理与规划。该框架学习一个因果世界模型,将因果变量与自然语言表达关联,为LLM提供灵活的文本接口以描述和生成动作与状态。本质上,因果世界模型充当可查询的模拟器,供LLM交互使用。我们在不同时间尺度和环境复杂度下评估该框架在因果推断与规划任务上的表现。实验表明,该方法在长规划周期中明显优于基于LLM的推理器。
原文摘要 · Abstract (English)
Large Language Models (LLMs) have recently shown great promise in planning and reasoning applications. These tasks demand robust systems, which arguably require a causal understanding of the environment. While LLMs can acquire and reflect common sense causal knowledge from their pretraining data, this information is often incomplete, incorrect, or inapplicable to a specific environment. In contrast, causal representation learning (CRL) focuses on identifying the underlying causal structure within a given environment. We propose a framework that integrates CRLs with LLMs to enable causally-aware reasoning and planning. This framework learns a causal world model, with causal variables linked to natural language expressions. This mapping provides LLMs with a flexible interface to process and generate descriptions of actions and states in text form. Effectively, the causal world model acts as a simulator that the LLM can query and interact with. We evaluate the framework on causal inference and planning tasks across temporal scales and environmental complexities. Our experiments demonstrate the effectiveness of the approach, with the causally-aware method outperforming LLM-based reasoners, especially for longer planning horizons.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。