首页 > AI前沿 > DeferKV: Rethinking Eviction Timing for One-Shot KV Cache Compression

DeferKV: Rethinking Eviction Timing for One-Shot KV Cache Compression

arXiv自然语言 2026-10-05 21:13 6 阅读 查看原文

Long-context large language models (LLMs) have demonstrated strong capabilities across a wide range of tasks, but the growing KV cache introduces substantial memory and inference overhead.

Existing one-shot KV cache compression methods typically commit to irreversible eviction immediately after prefill, before any signal from actual generation becomes available.

Our quantitative analysis shows that early queries from the actual generation stage provide attention signals that are more consistent with subsequent decode attention, with the largest single-step gain occurring at the prefill-decode boundary.

Based on this observation, we propose DeferKV, which moves the eviction decision from the end of prefill to the first real decoding step and temporally combines prompt-side and decode-side observations, thereby better aligning KV importance estimation with subsequent generation requirements.

DeferKV requires no additional training, draft model, or future-query prediction module, making it simple and easy to deploy.

Experiments on LongBench, RULER, and Needle-in-a-Haystack demonstrate that DeferKV consistently improves model performance under KV cache compression while maintaining low inference latency.