arXiv:2602.19980cs.LG2026-02被引 3

非自回归模型利用逆向解码优势,高效解决多步规划任务

Discrete Diffusion Models Exploit Asymmetry to Solve Lookahead Planning Tasks

  • 用反向生成避开复杂路径搜索,利用未来信息逆推答案
  • 在相同任务上,NAR模型准确率达100%但仅需指数级更少训练数据
  • 适合追求高效推理、低资源训练的规划类应用开发者

尽管自回归(AR)Transformer生成模型常用于前瞻任务,但近期研究指出其在需要多步前瞻的规划任务中可能存在能力差异。本文对比了在前瞻任务上训练自回归(AR)与非自回归(NAR)模型(如离散扩散模型,dLLMs)时产生的不同涌现机制。通过要求模型前瞻以达成正确结论,我们发现规划问题存在关键不对称性:前向生成需在分支点进行复杂前瞻,而反向生成往往确定性更强。这一特性为NAR模型创造了机会。通过分析训练与推理动态,我们证明NAR模型通过利用未来标记反向解码,无需学习复杂的遍历机制即可完成任务。因此,两种模型均在该任务上达到100%准确率,但NAR模型所需训练样本呈指数级减少,且架构更浅;而AR模型通常需特定课程设计才可收敛。

原文摘要 · Abstract (English)

While Autoregressive (AR) Transformer-based Generative Language Models are frequently employed for lookahead tasks, recent research suggests a potential discrepancy in their ability to perform planning tasks that require multi-step lookahead. In this work, we investigate the distinct emergent mechanisms that arise when training AR versus Non-Autoregressive (NAR) models, such as Discrete Diffusion Models (dLLMs), on lookahead tasks. By requiring the models to plan ahead to reach the correct conclusion, we analyze how these two paradigms fundamentally differ in their approach to the problem. We identify a critical asymmetry in planning problems: while forward generation requires complex lookahead at branching junctions, reverse generation is often deterministic. This asymmetry creates an opportunity for NAR models. Through mechanistic analysis of training and inference dynamics, we demonstrate that NAR models learn to solve planning tasks by utilizing future tokens to decode backwards, avoiding the need to learn complex traversal mechanisms entirely. Consequently, we report that both AR and NAR models are able to achieve perfect accuracy on the lookahead task. However, NAR models require exponentially fewer training examples and shallower architectures compared to AR models, which often fail to converge without specific curriculum adjustments.

规划任务扩散模型非自回归

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