arXiv:2409.02908cs.LGcs.AI2024-09ICLR被引 208

发现掩码扩散模型其实本质是掩码模型,采样速度提升20倍且存在采样偏差。

Masked Diffusion Models are Secretly Time-Agnostic Masked Models and Exploit Inaccurate Categorical Sampling

  • 将掩码扩散模型重新解读为无需时间变量的掩码模型。
  • 提出首击采样器,实现20倍加速并解决分类采样不准确问题。
  • 揭示现有评估方法可能高估其性能,适合关注生成质量的读者。

掩码扩散模型(MDMs)因其在离散数据生成任务中表现优于其他离散扩散模型,正成为语言建模的重要候选方案。然而本文揭示,MDMs的训练与采样过程在理论上完全独立于时间变量,本质上等同于掩码模型。我们提出的首击采样器(FHS)在理论上与原生成流程等价,显著缓解了耗时的分类采样问题,实现20倍加速。此外,研究首次指出即使使用32位浮点精度,也存在数值误差导致分类采样不准确,理论与实证均表明这会降低有效温度,减少词元多样性。因此,仅依赖不完整生成困惑度评估的先前研究可能对MDMs存在偏颇,影响公平比较。

原文摘要 · Abstract (English)

Masked diffusion models (MDMs) have emerged as a popular research topic for generative modeling of discrete data, thanks to their superior performance over other discrete diffusion models, and are rivaling the auto-regressive models (ARMs) for language modeling tasks. The recent effort in simplifying the masked diffusion framework further leads to alignment with continuous-space diffusion models and more principled training and sampling recipes. In this paper, however, we reveal that both training and sampling of MDMs are theoretically free from the time variable, arguably the key signature of diffusion models, and are instead equivalent to masked models. The connection on the sampling aspect is drawn by our proposed first-hitting sampler (FHS). Specifically, we show that the FHS is theoretically equivalent to MDMs' original generation process while significantly alleviating the time-consuming categorical sampling and achieving a 20$\times$ speedup. In addition, our investigation raises doubts about whether MDMs can truly beat ARMs in text generation. We identify, for the first time, an underlying numerical issue, even with the commonly used 32-bit floating-point precision, which results in inaccurate categorical sampling. We show that it lowers the effective temperature both theoretically and empirically, and the resulting decrease in token diversity makes previous evaluations, which assess the generation quality solely through the incomplete generative perplexity metric, somewhat unfair.

扩散模型文本生成采样优化数值误差

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