跳到正文
原文
HuggingFace Daily Papers(社区热门论文)·· 2026-04-15AI 评分41

LangFlow:连续扩散语言模型首次匹敌离散扩散

LangFlow: Continuous Diffusion Rivals Discrete in Language Modeling

AI 导读

研究团队推出LangFlow,首个性能比肩离散扩散模型的连续扩散语言模型。该方法通过Bregman散度将嵌入空间与Flow Matching相连,提出基于ODE的NLL评估边界、信息均匀噪声调度及自条件训练三项创新。LangFlow在LM1B和OpenWebText上分别实现30.0和24.6的困惑度(PPL),并在7项零样本迁移基准中的4项上超越自回归基线,首次证实连续扩散在语言建模中的竞争力。

正文

Continuous diffusion has been the foundation of high-fidelity, controllable, and few-step generation of many data modalities such as images. However, in language modeling, prior continuous diffusion language models (DLMs) lag behind discrete counterparts due to the sparse data space and the underexplored design space. In this work, we close this gap with LangFlow, the first continuous DLM to rival discrete diffusion, by connecting embedding-space DLMs to Flow Matching via Bregman divergence, alongside three key innovations: (1) we derive a novel ODE-based NLL bound for principled evaluation of continuous flow-based language models; (2) we propose an information-uniform principle for setting the noise schedule, which motivates a learnable noise scheduler based on a Gumbel distribution; and (3) we revise prior training protocols by incorporating self-conditioning, as we find it improves both likelihood and sample quality of embedding-space DLMs with effects substantially different from discrete diffusion. Putting everything together, LangFlow rivals top discrete DLMs on both the perplexity (PPL) and the generative perplexity (Gen. PPL), reaching a PPL of 30.0 on LM1B and 24.6 on OpenWebText. It even exceeds autoregressive baselines in zero-shot transfer on 4 out of 7 benchmarks. LangFlow provides the first clear evidence that continuous diffusion is a promising paradigm for language modeling. Homepage: https://github.com/nealchen2003/LangFlow

来源:HuggingFace Daily Papers(社区热门论文) · arxiv.org