arXiv:2608.00146cs.CLcs.AI2026-08

用扩散模型加速文本生成,256词并行输出,速度超传统模型。

DiffusionGemma Technical Report

论文配图:DiffusionGemma Technical Report
图 1 · 摘自论文原文
  • 通过扩散机制并行生成256词块,突破逐词生成瓶颈。
  • 单卡H100每秒生成约1500词,比先进推测解码快得多。
  • 保留原模型思维模式与长上下文能力,可混合使用扩散与自回归解码。

我们提出DiffusionGemma,一种实验性开源语言模型,利用离散扩散机制实现极高速文本生成。不同于逐词解码的自回归(AR)大模型,DiffusionGemma在每轮前向传播中并行精炼256个词块,有效规避了序列解码瓶颈。该模型通过微调具有38亿激活参数和252亿总参数的混合专家模型Gemma 4获得。采用计算高效的两阶段训练流程,仅消耗原始AR模型10%以下的训练令牌预算。第一阶段采用监督微调实现双向去噪;第二阶段结合强化学习与采样器蒸馏,协同优化生成质量与推理效率。在全评估套件下,DiffusionGemma平均每轮生成约20词,在单张NVIDIA H100 GPU上达到约1500词/秒的生成速度,显著超越当前最先进的推测解码方案。同时,它仍保留原模型的思维模式支持、多模态输入与长上下文能力。尽管经过扩散微调,其自回归生成性能仅轻微下降,为混合扩散-自回归解码提供了可行路径。

原文摘要 · Abstract (English)

We introduce DiffusionGemma, an experimental open-weight language model that uses discrete diffusion to generate text at exceptionally high speed. Rather than decoding one token at a time, DiffusionGemma iteratively refines blocks of 256 tokens in parallel, avoiding the sequential decoding bottleneck of conventional autoregressive (AR) large language models. Instead of training from scratch, we obtain DiffusionGemma by fine-tuning the mixture-of-experts Gemma 4 model with 3.8B activated and 25.2B total parameters. Our compute-efficient two-stage training pipeline uses fewer than 10% of the starting AR model's total training token budget. The first stage uses supervised fine-tuning to teach bidirectional denoising, while the second stage combines reinforcement learning with sampler distillation to jointly improve generation quality and inference efficiency. DiffusionGemma establishes a new Pareto frontier for the trade-off between generation speed and model capability. Averaged across our full evaluation suite, it generates around 20 tokens per forward pass and achieves roughly 1,500 output tokens per second on a single NVIDIA H100 GPU, which is substantially faster than AR models even with state-of-the-art speculative decoding. DiffusionGemma also retains the starting model's support for thinking mode, multimodal inputs, and long contexts. Despite diffusion fine-tuning, it remains capable of AR generation with only minor performance degradation, suggesting a path toward hybrid diffusion-AR decoding.

扩散模型文本生成高效推理混合解码

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