提出一种新型连续扩散模型,让语言建模更高效。
Continuous Diffusion Model for Language Modeling
- 基于统计流形几何设计连续扩散过程
- 在多个语言任务上接近自回归模型表现
- 无需模拟训练,适合高维数据建模
扩散模型作为离散类别数据建模的有前景替代方案,但直接作用于离散空间的模型在状态转移中会丢失信号,难以发挥迭代优化优势。现有连续扩散模型在离散数据上表现逊于离散方法,且与离散方法缺乏清晰关联。本文提出一种面向语言建模的连续扩散模型,融合底层类别分布的几何结构,建立离散扩散与统计流形上的连续流之间的联系,并基于此提出一个广义化的简单扩散过程。进一步提出无模拟训练框架,结合径向对称性设计与高维流形处理技术。在语言建模基准及其他模态上进行全面实验,结果表明该方法优于现有离散扩散模型,性能逼近自回归模型。代码已公开于 https://github.com/harryjo97/RDLM。
原文摘要 · Abstract (English)
Diffusion models have emerged as a promising alternative to autoregressive models in modeling discrete categorical data. However, diffusion models that directly work on discrete data space fail to fully exploit the power of iterative refinement, as the signals are lost during transitions between discrete states. Existing continuous diffusion models for discrete data underperform compared to discrete methods, and the lack of a clear connection between the two approaches hinders the development of effective diffusion models for discrete data. In this work, we propose a continuous diffusion model for language modeling that incorporates the geometry of the underlying categorical distribution. We establish a connection between the discrete diffusion and continuous flow on the statistical manifold, and building on this analogy, introduce a simple diffusion process that generalizes existing discrete diffusion models. We further propose a simulation-free training framework based on radial symmetry, along with a simple technique to address the high dimensionality of the manifold. Comprehensive experiments on language modeling benchmarks and other modalities show that our method outperforms existing discrete diffusion models and approaches the performance of autoregressive models. The code is available at https://github.com/harryjo97/RDLM.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。