HARD-KV: Head-Adaptive Regularization for Decoding-time KV Compression
On the map: Efficient LLM inference and serving → KV cache optimization
Abstract (arXiv)
Long-context LLM inference faces a fundamental conflict: head-adaptive compression algorithms (e.g., Top-p nucleus sampling) offer superior accuracy by dynamically fluctuating memory budgets, yet modern inference engines (e.g., vLLM) demand rigid, static memory patterns to leverage CUDA Graphs and PagedAttention. We resolve this ``Static-Dynamic'' mismatch with HARD-KV, a unified framework that that bridges dynamic selection with rigid system constraints. HARD-KV introduces a Cascade Cache hierarchy, managing the token lifecycle across dense, sparse, and condensed tiers. Crucially, we propose a Logits Calibration mechanism that normalizes diverse importance metrics into a unified probability space, enabling consistent Top-p budgeting across heterogeneous heads. To bridge the efficiency gap, we offer a system-level solution, which rewrites fragmented, dynamic indices into contiguous physical layouts compatible with high-performance inference engine. Extensive experiments on math-reasoning benchmarks (AIME, U-Math) verify that HARD-KV achieves up to 2× throughput improvement over static baselines while maintaining high-fidelity generation in 10k+ token scenarios. Code is available at https://github.com/SuDIS-ZJU/HARDInfer.
Key facts. Cited by 2 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 38 of its 48 references that are in the map. First posted on arXiv 27 Jun 2026.
Builds on
38 of 48 references.
Closest ancestors · 15
Same direction, newest first
- 2025G-KV: Decoding-Time KV Cache Eviction with Global Attention29 Nov
- 2025Taming the Fragility of KV Cache Eviction in LLM Inference15 Oct
- 2025The Pitfalls of KV Cache Compression30 Sep
- 2025KeyDiff: Key Similarity-Based KV Cache Eviction for Long-Context LLM Inference in Resource-Constrained Environments21 Apr
- 2025RocketKV: Accelerating Long-Context LLM Inference via Two-Stage KV Cache Compression19 Feb
- 2024SCOPE: Optimizing Key-Value Cache Compression in Long-context Generation18 Dec
- 2024DiffKV: Differentiated Memory Management for Large Language Models with Parallel KV Compaction4 Dec
- 2024Not All Heads Matter: A Head-Level KV Cache Compression Method with Integrated Retrieval and Reasoning25 Oct
- 2024DuoAttention: Efficient Long-Context LLM Inference with Retrieval and Streaming Heads14 Oct
- 2024NACL: A General and Effective KV Cache Eviction Framework for LLMs at Inference Time7 Aug
- 2024ThinK: Thinner Key Cache by Query-Driven Pruning30 Jul
- 2024RazorAttention: Efficient KV Cache Compression Through Retrieval Heads22 Jul
3 more
Other references · 23
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2025Qwen3 Technical Reportcited by 8,195
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2019The Curious Case of Neural Text Degenerationcited by 2,881
- 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
- 2023SGLang: Efficient Execution of Structured Language Model Programscited by 1,107
- 2016An overview of gradient descent optimization algorithmscited by 1,021
- 2024Medusa: Simple LLM Inference Acceleration Framework with Multiple Decoding Headscited by 821
- 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 373
11 more
- 2024LongBench v2: Towards Deeper Understanding and Reasoning on Realistic Long-context Multitaskscited by 367
- 2024MagicPIG: LSH Sampling for Efficient LLM Generationcited by 112
- 2025Twilight: Adaptive Attention Sparsity with Hierarchical Top-p Pruningcited by 48
- 2025StreamingTOM: Streaming Token Compression for Efficient Video Understandingcited by 41
- 2025A Systematic Analysis of Hybrid Linear Attentioncited by 38
- 2025Accelerating Streaming Video Large Language Models via Hierarchical Token Compressioncited by 34
- 2024U-MATH: A University-Level Benchmark for Evaluating Mathematical Skills in LLMscited by 24
- 2025Sliding Window Attention Training for Efficient Large Language Modelscited by 21
- 2025LazyEviction: Lagged KV Eviction with Attention Pattern Observation for Efficient Long Reasoningcited by 17
- 2025Which Heads Matter for Reasoning? RL-Guided KV Cache Compressioncited by 10
- 2025Lethe: Layer- and Time-Adaptive KV Cache Pruning for Reasoning-Intensive LLM Servingcited by 5
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2025KVCompose: Efficient Structured KV Cache Compression with Composite Tokens
- 2026Adaptive Mass-Segmented KV Compression for Long-Context Reasoning
- 2026KVSculpt: KV Cache Compression as Distillation
- 2026From Rigid to Dynamic: Entropy-Guided Adaptive Inference for Long-Context LLMs
- 2026Tailoring the Quantization Space for 1-Bit KV Cache Compression
- 2026MetaKV: Adaptive KV Cache Compression for Constrained LLM Inference
- 2025ReFreeKV: Towards Threshold-Free KV Cache Compression
- 2024ZigZagkv: Dynamic KV Cache Compression for Long-context Modeling based on Layer Uncertainty