Rebas Daily PERSONAL AI DAILY — 自动选题 · 核查 · 撰写 NO.007 — 2026-07-11
PAPER H 17 约 1 分钟

KV Cache优化终于能横向比了

统一比较四类KV Cache优化:省得多,不一定跑得快、答得好。

让 AI 读一摞长文档时,它会保留一份 KV Cache——生成回答时随手查阅的“草稿纸”,免得反复计算。文本越长,这份草稿越占显存。量化、剪枝和合并等压缩办法都想给它瘦身,但过去常用不同模型、任务和系统各测各的,数字很难横向比较。

Agrawal 和 Mayer 因此搭建了一套工作负载感知基准:把 KIVI、TurboQuant、SnapKV 和 CaM 放进较统一的条件,用 Llama-3.1-8B-Instruct 与 Mistral-7B-Instruct-v0.3,覆盖多文档问答、单文档问答、少样本学习和摘要;既看回答质量,也测输出吞吐量、首个词元等待时间和实际压缩率。

最值得注意的结论是:压缩率高,并不能可靠预测端到端表现。作者报告,KIVI4 在不同模型上的质量最稳定,SnapKV 的长上下文吞吐量最强;CaM 在部分问答任务上收益较大,却会随任务明显波动。换句话说,KV Cache 优化没有通吃方案,部署时得先看具体工作负载,而不能只盯着“省了多少显存”。


供稿材料 SOURCES — 1

← 返回 2026-07-11 · 学术板块