cs.LGAug 25, 2026

Trust the Mass: Forced Weights in KV-Cache Eviction

Authors: Jack Shi, Jerry Gu

Organizations: Stanford University

Abstract

Every deployed sparse-attention or KV-cache-eviction rule keeps a subset of the keys, discards the rest, and renormalizes the attention weights over the kept set. Enumerating the exact best subset under that constraint on 168,192168{,}192 attention rows from five models shows that keeping the largest weights is already near-optimal, since the best subset closes only a median 22 to 5%5\% of the remaining gap to full attention. If selection closes this little, published margins between eviction methods must come from elsewhere, so we measure the bytes each method holds. In the shared evaluation pipeline, the strongest query-agnostic methods hold the full cache because their per-head selections are stored as masks, and only ragged per-head storage frees that memory. Enforcing a nominal budget on one fixed selection costs 1414 to 6262 benchmark points. We trace an 87.687.6-point retrieval margin to rankings computed while the question is visible. ContourKV, a training-free allocator built from the dropped-mass statistic, wins 9393 of 160160 paired comparisons against that state of the art and loses 2222 at the byte count of the budget-enforcing baselines, and it ties the strongest of them.

Figures & tables

Appendix figures & tables5 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Beyond Sparse Weights: When Is Attention Compressible?

    Aug 21, 2026Chiwun Yang, Xiaoyu LiKV-Cache CompressionMemory Compression

  2. Page-EntroKV: Hardware-Aligned, Entropy-Weighted KV-Cache Eviction under Grouped-Query Attention

    Oct 2, 2026Inbasekaran SKV-Cache EvictionMemory-Efficient Inference

  3. One Distribution, Two Channels: Coverage-Calibrated KV Retention and Compensation

    Sep 28, 2026Shuxin Liu, Qing Liu, Yi Du +1KV-Cache EvictionKV-Cache Management