Taming the Fragility of KV Cache Eviction in LLM Inference
On the map: Efficient LLM inference and serving → KV cache optimization
Abstract (arXiv)
Large language models have revolutionized natural language processing, yet their deployment remains hampered by the substantial memory and runtime overhead of the transformer's Key-Value cache. To mitigate this, recent methods employ a scoring-aggregation framework to evict unimportant cache entries, based on the stability assumption-that a fixed subset of entries remains consistently important during generation. However, prior work has largely focused on refining importance indicators for scoring, while defaulting to mean aggregation due to a faithful trust in the stability assumption. In this work, we argue that this underlying assumption is inherently fragile, making mean aggregation highly vulnerable in extreme cases. To counter this, we propose a simple yet elegant defensive aggregation strategy: a two-step, linear-time approach that controls worst-case risk, thereby defending against extreme cases with negligible computational overhead. Embodying this strategy, we propose a novel cache eviction method, DefensiveKV and its extension, Layer-DefensiveKV, which incorporates layer-wise budget allocation. Across seven task domains (18 datasets), our methods reduce generation quality loss by 2.3x and 4.3x respectively, versus the strongest baseline under a 20% cache size. These results set new performance benchmarks and pioneer a promising direction for optimizing cache eviction against underlying fragility through worst-case risk management. Our code is available at https://github.com/FFY0/DefensiveKV.
Key facts. Cited by 14 papers in the map (data through 7 Oct 2026). Placed in KV cache optimization, one of 12 directions in Efficient LLM inference and serving. Builds on 43 of its 51 references that are in the map. First posted on arXiv 15 Oct 2025.
Builds on
43 of 51 references.
Closest ancestors · 14
Same direction, newest first
- 2025CAKE: Cascading and Adaptive KV Cache Eviction with Layer Preferences16 Mar
- 2025CriticalKV: Optimizing KV Cache Eviction from an Output Perturbation Perspective6 Feb
- 2024Not All Heads Matter: A Head-Level KV Cache Compression Method with Integrated Retrieval and Reasoning25 Oct
- 2024KVSharer: Efficient Inference via Layer-Wise Dissimilar KV Cache Sharing24 Oct
- 2024DuoAttention: Efficient Long-Context LLM Inference with Retrieval and Streaming Heads14 Oct
- 2024ThinK: Thinner Key Cache by Query-Driven Pruning30 Jul
- 2024Ada-KV: Optimizing KV Cache Eviction by Adaptive Budget Allocation for Efficient LLM Inference16 Jul
- 2024Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference16 Jun
- 2024PyramidKV: Dynamic KV Cache Compression based on Pyramidal Information Funneling4 Jun
- 2024MiniCache: KV Cache Compression in Depth Dimension for Large Language Models23 May
- 2024SnapKV: LLM Knows What You are Looking for Before Generation22 Apr
- 2024On the Efficacy of Eviction Policy for Key-Value Constrained Generative Language Model Inference9 Feb
2 more
Other references · 29
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,686
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2023Mistral 7Bcited by 3,065
- 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,895
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
- 2023Efficient Streaming Language Models with Attention Sinkscited by 2,115
- 2023LongBench: A Bilingual, Multitask Benchmark for Long Context Understandingcited by 1,378
- 2024RULER: What's the Real Context Size of Your Long-Context Language Models?cited by 1,270
- 2020Constructing A Multi-hop QA Dataset for Comprehensive Evaluation of Reasoning Stepscited by 1,259
- 2021MuSiQue: Multihop Questions via Single-hop Question Compositioncited by 893
- 2024KIVI: A Tuning-Free Asymmetric 2bit Quantization for KV Cachecited by 604
17 more
- 2017The NarrativeQA Reading Comprehension Challengecited by 590
- 2024KVQuant: Towards 10 Million Context Length LLM Inference with KV Cache Quantizationcited by 571
- 2019SAMSum Corpus: A Human-annotated Dialogue Dataset for Abstractive Summarizationcited by 530
- 2021A Dataset of Information-Seeking Questions and Answers Anchored in Research Paperscited by 475
- 2019Multi-News: a Large-Scale Multi-Document Summarization Dataset and Abstractive Hierarchical Modelcited by 431
- 2024MInference 1.0: Accelerating Pre-filling for Long-Context LLMs via Dynamic Sparse Attentioncited by 375
- 2021QMSum: A New Benchmark for Query-based Multi-domain Meeting Summarizationcited by 363
- 2023RepoBench: Benchmarking Repository-Level Code Auto-Completion Systemscited by 346
- 2021Efficient Attentions for Long Document Summarizationcited by 342
- 2024InfLLM: Training-Free Long-Context Extrapolation for LLMs with an Efficient Context Memorycited by 192
- 2024A Survey on Recent Advances in LLM-Based Multi-turn Dialogue Systemscited by 170
- 2024Transformers are Multi-State RNNscited by 120
- 2024RetrievalAttention: Accelerating Long-Context LLM Inference via Vector Retrievalcited by 114
- 2023LongCoder: A Long-Range Pre-trained Language Model for Code Completioncited by 110
- 2024TriForce: Lossless Acceleration of Long Sequence Generation with Hierarchical Speculative Decodingcited by 98
- 2025The Sparse Frontier: Sparse Attention Trade-offs in Transformer LLMscited by 28
- 2024LightTransfer: Your Long-Context LLM is Secretly a Hybrid Model with Effortless Adaptationcited by 11
Built on by
Citing papers per year: 2026: 14.
14 papers.
Sort:
By area:
- 2026Efficient Inference for Large Vision-Language Models: Bottlenecks, Techniques, and Prospectscited by 8
- 2026MomentKV: Closing the Directional Gap in KV Cache Eviction for Long-Context Inferencecited by 5
- 2026Predicting Future Utility: Global Combinatorial Optimization for Task-Agnostic KV Cache Evictioncited by 4
- 2026LKV: End-to-End Learning of Head-wise Budgets and Token Selection for LLM KV Cache Evictioncited by 4
- 2026When Does Value-Aware KV Eviction Help? A Fixed-Contract Diagnostic for Non-Monotone Cache Compressioncited by 4
- 2026HARD-KV: Head-Adaptive Regularization for Decoding-time KV Compressioncited by 2
- 2026KARA: Efficient Reasoning LLM Serving via Sliding-Window KV Cache Compressioncited by 2
- 2026Adaptive Mass-Segmented KV Compression for Long-Context Reasoningcited by 1
- 2026When Replanning Becomes the Bottleneck: Budgeted Replanning for Embodied Agentscited by 1
- 2026Monte Carlo Estimation for KV Cache Evictioncited by 1
- 2026KVSlimmer: Theoretical Insights and Practical Optimizations for Asymmetric KV Mergingcited by 0
- 2026Risk-Controlled KV-Cache Eviction: From Memory Budgets to Risk Targetscited by 0
- 2026VisionWeave: Weaving Elastic Visual Representations as a Native Capability of MLLMs6 Oct 2026
- 2026Monte Carlo Estimation for KV Cache Eviction6 Oct 2026
- 2026Distance-KV: Exploiting Relative Distance for Efficient Long-Context Inference26 Sep 2026
- 2026Risk-Controlled KV-Cache Eviction: From Memory Budgets to Risk Targets23 Sep 2026
- 2026When Replanning Becomes the Bottleneck: Budgeted Replanning for Embodied Agents2 Aug 2026
- 2026HARD-KV: Head-Adaptive Regularization for Decoding-time KV Compression27 Jun 2026
- 2026MomentKV: Closing the Directional Gap in KV Cache Eviction for Long-Context Inference1 Jun 2026
- 2026Adaptive Mass-Segmented KV Compression for Long-Context Reasoning22 May 2026
- 2026When Does Value-Aware KV Eviction Help? A Fixed-Contract Diagnostic for Non-Monotone Cache Compression7 May 2026
- 2026KARA: Efficient Reasoning LLM Serving via Sliding-Window KV Cache Compression1 May 2026
- 2026LKV: End-to-End Learning of Head-wise Budgets and Token Selection for LLM KV Cache Eviction22 Apr 2026
- 2026Efficient Inference for Large Vision-Language Models: Bottlenecks, Techniques, and Prospects7 Apr 2026
- 2026MomentKV: Closing the Directional Gap in KV Cache Eviction for Long-Context Inferencecited by 5
- 2026Predicting Future Utility: Global Combinatorial Optimization for Task-Agnostic KV Cache Evictioncited by 4
- 2026LKV: End-to-End Learning of Head-wise Budgets and Token Selection for LLM KV Cache Evictioncited by 4
- 2026When Does Value-Aware KV Eviction Help? A Fixed-Contract Diagnostic for Non-Monotone Cache Compressioncited by 4
- 2026HARD-KV: Head-Adaptive Regularization for Decoding-time KV Compressioncited by 2
- 2026Monte Carlo Estimation for KV Cache Evictioncited by 1
- 2026Risk-Controlled KV-Cache Eviction: From Memory Budgets to Risk Targetscited by 0
- 2026Distance-KV: Exploiting Relative Distance for Efficient Long-Context Inferencecited by 0
- 2026Monte Carlo Estimation for KV Cache Eviction6 Oct 2026
- 2026Distance-KV: Exploiting Relative Distance for Efficient Long-Context Inference26 Sep 2026
- 2026Risk-Controlled KV-Cache Eviction: From Memory Budgets to Risk Targets23 Sep 2026
- 2026HARD-KV: Head-Adaptive Regularization for Decoding-time KV Compression27 Jun 2026
- 2026MomentKV: Closing the Directional Gap in KV Cache Eviction for Long-Context Inference1 Jun 2026
- 2026When Does Value-Aware KV Eviction Help? A Fixed-Contract Diagnostic for Non-Monotone Cache Compression7 May 2026
- 2026LKV: End-to-End Learning of Head-wise Budgets and Token Selection for LLM KV Cache Eviction22 Apr 2026
- 2026Predicting Future Utility: Global Combinatorial Optimization for Task-Agnostic KV Cache Eviction9 Feb 2026
Similar, not cited
Close in wording, no citation either way
- 2026Coverage-Driven KV Cache Eviction for Efficient and Improved Inference of LLM
- 2026Reformulating KV Cache Eviction Problem for Long-Context LLM Inference
- 2024NACL: A General and Effective KV Cache Eviction Framework for LLMs at Inference Time
- 2026Rethinking KV Cache Eviction via a Unified Information-Theoretic Objective
- 2026KV Cache Optimization Strategies for Scalable and Efficient LLM Inference
- 2026LookaheadKV: Fast and Accurate KV Cache Eviction by Glimpsing into the Future without Generation
- 2026Randomization Boosts KV Caching, Learning Balances Query Load: A Joint Perspective
- 2025Stateful KV Cache Management for LLMs: Balancing Space, Time, Accuracy, and Positional Fidelity