arXiv cs.LG·13 天前★ 3

PeekKV 用单 token 探测实现按注意力头分配 KV 缓存

原文标题:PeekKV: The First Content Token Already Knows Which Heads Matter

重要度 3/5 · AI 综合评估报道影响力与新鲜度,4–5 分为高价值资讯。

AI 智能摘要

KV 缓存随线性增长成为 内存瓶颈。研究观察到 decode 阶段活跃在内容 间保持稳定,据此提出 PeekKV:在 后观察单个内容 ,联合获取头组得分与 token 重要性得分,按比例分配缓存预算。无需离线分析或训练,在 上以 10% 内存预算保留 95.7% 全缓存精度。

深读 · 结论先行AI 生成展开看要点与影响 ↓

PeekKV 仅用单 token 探测分配 KV 缓存, 上 10% 内存保留 95.7% 精度

  • •核心观察:prefill 与 decode 阶段活跃头会变化,但 decode 阶段过滤格式 token 后活跃头组成在内容 token 间稳定
  • •方法:prefill 后观察单个内容 token,联合得到头组得分与 token 重要性得分,按比例分配缓存预算
  • •无需离线分析或训练,可直接用于未见工作负载与新模型
  • •四个公开基准中过半模型对比取得最高精度, 上最接近全缓存精度

为什么重要:KV 缓存是的内存瓶颈,免训练、免离线剖析的方案对部署新模型和波动负载的推理服务方更实用。

以上深读由 AI 依据标题/摘要/正文节选生成,仅供参考;事实以原文为准。

阅读原文 +0
AI 处理过程展开/收起
原始标题:PeekKV: The First Content Token Already Knows Which Heads Matter
AI 中文标题:PeekKV 用单 token 探测实现按注意力头分配 KV 缓存
来源:arXiv cs.LG · 发布于 13 天前
分类:大模型 · AI 重要度:★ 3(3)
标签:KV 缓存 / 推理优化 / 注意力机制 / LLM 服务
关键词:KV 缓存压缩 / 推理优化 / 注意力头 / Llama 3.1 / 大模型部署
AI 处理:2026/10/20 15:02:19 · 模型 deepseek-chat · 规则版本 2026-10-10.1
AI 生成以上内容由 AI 自动生成:中文标题、摘要、分类与重要度;同一事件的多家报道会自动归并展示。 内容仅供参考,可能存在偏差,请以原文为准, 不构成任何投资或专业建议。
版权归原作者与原发布平台所有;若涉及侵权或信息错误,请在举报与侵权投诉页反馈,我们会及时处理。
每天 09:30,3 条 AI 圈真正重要的变化,免费订阅 →
PeekKV 用单 token 探测实现按注意力头分配 KV 缓存 - 智讯 AI News