LLM Memory & RAG 相关度: 9/10

KV-CoRE: Benchmarking Data-Dependent Low-Rank Compressibility of KV-Caches in LLMs

Jian Chen, Zhuoran Wang, Jiayu Qin, Ming Li, Meng Wang, Changyou Chen, Yin Chen, Qizhen Weng, Yirui Liu
arXiv: 2602.05929v1 发布: 2026-02-05 更新: 2026-02-05

AI 摘要

该论文提出KV-CoRE方法评估LLM中KV-cache的数据依赖低秩可压缩性,并进行了大规模基准测试。

主要贡献

  • 提出KV-CoRE方法评估KV-cache可压缩性
  • 构建大规模KV-cache可压缩性基准测试
  • 分析模型、数据和语言与可压缩性的关系

方法论

使用基于SVD的KV-CoRE方法,通过计算Frobenius范数下的最优低秩近似,进行数据集层面和层级的评估。

原文摘要

Large language models rely on kv-caches to avoid redundant computation during autoregressive decoding, but as context length grows, reading and writing the cache can quickly saturate GPU memory bandwidth. Recent work has explored KV-cache compression, yet most approaches neglect the data-dependent nature of kv-caches and their variation across layers. We introduce KV-CoRE KV-cache Compressibility by Rank Evaluation), an SVD-based method for quantifying the data-dependent low-rank compressibility of kv-caches. KV-CoRE computes the optimal low-rank approximation under the Frobenius norm and, being gradient-free and incremental, enables efficient dataset-level, layer-wise evaluation. Using this method, we analyze multiple models and datasets spanning five English domains and sixteen languages, uncovering systematic patterns that link compressibility to model architecture, training data, and language coverage. As part of this analysis, we employ the Normalized Effective Rank as a metric of compressibility and show that it correlates strongly with performance degradation under compression. Our study establishes a principled evaluation framework and the first large-scale benchmark of kv-cache compressibility in LLMs, offering insights for dynamic, data-aware compression and data-centric model development.

标签

KV-cache 压缩 低秩近似 LLM 基准测试

arXiv 分类

cs.CL