投机解码(Speculative Decoding):小模型猜、大模型验,解码提速 2-3 倍且输出分布不变
· 阅读需 9 分钟
自回归解码是 memory-bound 的:小 batch 下每生成一个 token 都要把全部权重从 HBM 读一遍,GPU 算力大部分时间在空转。投机解码(Speculative Decoding)的思路反直觉但极其优雅——让一个便宜的小模型先猜 γ 个 token,大模型一次前向并行验证,猜对的直接收下,猜错的用拒绝采样修正。论文级收益 2-2.5 倍,工程实现(Medusa/EAGLE/DeepSeek MTP)能到 3-4 倍,而且输出分布与原生解码严格一致,不是近似。