Every Cache Entry Earns Its Place: Global Allocation of Resolution and Coverage for KV Cache Compression
cs.LG
Submitted: 2026-08-07
Updated: 2026-10-10
License: http://creativecommons.org/licenses/by/4.0/
The gist: As large language models (LLMs) process increasingly long contexts, KV cache storage and repeated access have become a major bottleneck.
Terminology
Abstract
As large language models (LLMs) process increasingly long contexts, KV cache storage and repeated access have become a major bottleneck. Existing KV cache compression methods rely on predefined, fixed compression rules and are typically developed around either token eviction or merging. As a result, cache resources can neither flow freely across layers, heads, and context slots, nor be jointly allocated to balance local resolution and information coverage. Therefore, we propose GraceKV, a global approach for the allocation of resolution and coverage in KV cache compression, and formulate the compression process as a global resource allocation problem under a fixed cache budget. GraceKV treats each layer-KV head-slot combination as an atomic unit and builds a prototype tree. Leaf nodes correspond to token-level KV entries, while each internal node uses a single prototype to compress the KV space covered by its children. A set of non-overlapping nodes in the tree forms the representation of an atomic unit. Adding the root of a new tree expands information coverage, whereas splitting a selected node improves local resolution. All candidate actions compete globally for a shared cache budget. Finally, the nodes retained across all trees form the compressed KV cache. This process adaptively determines the allocation of cache resources among atomic units globally and the balance between resolution and coverage. GraceKV requires no additional training, and the entire compression and inference process is performed on the GPU. Systematic experiments across diverse long-context tasks and compression ratios show that GraceKV ranks first in 24 of 32 settings and remains robust up to 128-fold compression. These results validate the effectiveness of global budget allocation in coordinating information coverage and local resolution.
Sources
- SeKV: Resolution-Adaptive KV Cache with Hierarchical Semantic Memory for Long-Context LLM Inference
- PyramidKV: Dynamic KV Cache Compression based on Pyramidal Information Funneling
- Model Tells You What to Discard: Adaptive KV Cache Compression for LLMs
- The Llama 3 Herd of Models
- RULER: What's the Real Context Size of Your Long-Context Language Models?
- CentroidKV: Efficient Long-Context LLM Inference via KV Cache Clustering
- SpeCache: Speculative Key-Value Caching for Efficient Generation of LLMs
- ZSMerge: Zero-Shot KV Cache Compression for Memory-Efficient Long-Context LLMs
- Qwen2.5 Technical Report
- D2O: Dynamic Discriminative Operations for Efficient Long-Context Inference of Large Language Models
- Model Tells You Where to Merge: Adaptive KV Cache Merging for LLMs on Long-Context Tasks
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks