Long-Context Drafting Does Not Require Long-Context Drafters
论文指出超长上下文下投机解码的加速会消失,因为草稿模型需处理完整前缀。作者提出无需训练长上下文草稿模型:冻结短上下文草稿器,通过限制注意力缓存和重定位旋转位置索引两个推理期机制,让其在200万token下保持短上下文时的加速比,无需改权重,vLLM侧单一配置即可覆盖4k到2M token。
先用站内已采集内容作答;站内不足时联网检索,每条结论都给出来源链接
2 条 · 点击可看原文全文与中文译文
论文指出超长上下文下投机解码的加速会消失,因为草稿模型需处理完整前缀。作者提出无需训练长上下文草稿模型:冻结短上下文草稿器,通过限制注意力缓存和重定位旋转位置索引两个推理期机制,让其在200万token下保持短上下文时的加速比,无需改权重,vLLM侧单一配置即可覆盖4k到2M token。
AbridgeKV 训练模型把已完成的长推理文本块压缩进少量可学习 pause token 的 KV 条目,从而释放原始缓存,后续 token 改为关注这些 pause token。在 AIME 等 446 道竞赛题、32K 生成下,仅保留四分之一缓存即可匹配全 KV 模型精度,vLLM 中单序列缓存缩小 3.5 倍、H100 完成速度提升 3.2 倍。
现有资料只能给出方向性结论,不足以给出 vLLM 中可直接落地的具体配置项、参数名或代码。资料中唯一直接讨论「长上下文 + 投机解码 + 草稿模型」的是 1,但它描述的是方法层面的两个推理期机制,并未列出 vLLM 的配置字段。 可确认的部分(来自 [1])
1. 限制注意力缓存(限制草稿器可见的缓存范围); 2. 重定位旋转位置索引(把位置索引重新映射,避免超出短上下文草稿器的训练分布)。
资料未覆盖的部分
旁证(非草稿模型配置,仅供参考) 2 讨论的是长上下文下的 KV 缓存压缩而非投机解码:把长推理压缩进可学习 pause token 的 KV 条目,在 vLLM 中单序列缓存缩小 3.5 倍、H100 完成速度提升 3.2 倍2。它与 1 属于不同技术路线,不能直接当作草稿模型配置依据。 建议:若要落地,需回到 1 原文确认其 vLLM 实现细节(是否以插件/分支形式提供),现有摘要信息不足以支撑写出配置代码。