AI觉醒星球
Awakening is here
Knowledge File / AI小生意项目库
2026-08-09 0 浏览 会员

谷歌的DiffusionGemma证明构建文本扩散模型无需从头训练

谷歌DeepMind发布DiffusionGemma技术报告,展示如何将现有Gemma模型改装为扩散模型,无需从头训练,生成速度大幅提升,但绝对性能有所下降。

SOURCE / AI小生意项目库 MIN / 9 ACCESS / 会员 POST / 2026-08-09 18:01:26

原贴

查看原文
作者:Jonathan Kemper 来源站点:the-decoder.com 原贴时间:

原文

Instead of training a new model from scratch, Google DeepMind retrofitted Gemma 4 into a diffusion model. The newly published report explains how it works and where the tradeoffs are. Google DeepMind released DiffusionGemma as a model in mid-June and has now followed up with the technical report. Unlike standard language models that generate text one token at a time, DiffusionGemma refines blocks of 256 tokens in parallel, similar to how image AIs pull a picture out of noise. On an Nvidia H100 accelerator, the model hits about 1,500 tokens per second. Building a new model from scratch wasn't necessary. The team started with the existing Gemma-4-26B-A4B and converted it into a diffusion model using less than ten percent of the original training token budget, according to the report . In the first of two steps, the model learns to reconstruct noisy text blocks from example data. A combined phase of reinforcement learning and sampler distillation follows, which Google calls SD·RL. Reinforcement learning typically boosts answer quality, while sampler distillation lets the model get by with fewer compute steps. Google merges both into a single process. According to the report, this combined approach raises quality on reasoning benchmarks by an average of ten points while nearly quadrupling the number of tokens per compute step. As a side effect, DiffusionGemma's answers run about 50 percent shorter, which further boosts speed. Standard language models have to commit to the first digit of an answer before they've worked through the reasoning. In a math problem from the report, Gemma 4 starts its response with "-1," realizes during its derivation that "-25" is correct, and tacks on a correction afterward. DiffusionGemma develops the answer and reasoning in parallel, so it can fix mistakes before the output is finalized. Sudoku solving works on the same principle, since every entry depends on entries that come later. After minimal fine-tuning, DiffusionGemma solves close to 85 percent of puzzles correctly, while the base model fails at the task entirely. Structured outputs like JSON or code repairs finish after just two to three refinement steps, according to the report, because the input already determines most tokens. DiffusionGemma also keeps its original ability to generate text word by word, letting users switch between both modes depending on the task. Absolute performance falls short of the autoregressive base model. Google points to several reasons for this. DiffusionGemma wasn't trained as a diffusion model from the start but was retrofitted after the fact. The subsequent training phase was relatively short, and the second step, SD·RL, prioritized speed over peak quality. The architecture, training data, and other settings were also carried over from the original Gemma 4 model, which aren't necessarily ideal for diffusion. The model occasionally gets stuck in repetition loops, producing individual words multiple times in a row. This is an artifact of the aggressively reduced compute steps. On multimodal tasks, DiffusionGemma sometimes forgets to close its reasoning section properly, which artificially drags down benchmark scores. The speed advantage also holds mainly for single-user scenarios. Once about 32 concurrent requests hit the model, standard language models catch up on throughput. Google explicitly calls DiffusionGemma an experimental model and says the release is meant to speed up research on text diffusion while giving the community a foundation for specialized, resource-efficient adaptations.

中文翻译

谷歌DeepMind没有从头训练新模型,而是将Gemma 4改装成了扩散模型。新发表的报告解释了其工作原理及权衡之处。谷歌DeepMind于6月中旬发布了DiffusionGemma模型,如今又发布了技术报告。与逐一生成token的标准语言模型不同,DiffusionGemma并行精炼256个token的块,类似于图像AI从噪声中生成图片。在Nvidia H100加速器上,该模型每秒可处理约1500个token。从头构建新模型并非必要。研究团队从现有的Gemma-4-26B-A4B出发,将其转换为扩散模型,据报告称,使用的训练token预算不足原始预算的10%。在两个步骤中的第一步,模型学习从示例数据中重建有噪声的文本块。接下来是强化学习和采样器蒸馏的结合阶段,谷歌称之为SD·RL。强化学习通常提升答案质量,而采样器蒸馏让模型以更少的计算步骤完成。谷歌将两者合并为一个过程。据报告称,这种结合方法在推理基准上的质量平均提高了10个百分点,同时每个计算步骤的token数量几乎翻了两番。副作用是,DiffusionGemma的回答长度缩短了约50%,进一步提升了速度。标准语言模型在完成推理之前必须确定答案的第一个数字。在报告中的一道数学题中,Gemma 4以“-1”开头,在推导过程中意识到“-25”才是正确的,然后附上修正。DiffusionGemma并行构建答案和推理,因此可以在输出最终确定前修复错误。数独解题也基于同样原理,因为每个条目都依赖于后续条目。经过极简微调,DiffusionGemma正确解决了近85%的谜题,而基础模型完全无法完成该任务。据报告称,对于JSON或代码修复等结构化输出,仅需两到三个精炼步骤即可完成,因为输入已经确定了大部分token。DiffusionGemma还保留了逐词生成文本的原始能力,允许用户根据任务在两种模式间切换。绝对性能不及自回归基础模型。谷歌指出了几个原因。DiffusionGemma并非一开始就作为扩散模型训练,而是事后改装的。后续训练阶段相对较短,第二步SD·RL优先考虑速度而非峰值质量。架构、训练数据和其他设置也沿用了原始Gemma 4模型,不一定适合扩散。该模型有时会陷入重复循环,连续多次产生相同的词。这是激进减少计算步骤导致的伪影。在多模态任务上,DiffusionGemma有时忘记正确关闭推理部分,人为拉低基准分数。速度优势主要适用于单用户场景。一旦约32个并发请求到达模型,标准语言模型在吞吐量上就能赶上。谷歌明确将DiffusionGemma称为实验性模型,并表示发布该模型旨在加速文本扩散研究,同时为社区提供专用、资源高效适配的基础。

核心信息

谷歌DeepMind发布DiffusionGemma技术报告,展示如何将现有Gemma模型改装为扩散模型,无需从头训练,生成速度大幅提升,但绝对性能有所下降。

  • 谷歌DeepMind发布DiffusionGemma技术报告,展示如何将现有Gemma模型改装为扩散模型,无需从头训练,生成速度大幅提升,但绝对性能有所下降。
  • 原贴提到:Instead of training a new model from scratch, Google DeepMind retrofitte
  • 来源:the-decoder.com
试看内容

成为会员查看完整内容

你已经看到了这篇内容的前置整理,剩余深度部分仅对会员开放。

详细解读 信息差价值 参考来源
成为会员查看完整内容
上一篇 Google Deepmind 的 WeatherNext 同时预测气旋路径和强度 下一篇 AI的能源需求推动英伟达和亚马逊投入数十亿美元建设大规模电力基础设施