arXiv:2410.21357cs.CLcs.LG2024-10ICLR被引 98

用能量模型提升文本扩散生成,速度更快且更准。

Energy-Based Diffusion Language Models for Text Generation

  • 引入序列级能量模型改进扩散过程的近似精度。
  • 在多个基准上接近自回归模型的困惑度,采样速度提升1.3倍。
  • 适合追求高效文本生成或研究扩散模型的新手。

尽管自回归语言模型取得显著进展,但超越从左到右生成范式的替代生成范式仍处于活跃探索中。离散扩散模型因其并行生成能力成为有前景的替代方案。然而,这些模型性能仍落后于自回归模型,尤其在减少采样步数时差距扩大。我们的分析表明,这种退化源于扩散模型中不完善的近似。本文提出能量基扩散语言模型(EDLM),一种在每一步扩散中作用于完整序列的能量基模型,以改进扩散模型的底层近似。具体而言,我们采用残差形式的能量模型,并通过预训练自回归模型或利用噪声对比估计微调双向Transformer来获取其参数。我们还提出一种高效的并行重要性采样生成算法。在语言建模基准上的全面实验表明,我们的模型能显著优于现有最先进扩散模型,逼近自回归模型的困惑度。进一步显示,在不损失生成性能的前提下,框架相比现有扩散模型实现1.3倍的采样加速。代码已开源:https://github.com/MinkaiXu/Energy-Diffusion-LLM。

原文摘要 · Abstract (English)

Despite remarkable progress in autoregressive language models, alternative generative paradigms beyond left-to-right generation are still being actively explored. Discrete diffusion models, with the capacity for parallel generation, have recently emerged as a promising alternative. Unfortunately, these models still underperform the autoregressive counterparts, with the performance gap increasing when reducing the number of sampling steps. Our analysis reveals that this degradation is a consequence of an imperfect approximation used by diffusion models. In this work, we propose Energy-based Diffusion Language Model (EDLM), an energy-based model operating at the full sequence level for each diffusion step, introduced to improve the underlying approximation used by diffusion models. More specifically, we introduce an EBM in a residual form, and show that its parameters can be obtained by leveraging a pretrained autoregressive model or by finetuning a bidirectional transformer via noise contrastive estimation. We also propose an efficient generation algorithm via parallel important sampling. Comprehensive experiments on language modeling benchmarks show that our model can consistently outperform state-of-the-art diffusion models by a significant margin, and approaches autoregressive models' perplexity. We further show that, without any generation performance drop, our framework offers a 1.3$\times$ sampling speedup over existing diffusion models. Reproduced code is available at https://github.com/MinkaiXu/Energy-Diffusion-LLM.

扩散模型文本生成能量模型高效采样

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