English

SurfaceLogicKV: Surface and Logic Attention Behaviors are All You Need for Robust KV Cache Compression

Computation and Language 2025-08-25 v1 Artificial Intelligence

Abstract

The increasing input sequence length in Large Language Models (LLMs) puts significant pressure on key-value (KV) cache storage, making efficient inference challenging. Explicitly distinguishing attention behavior into our self-defined surface memorization and logic construction reveals essential roles in long-context reasoning. We observe that an individual attention head can display various behaviors, with nearly 98.5% effectively ignoring completely irrelevant information. The remaining 1.5% behaves as logic construction, and 0.5% behaves as surface memorization. Based on layer- and head-wise integration, we propose a novel two-stage SurfaceLogicKV method to utilize these attention behaviors for KV Cache compression. As a result, it achieves improved compressing robustness while maintaining competitive performance across various tasks and long sequences compared to baselines or even FullKV in some specific situations

Keywords

Cite

@article{arxiv.2508.15806,
  title  = {SurfaceLogicKV: Surface and Logic Attention Behaviors are All You Need for Robust KV Cache Compression},
  author = {Mengjie Li and William J. Song},
  journal= {arXiv preprint arXiv:2508.15806},
  year   = {2025}
}

Comments

18 pages, 9 tables, 10 pages