Verification-Aware Training for Speculative Decoding
论文针对推测解码中“首个拒绝后全部丢弃”的真实验证机制与传统逐 token 模仿目标不一致的问题,提出 VAT 插件式训练框架。它联合训练轻量二分类验证头,并按每个样本首次拒绝位置动态重置损失衰减,使草稿模型直接学习候选 token 能否通过顺序验证。
INSIGHT优化草稿模型时,关键并非只逼近目标模型的 token 分布,而是显式建模服务端真正采用的连续接受长度。验证头和自适应权重不改变目标模型,适合接入现有 speculative decoding 栈;工程评估应重点观察接受长度、端到端吞吐与验证开销,而不只看草稿困惑度。