AFTERGLOW:给KV缓存淘汰打分加一道前向衰减,长文本答案不再被截断

alphaXiv 收录的论文摘要提出一种免训练插件 AFTERGLOW,修补现有 KV 缓存淘汰方法只看答案开头、丢弃后半段的毛病。在 RULER-4K 检索任务、87.5% 淘汰率下,它把四种方法的准确率从 49–74 提升到 98–100。

一句话看懂:alphaXiv 收录的论文摘要提出一种免训练插件 AFTERGLOW,修补现有 KV 缓存淘汰方法只看答案开头、丢弃后半段的毛病。在 RULER-4K 检索任务、87.5% 淘汰率下,它把四种方法的准确率从 49–74 提升到 98–100。

事件核心:发生了什么

长上下文推理时,大模型要用 KV 缓存保存历史 token 的键和值,内存和注意力开销都随长度增长。查询感知淘汰法按注意力分数删掉“不重要”的缓存,但作者发现这类分数只能标出相关片段的起点:以 Llama-3.1-8B 为例,87.5% 淘汰率下,UUID 答案第一个 token 平均在 0.47 的 KV head 对中保留,第 9–16 个 token 只剩 0.06,答案因此容易被截断或复述出错。

AFTERGLOW 的做法是把每个 token 的分数设为它自身分数与前序 token 衰减分数的最大值,其中 γ 取 0.95,前四个注意力汇 token 不参与传播;淘汰流程本身不变。摘要称,这在 Qwen3-8B 和 Llama-3.1-8B 上覆盖 4K 至 64K 上下文、4B 至 14B 参数规模,并在 LongBench 上持平或提升。

为什么重要

KV 缓存淘汰是降低长文本推理显存占用的主流手段,但“省内存”和“答案完整”一直存在取舍。AFTERGLOW 不改模型、不重训、不增加缓存预算,只调整打分规则,意味着现有推理框架的淘汰策略可能直接受益。目前公开信息显示,上述结论来自论文摘要与示例,并非已上线产品,也未经独立复现。

对用户/开发者/创作者的影响

对跑本地推理或长文档问答的开发者,这类插件式改动实现成本低,适合先在小规模评测上验证:重点看精确复制类任务,例如工具输出、编号、标题等容易被截断的字段。对依赖 Agent 读取长工具返回结果的团队,答案尾部是否保留,直接影响下游解析的准确性。

GamsGo AI

AI 工具推荐

想把多个 AI 模型放在一个入口?

GamsGo AI 集成 ChatGPT、DeepSeek、Gemini、Claude、Midjourney、Veo 等常用模型,适合写作、绘图、视频和日常 AI 工作流。

了解 GamsGo AI

推广链接:通过此链接购买,我可能获得佣金,不影响你的价格。

值得关注的后续

一是 AFTERGLOW 是否被 vLLM 等推理框架或主流淘汰方法采纳为默认选项;二是不同模型、不同 γ 取值下效果是否稳定;三是真实工具调用场景的复现结果能否与摘要一致。

来源:alphaXiv

celebrityanime
celebrityanime
文章: 29843

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注