跳到正文
我的
arXiv · cs.CL· Hao Gu·· 1 天前AI 评分33

KV Cache 淘汰的门控机制

The Gating Mechanism of KV Cache Eviction

来源与原文

arXiv · cs.CL

作者:Hao Gu

发布时间:(北京时间)

阅读原始来源 · arxiv.org ↗
完整链接与内容说明

https://arxiv.org/abs/2611.07605v1

本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。

AI 导读

KV Cache 淘汰需要一个可学习门控来判断哪些条目承载其余缓存无法提供的信息,但该门控在语言建模损失下训练时始终接近常数分数,因为冻结骨干本就适配稠密注意力,其损失偏好空操作。将稠密注意力置于门控触及范围之外并以稠密模型预测为目标,可同时消除这一障碍与梯度归责问题;在 75% KV Cache 压缩下,该门控在 Qwen3-8B 上于 8K 的 RULER 达到 90.14、LongBench 达到 46.03,分别低于全缓存推理 3.72 和 1.36 分。

来源:arXiv · cs.CL · arxiv.org

这条消息,你怎么看?

把新闻带到 MickerBook,和社区伙伴聊聊;也可以去 AIHER 看看模型背后的角色与梗。