Behavior-Preserving KV Cache Compression

Doo Hwan Hwang, Junyoung Jang, Junho Na, Hosung Lim, Kee-Eung Kim


Abstract
KV caches are a major bottleneck in long-context inference and long-form generation with large language models. Existing training-free eviction policies largely rely on proxy importance signals, such as attention mass, to decide which past tokens to retain. We argue that cache compression should instead preserve the predictive behavior of the full-cache model, retaining entries whose removal would substantially change the model’s output distribution. We propose Behavior-Preserving KV Cache Compression, a training-free framework that scores candidate evictions by estimating the compressed-cache logits induced by their removal and evaluating the resulting KL to the full-cache next-token distribution. Using pre-eviction forward statistics, the method avoids running separate masked forward passes for each candidate. Across diverse architectures and both prefill-time and generation-time compression, our method delivers substantial gains in downstream task quality over lightweight attention-based heuristics at matched retained-KV budgets, with the largest gains under aggressive compression. It achieves these gains with additional compression-time computation while retaining an end-to-end speedup over full-cache inference in our evaluated settings.
Anthology ID:
2026.emnlp-main.2010
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
45807–45829
Language:
URL:
https://aclanthology.org/2026.emnlp-main.2010/
DOI:
Bibkey:
Cite (ACL):
Doo Hwan Hwang, Junyoung Jang, Junho Na, Hosung Lim, and Kee-Eung Kim. 2026. Behavior-Preserving KV Cache Compression. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 45807–45829, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Behavior-Preserving KV Cache Compression (Hwang et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.2010.pdf