Epiphany-Aware KV Cache Eviction Without the Attention Matrix
About
As reasoning models emit chains of thought tens of thousands of tokens long, KV cache increasingly becomes a deployment bottleneck. Existing cache eviction methods rank tokens by attention weight, which is a noisy importance proxy in long reasoning traces, and prohibits the use of fused kernels in production inference by forcing the model to materialize the attention matrix. In this work, we instead score tokens with a metric we term the epiphany score: the change in the model's internal representation, read directly from the forward pass with no attention matrix and negligible extra state. Our resulting cache eviction method, EpiKV, requires no training, classifier, or custom kernel, and can be used directly in FlashAttention inference stacks unchanged -- scaling to a 16x longer feasible context than attention-based scoring. upper-mid layers negatively) and remove a positional trend with a causal rolling z-score. At a 4096-token cache EpiKV reaches 72% on MATH-500, matching the strongest attention-based baseline (ThinKV 71%, H2O 67%); a lag-normalized KV variant reaches 37% on AIME-2024 at 8192 tokens against the best of them (33%), at up to 2.8x the speed.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Mathematical Reasoning | MATH 500 | Accuracy72 | 589 | |
| Mathematical Reasoning | AIME 2024 (test) | Accuracy36.7 | 294 | |
| Math Reasoning | AIME 2024 | Mean Wall-Clock Time (s)440.5 | 61 | |
| Mathematical Reasoning | MATH 500 | Mean Wall-Clock Time113.8 | 49 | |
| Math problem solving | AIME 2024 | Mean Peak GPU Memory (MB)1.55e+4 | 41 | |
| Mathematical Reasoning | MATH 500 | Mean Peak GPU Memory (MB)1.55e+4 | 37 |