arXiv:2606.29563cs.CLcs.AI2026-06

提升大模型推理效率,通过关注关键信息覆盖度优化缓存淘汰策略。

Coverage-Driven KV Cache Eviction for Efficient and Improved Inference of LLM

论文配图:Coverage-Driven KV Cache Eviction for Efficient and Improved Inference of LLM
图 1 · 摘自论文原文
  • 基于令牌覆盖率设计新型缓存淘汰机制,跨头与跨层增强信息保留。
  • 在相同内存约束下,性能最高提升10.35分,显著优于现有方法。
  • 适合资源受限场景下的大模型高效部署,尤其对长文本推理有帮助。

大型语言模型(LLMs)凭借处理长上下文输入的能力,在问答和摘要等复杂任务中表现优异。然而,其部署成本高昂,不仅源于自注意力机制的二次计算复杂度和自回归生成的高开销,还因推理过程中需存储大量键值(KV)缓存带来的显著内存压力。为降低内存消耗,现有缓存淘汰策略利用注意力稀疏性选择性保留部分标记。但此类方法导致性能明显下降,尤其在需要长上下文推理的任务中。我们发现性能下降与唯一标记覆盖率降低相关,并从理论上证明:覆盖率下降会限制输入与输出间的互信息,从而损害预测准确性。为此,我们提出K-VEC,一种覆盖感知的新型缓存淘汰策略,优先保障标记覆盖率。K-VEC引入跨注意力头与跨模型层的覆盖率模块,增强多头、多层的信息保留,缓解低覆盖率导致的性能退化。在16个LongBench子集上的评估显示,在相同淘汰率与内存约束下,相比现有方法,性能最高提升10.35分。全面验证表明该方法在资源受限环境下具有高效部署潜力。

原文摘要 · Abstract (English)

Large language models (LLMs) excel at complex tasks like question answering and summarization, thanks to their ability to handle long-context inputs. However, deploying LLMs is costly, not only due to the high computational demands of quadratic complexity of self-attention and auto-regressive generation, but also because of the significant memory overhead required for storing the key-value (KV) cache during inference. To reduce the memory cost, existing KV-cache eviction strategies leverage the sparsity in attention to selectively store a subset of tokens. While reducing the memory footprint, such approaches show a considerable drop in performance, especially in tasks that require long-context reasoning. We identify that the drop in performance is linked to a reduction in the coverage of unique tokens. Additionally, we theoretically show that reduced coverage limits the mutual information between inputs and outputs, thereby impairing predictive accuracy. To this end, we introduce K-VEC, a novel coverage-aware KV-cache eviction strategy that prioritizes token coverage while evicting tokens in the cache. K-VEC introduces a cross-head and a cross-layer coverage module to enhance token retention across attention heads and model layers, mitigating performance degradation caused by low coverage. Evaluated on 16 LongBench subsets, K-VEC exhibit up to 10.35 points improvement over the existing methods under the same eviction rate and memory constraint. Comprehensive evaluations validate the effectiveness of our approach and demonstrate its potential for efficient LLM deployment in resource-constrained settings.

大模型推理缓存优化长上下文

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。