arXiv:2603.04411cs.CLcs.AI2026-03

按语义动态压缩键值缓存,大幅省内存且保持生成质量

One Size Does Not Fit All: Token-Wise Adaptive Compression for KV Cache

  • 根据每个词元语义重要性动态分配压缩率
  • 在长文本任务上仅用6%缓存仍保持94%性能
  • 可与剪枝方法结合,适合高吞吐推理场景

尽管大型语言模型进展迅速,其键值(KV)缓存不断增长的内存开销仍是高效推理的关键瓶颈。虽然降维提供了一条有前景的压缩路径,但现有方法要么需要从头开始代价高昂的预训练,要么在高压缩率下导致严重性能下降。本文提出DynaKV,一种新型后训练低秩KV缓存压缩框架。据我们所知,DynaKV是首个根据词元语义意义动态分配压缩率的方法,从而在激进压缩比下实现更高保真度。大量实验表明,该方法持续优于现有最先进压缩技术,在显著减少内存占用的同时保持优异生成质量。此外,本方法与序列级剪枝方法正交;与SnapKV结合使用时,仅保留6%的KV缓存即可在LongBench基准上维持94%的基线性能。

原文摘要 · Abstract (English)

Despite the remarkable progress of Large Language Models (LLMs), the escalating memory footprint of the Key-Value (KV) cache remains a critical bottleneck for efficient inference. While dimensionality reduction offers a promising compression avenue, existing approaches typically either necessitate prohibitively expensive pre-training from scratch or suffer from severe performance deterioration under high compression regimes. In this work, we propose DynaKV, a novel post-training framework for low-rank KV cache compression. To the best of our knowledge, DynaKV is the first method to dynamically allocate compression rates to individual tokens according to their semantic meaning, which allows it to achieve better fidelity at aggressive compression ratios. Extensive experiments demonstrate that our method consistently outperforms existing state-of-the-art compression techniques, achieving significant memory reduction while maintaining competitive generation quality. Furthermore, our approach is orthogonal to sequence-level pruning methods. When integrated with SnapKV, DynaKV retains only 6% of the KV cache while maintaining 94% of the baseline performance on the LongBench benchmark.

KV缓存压缩大模型推理动态压缩后训练

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