[控场AI]
概念draft model

草稿模型

投机解码中使用的小型辅助模型,负责快速自回归生成候选token序列,供目标大模型并行验证,其与目标模型的分布对齐程度直接决定接受率和加速效果。

时间轴 (近 90 天)

11月3日

通过将幂参数从零退火到一,TVP在三种草稿模型架构和三种目标模型上取得了最高或并列最高的接受长度。

待验证50%
11月3日

现有草稿模型多采用交叉熵(CE)或KL散度训练。

待验证50%
10月22日

传统软标签蒸馏要求草稿模型在每一个位置上都逼近目标模型的完整概率分布,包括高熵区域,这与推理时最大化接受长度的目标不一致。

待验证50%
10月20日

投机解码的核心思想是:用一个较小的草稿模型快速生成多个候选 token,然后让目标大模型一次性并行验证这些 token 是否可接受。

待验证50%
10月20日

蒸馏模型通过让草稿模型拟合目标模型的输出分布来进一步提升对齐度,从而抬高接受率。

待验证50%
10月20日

同系列小模型(如大模型的 1/10 参数量版本)因训练数据与架构相近,输出分布与目标模型较为接近,是草稿模型的常见选择。

待验证50%
10月19日

现有草稿模型在智能体场景下的接受长度最多下降80%。

待验证50%
9月11日

推测解码实现需要在草稿模型的推理速度和预测准确率之间找到平衡点

待验证50%
9月11日

草稿模型通常一次生成3到8个候选token

待验证50%
9月11日

协同设计从一开始就将草稿模型与目标模型作为一个整体系统来设计和训练,以提升token接受率

待验证50%

还有 3 条时间轴事件

全部知识事实 (13)

待验证

通过将幂参数从零退火到一,TVP在三种草稿模型架构和三种目标模型上取得了最高或并列最高的接受长度。

50%
待验证

现有草稿模型多采用交叉熵(CE)或KL散度训练。

50%
待验证

传统软标签蒸馏要求草稿模型在每一个位置上都逼近目标模型的完整概率分布,包括高熵区域,这与推理时最大化接受长度的目标不一致。

50%
待验证

投机解码的核心思想是:用一个较小的草稿模型快速生成多个候选 token,然后让目标大模型一次性并行验证这些 token 是否可接受。

50%
待验证

蒸馏模型通过让草稿模型拟合目标模型的输出分布来进一步提升对齐度,从而抬高接受率。

50%
待验证

同系列小模型(如大模型的 1/10 参数量版本)因训练数据与架构相近,输出分布与目标模型较为接近,是草稿模型的常见选择。

50%
待验证

现有草稿模型在智能体场景下的接受长度最多下降80%。

50%
待验证

推测解码实现需要在草稿模型的推理速度和预测准确率之间找到平衡点

50%
待验证

草稿模型通常一次生成3到8个候选token

50%
待验证

协同设计从一开始就将草稿模型与目标模型作为一个整体系统来设计和训练,以提升token接受率

50%
待验证

传统草稿模型针对单一目标模型在狭窄分布上训练,工作负载变化时接受率会急剧下降,且每更换一个目标模型就必须重新完成整个预训练过程

50%
待验证

传统草稿模型训练方案需要消费目标模型的隐藏状态并在目标模型logits上进行蒸馏,产生对目标模型的强绑定依赖

50%
待验证

投机解码的草稿模型推理成本通常仅为主模型的1/5-1/10

50%

来源文章