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 侧的代码、类名、配置项或补丁细节。资料2虽然也提到 vLLM,但讲的是另一套 KV 压缩方案,与「限制注意力缓存 + 重定位旋转位置索引」不是同一机制,不能用来补足。 资料1中能确认的部分:
因此,关于「具体实现改动」(例如缓存窗口如何截断、RoPE 位置索引如何重映射、是否涉及 attention metadata 或 block table 的修改),资料中没有可引用的内容,不做推测。 可对照的另一条线索(不同机制,仅供区分):资料2的 AbridgeKV 走的是训练侧路线,把长推理压缩进可学习 pause token 的 KV 条目,在 vLLM 中实现单序列缓存缩小 3.5 倍、H100 完成速度提升 3.2 倍2;它并未涉及旋转位置索引重定位2。