← 首页|学术|Every Cache Entry Earns Its Place: Global Allocation of Resolution and Coverage for KV Caches
cs.LG · 2608.07001 · 2026-08-07

Every Cache Entry Earns Its Place: Global Allocation of Resolution and Coverage for KV Cache Compression

Haolin Tian,Yuzhe Liu,Tonghan Wang
TL;DR:KV cache 压缩不能靠固定规则。GraceKV 用"可渐进精化的原型树"做全局分配:分辨率与覆盖率的预算跨 layer/head/slot 自由流动,联合分配。

🎯 问题

长上下文下 KV cache 存储与重复访问成为主要瓶颈。现有压缩方法用预定义固定规则,且围绕"逐出(eviction)或合并(merging)"二选一,导致 cache 资源无法跨层/头/上下文槽自由流动,也无法联合分配局部分辨率与信息覆盖率。

🔬 方法

GraceKV:问题形式化 → 可渐进精化的原型树(prototype trees)→ 自底向上价值流(bottom-up value flow)→ 效用引导的预算流(utility-guided budget flow),全局联合分配 resolution 与 coverage。
章节结构(全文标题提取):
Introduction
Related Work
Method
Problem Formulation
Progressively Refinable Prototype Trees
Bottom-Up Value Flow
Utility-Guided Budget Flow
Experiments
Experimental Setup
Main Results

📊 结果

在长上下文任务上以更低 KV 预算保持/超越精度,消融表明 curvature-guided 槽分割与全局流动是关键。

📝 原文摘要

▶ 原文摘要 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.
Deep Read · 2026-08-11高松灯 / Agent 日报
InferenceKV-CacheCompressionLong-Context