arXiv:2601.16971cs.LG2026-01被引 4

提出ARMD模型,让扩散模型兼具自回归效率与并行生成能力。

Auto-Regressive Masked Diffusion Models

  • 将掩码扩散重构为分块因果模型,实现单次前向传播计算所有条件概率。
  • 在标准语言建模任务上超越现有扩散基模型,训练步数显著减少。
  • 支持分块并行生成,适合追求高效推理的文本生成场景。

掩码扩散模型(MDMs)在语言建模中展现出潜力,但性能仍落后于自回归模型(ARMs),且需更多训练迭代。本文提出自回归掩码扩散(ARMD)模型,通过将掩码扩散过程重构为分块因果模型,设计出严格因果、排列等变的架构,可在一次前向传播中完成多步去噪的所有条件概率计算。该架构支持自回归式解码与渐进排列训练,使模型能学习从左到右及随机标记顺序。进一步提出一种新型分块并行生成策略,在保持全局一致性的同时加速推理。实验表明,ARMD在标准语言建模基准上达到当前最优性能,超越现有扩散基模型,且训练步数大幅减少,有效弥合了并行与串行解码间的性能差距。

原文摘要 · Abstract (English)

Masked diffusion models (MDMs) have emerged as a promising approach for language modeling, yet they face a performance gap compared to autoregressive models (ARMs) and require more training iterations. In this work, we present the Auto-Regressive Masked Diffusion (ARMD) model, an architecture designed to close this gap by unifying the training efficiency of autoregressive models with the parallel generation capabilities of diffusion-based models. Our key insight is to reframe the masked diffusion process as a block-wise causal model. This perspective allows us to design a strictly causal, permutation-equivariant architecture that computes all conditional probabilities across multiple denoising steps in a single, parallel forward pass. The resulting architecture supports efficient, autoregressive-style decoding and a progressive permutation training scheme, allowing the model to learn both canonical left-to-right and random token orderings. Leveraging this flexibility, we introduce a novel strided parallel generation strategy that accelerates inference by generating tokens in parallel streams while maintaining global coherence. Empirical results demonstrate that ARMD achieves state-of-the-art performance on standard language modeling benchmarks, outperforming established diffusion baselines while requiring significantly fewer training steps. Furthermore, it establishes a new benchmark for parallel text generation, effectively bridging the performance gap between parallel and sequential decoding.

扩散模型语言建模并行生成自回归

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