arXiv:2603.00045cs.LGcs.AI2026-03被引 7

突破扩散语言模型的并行生成瓶颈,实现高效高质量文本生成

Breaking the Factorization Barrier in Diffusion Language Models

  • 用轻量级推理层替代全因子化输出,建模复杂联合依赖
  • 在少步生成下避免性能崩溃,延迟显著降低
  • 适配多种模型架构,训练成本仅为强化学习基线的几分之一

扩散语言模型理论上支持高效并行生成,但实际受限于‘因子化障碍’:即假设同时预测的词元相互独立。这一限制导致权衡:要么牺牲速度逐个处理依赖,要么因因子化导致语义不连贯。我们指出该障碍并非源于骨干模型表达能力不足,而是结构设计缺陷——显式参数化联合分布会导致Transformer输出参数量爆炸。为此提出耦合离散扩散(CoDD),用轻量级、可计算的概率推理层取代全因子化输出分布。该框架生成的分布家族远比标准因子化先验更具表达力,能有效建模复杂联合依赖,同时保持参数规模可控。实验表明,CoDD可无缝提升多种扩散语言模型架构,几乎无额外开销,推理性能媲美计算密集型强化学习基线,训练成本仅为后者的几分之一。此外,它有效防止少步生成中的性能退化,实现低延迟下的高质量输出。代码已公开:https://github.com/liuanji/CoDD。

原文摘要 · Abstract (English)

Diffusion language models theoretically allow for efficient parallel generation but are practically hindered by the ``factorization barrier'': the assumption that simultaneously predicted tokens are independent. This limitation forces a trade-off: models must either sacrifice speed by resolving dependencies sequentially or suffer from incoherence due to factorization. We argue that this barrier arises not from limited backbone expressivity, but from a structural misspecification: models are restricted to fully factorized outputs because explicitly parameterizing a joint distribution would require the Transformer to output a prohibitively large number of parameters. We propose Coupled Discrete Diffusion (CoDD), a hybrid framework that breaks this barrier by replacing the fully-factorized output distribution with a lightweight, tractable probabilistic inference layer. This formulation yields a distribution family that is significantly more expressive than standard factorized priors, enabling the modeling of complex joint dependencies, yet remains compact enough to avoid the prohibitive parameter explosion associated with full joint modeling. Empirically, CoDD seamlessly enhances diverse diffusion language model architectures with negligible overhead, matching the reasoning performance of computationally intensive Reinforcement Learning baselines at a fraction of the training cost. Furthermore, it prevents performance collapse in few-step generation, enabling high-quality outputs at significantly reduced latencies. Code available at: https://github.com/liuanji/CoDD.

扩散模型文本生成并行推理

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