ValueDiff:按价值向量偏离度淘汰 KV Cache,弱汇聚模型也能省显存
ValueDiff: Value-Geometric KV Cache Eviction for Sink-Suppressed LLMs
KV Cache 省显存的新思路:不看注意力看价值向量偏离度,MATH-500 上比老方法最高多约 20 分。
论文提出 ValueDiff,一种针对弱注意力汇聚模型的 KV Cache 淘汰方法,按 token 价值向量相对缓存均值的 L2 偏离度排序。在 RULER 基准、2k token 缓存预算下,该方法在七个弱汇聚模型上保留 88–99% 的稠密注意力成绩,其中 6 个模型表现最佳。LongBench 4k 预算下平均保留 92%,高于此前最强基线的 83%。在 MATH-500 的 25% 缓存预算下,它是所有受测弱汇聚模型上最强的非稠密方法,在门控注意力模型上比先前方法最高高出约 20 分。
ValueDiff: Value-Geometric KV Cache Eviction for Sink-Suppressed LLMs
Modern LLMs with QK-normalization, gated attention, learned attention sinks, or logit softcapping exhibit weaker persistent attention sinks, on which existing KV cache eviction methods primarily rely. We observe that across these models, weaker sinks co-occur with greater value-vector dispersion relative to key-vector dispersion. Motivated by this value-side dispersion, we present ValueDiff, a value-geometric eviction that ranks tokens by the L2 deviation of their value vectors from the cache mean. The same score arises as the minimal-disturbance eviction under a max-entropy assumption about future attention. We evaluate under fixed cache budgets, with eviction at every block boundary during prefill and at every decoding step during generation. On RULER at a tight 2k token budget, ValueDiff retains 88--99\% of dense across seven sink-suppressed models (best on 6 out of 7). On LongBench at the 4k budget, ValueDiff averages 92\% retention across sink-suppressed models versus 83\% for the strongest prior baseline. On MATH-500, ValueDiff is the strongest non-dense method on every sink-suppressed model tested at the 25\% cache budget, outperforming prior methods by up to $\sim$20 points on gated-attention models. Across all three benchmarks, value geometry emerges as the more reliable query-invariant eviction signal for sink-suppressed models.