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

KVpop -- Key-Value Cache Compression with Predictive Online Pruning

About

Key-value (KV) cache growth is a major bottleneck in autoregressive decoding, as memory and bandwidth scale linearly with context length. Existing KV eviction methods often rely on static heuristics or proxy scores, which poorly track future token utility and cause brittle eviction as relevance shifts. To address this, we introduce KVpop, which learns a fixed-budget KV eviction policy by directly supervising the keep-or-drop decision. The scorer is trained against a novel future-attention target, computed efficiently without materializing dense attention maps. We further introduce a delayed memory-based scorer that, uniquely among learned eviction methods, defers scoring for a fixed number of steps to exploit near-future context. On AIME and HMMT mathematical reasoning, KVpop retains 98% of full-attention performance on Qwen3-4B at 75% KV cache compression and 97% at 88% compression, consistently outperforming established eviction baselines. Qwen3-8B shows even stronger results, reaching near-full teacher performance. These results show that supervising eviction with future-attention signals cuts memory costs while maintaining quality.

Lukas Hauzenberger, Niklas Schmidinger, Anamaria-Roberta Hartl, David Stap, Thomas Schmied, Sebastian B\"ock, G\"unter Klambauer, Sepp Hochreiter• 2026

Related benchmarks

TaskDatasetResultRank
Scientific ReasoningGPQA Diamond
Accuracy59
94
Mathematical ReasoningHMMT 2502
Pass@131
26
Mathematical ReasoningHMMT 2511
Pass@140
26
Mathematical ReasoningAIME and HMMT Average
Pass@1 Absolute44
26
Code GenerationLiveCodeBench (LCB) V6
Pass@135
26
Showing 5 of 5 rows

Other info

Follow for update