arXiv:2602.16813cs.CLcs.AI2026-02被引 37

用连续流模型实现一步生成,速度与质量双提升

Flow Map Language Models: One-step Language Modeling via Continuous Denoising

  • 基于连续流映射的非离散去噪,直接学习生成路径
  • 在LM1B和OWT数据集上达到顶尖离散扩散模型水平
  • 一步生成超越8步离散模型,适合高速推理场景

基于离散扩散的语言模型因潜在加速生成能力受到广泛关注。然而,这类模型在少步生成时质量急剧下降,限制了实际提速效果。本文提出基于独热编码嵌入上连续流的语言模型(FLM),其独特流映射结构可直接用于高效少步推断,这是离散方法无法实现的。我们采用符合数据单纯形几何的交叉熵损失训练流及其映射,并比较了三种流映射蒸馏策略。实验表明,FLM在One Billion Words(LM1B)和OpenWebText(OWT)数据集上达到当前最优离散扩散基线水平。进一步将FLM蒸馏为流映射语言模型(FMLM),其单步生成质量超过近期少步离散扩散模型的8步表现。本工作挑战了离散噪声过程对离散模态生成必不可少的普遍认知,为大规模加速语言建模开辟新路径。代码已开源。

原文摘要 · Abstract (English)

Language models based on discrete diffusion have attracted widespread interest for their potential to provide faster generation than autoregressive models. Despite their promise, these models typically produce samples whose quality sharply degrades in the few-step regime, preventing a dramatic speedup in practice. Here, we show that language models based on continuous flows over one-hot token embeddings can outperform discrete diffusion in both quality and speed. Importantly, our continuous formulation defines a unique flow map that can be learned directly for efficient few-step inference, a structure we show is unavailable to discrete methods. In this setting, we show that both the flow and its associated flow map can be learned with simple cross-entropy objectives that respect the simplex geometry of the data, and we identify three distinct choices for flow map distillation whose performance we compare in practice. Using these insights, we build a flow language model (FLM), a continuous flow that matches state-of-the-art discrete diffusion baselines on the One Billion Words (LM1B) and OpenWebText (OWT) datasets. We then distill FLM into a flow map language model (FMLM), whose one-step generation exceeds the 8-step quality of recent few-step discrete diffusion language models. Our work challenges the widely-held hypothesis that discrete noising processes are necessary for generative modeling over discrete modalities and paves the way toward accelerated language modeling at scale. Code is available at https://github.com/david3684/flm.

语言模型连续流生成加速扩散模型

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