Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Value-Guided KV Compression for LLMs via Approximated CUR Decomposition

About

Key-value (KV) cache compression has emerged as a critical technique for reducing the memory and latency overhead of autoregressive language models during inference. Prior approaches predominantly rely on query-key attention scores to rank and evict cached tokens, assuming that attention intensity correlates with semantic importance. However, this heuristic overlooks the contribution of value vectors, which directly influence the attention output. In this paper, we propose CurDKV, a novel, value-centric KV compression method that selects keys and values based on leverage scores computed from CUR matrix decomposition. Our approach approximates the dominant subspace of the attention output $softmax(QK^T)V$, ensuring that the retained tokens best preserve the model's predictive behavior. Theoretically, we show that attention score approximation does not guarantee output preservation, and demonstrate that CUR-based selection minimizes end-to-end attention reconstruction loss. Empirically, CurDKV achieves up to 9.6% higher accuracy than state-of-the-art methods like SnapKV and ChunkKV under aggressive compression budgets on LLaMA and Mistral, while maintaining compatibility with FlashAttention and Grouped Query Attention. In addition to improved accuracy, CurDKV reduces generation latency by up to 40% at high compression, offering a practical speed-accuracy tradeoff.

Ayan Sengupta, Siddhant Chaudhary, Tanmoy Chakraborty• 2025

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningAIME 2025
Accuracy53.75
353
Mathematical ReasoningHMMT 2025
Accuracy39.58
241
Mathematical ReasoningAIME 2026
AIME 2026 Accuracy61.25
80
Scientific ReasoningGPQA Diamond
Accuracy48.48
54
Mathematical ReasoningAIME 26
Pass@1 Accuracy61.3
4
Mathematical ReasoningAIME25
Pass@1 Accuracy53.8
4
Mathematical ReasoningHMMT25
Accuracy (Pass@1)39.6
4
Mathematical ReasoningMATH
Pass@1 Accuracy75.4
4
Scientific ReasoningGPQA D
Pass@1 Accuracy48.5
4
Showing 9 of 9 rows

Other info

Follow for update