用符号规划任务验证语言模型能否真正支持推理规划。
From Next Token Prediction to (STRIPS) World Models
- 设计符号对齐的STRIPS Transformer与普通Transformer对比学习。
- 普通Transformer加特定注意力机制可完美训练并泛化到长序列。
- 即使无显式符号结构,模型也能提取出可用的符号规划模型。
我们研究了仅通过下一个词预测能否生成真正支持规划的世界模型,采用可控的符号设置:仅从动作轨迹中学习命题型STRIPS动作模型,并可精确评估正确性。提出两种架构:一是基于理论证明与STRIPS语言结构对齐的STRIPS Transformer;二是未显式引入符号结构的标准Transformer,考察不同位置编码和注意力聚合机制。在五个经典规划领域上评估,测量训练准确率、跨领域与问题规模的泛化能力及规划性能。有趣的是,两种方法均可用于生成可被现成STRIPS规划器使用的模型,覆盖指数级未见的初始状态与目标。尽管STRIPS Transformer具有强符号归纳偏置,但优化困难,需更大数据集才能可靠泛化;而标准Transformer结合棒断裂注意力机制,在训练上达到近完美准确率且泛化能力强。相反,无棒断裂注意力的标准Transformer无法泛化至长轨迹,而从短轨迹训练的Transformer中提取出的符号化STRIPS模型却具备该能力。
原文摘要 · Abstract (English)
We study whether next-token prediction can yield world models that truly support planning, in a controlled symbolic setting where propositional STRIPS action models are learned from action traces alone and correctness can be evaluated exactly. We introduce two architectures. The first is the STRIPS Transformer, a symbolically aligned model grounded in theoretical results linking transformers and the formal language structure of STRIPS domains. The second is a standard transformer architecture without explicit symbolic structure built in, for which we study different positional encoding schemes and attention aggregation mechanisms. We evaluate both architectures on five classical planning domains, measuring training accuracy, generalization, and planning performance across domains and problem sizes. Interestingly, both approaches can be used to produce models that support planning with off-the-shelf STRIPS planners over exponentially many unseen initial states and goals. Although the STRIPS Transformer incorporates a strong symbolic inductive bias, it is harder to optimize and requires larger datasets to generalize reliably. In contrast, a standard transformer with stick-breaking attention achieves near-perfect training accuracy and strong generalization. Finally, standard transformers without stick-breaking attention do not generalize to long traces, whereas a symbolic STRIPS model extracted from a transformer trained on shorter traces does.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。