草稿预判阶段小模型根据当前上下文,一口气连续预判出后面 K 个候选 token,常见取值为 4‑8 个。小模型体量小,这一步计算成本很低,很快输出一串草稿文本。
大模型一次性并行校验(核心加速点)把小模型猜出来的 K 个草稿,全部拼接到现有文本之后,大模型只执行一次前向传播运算。
重点:验证 K 个 token 和生成 1 个 token,读取大模型全套参数只需要一次。利用 Transformer 可以对多个序列位置并行做矩阵运算的特性,一次性算出每一个位置上大模型自己的概率分布,不需要跑 K 遍大模型。之前原生模式生成 K 个字,要读取 K 次整套大模型参数;现在只读取 1 次。多出来的验算计算,刚好消化掉 GPU 原本闲置的算力。