Gemma 4 的 MTP 草稿模型解析:投机解码,与一个只有六成体积的 4 比特版本
来源:dev.to — 2026-10-15
📋 概述
模型一次前向传播只写一个 token,而每一遍都要把所有权重读一遍:在小 GPU 上为单个用户服务时,算力很轻、显存流量很重,于是每一步都留下闲置算力。投机解码就是来用这份余量的:小模型先猜接下来几个 token,主模型(目标模型)一次前向并行校验这些猜测,从头接受直到第一个不同意的位置,丢弃其余并在该点补一个自己的 token,所以每一步至少产出一个 token,最多产出「草稿数加一」。Gemma 4 的草稿模型与经典做法不同,不是同族小模型,而是目标模型的附加件——Google 称之为多 token 预测,一个 4 层轻量模型,直接读取目标模型的激活与 KV cache(E2B 是 1536 宽隐藏状态,草稿本身只有 256 宽、词表与目标相同),因此它不保留独立的会话缓存,显存成本基本只有权重。作者给出 Google QAT 草稿模型的精确 Q4_0 重打包版,文件从 97.8 MB 缩到 59.2 MB。
🔑 核心要点
- 需要它的场景很明确:单用户时 GPU 等的是显存不是算力,一次校验两三个 token 就等于省下两三次前向传播。
- 校验规则是从头接受直到第一个不同意的猜测,丢弃其余,并在该点补上目标模型自己的一个 token。
- 输出始终属于目标模型:每个 token 要么与目标模型会选的相同,要么是它自己的替换。
- 草稿模型很轻:读目标的 1536 宽隐藏状态,自身是 256 宽 4 层的小 transformer,四层全部复用目标的 KV cache、只算 query。
- 体积的六成:Google 的 QAT 草稿模型按 4 比特网格存储后从 97.8 MB 降到 59.2 MB。
💡 金句
草稿模型只改变一件事:每一次昂贵的前向传播能产出多少个 token。
👍 0
👎 0
← 返回 dev.to 首页