NestedKV: Nested Memory Routing for Long-Context KV Cache Compression
cs.CL
Submitted: 2026-05-26
Updated: 2026-08-27
Comments: EMNLP 2026 Main
License: http://creativecommons.org/licenses/by/4.0/
The gist: Long-context language models are limited by the memory footprint of the key-value (KV) cache.
Terminology
Abstract
Long-context language models are limited by the memory footprint of the key-value (KV) cache. Existing training-free KV compression methods usually rank tokens by one importance signal -- attention, recency, layer-wise allocation, or key distinctiveness -- which becomes brittle when useful context is globally distinctive, locally episodic, or immediately relevant. We introduce NestedKV, a key-only KV cache compression method inspired by the Continuum Memory System in Nested Learning. NestedKV maintains global, block-level, and sliding-window key anchors, scores tokens by multi-time-scale cosine anomaly, and combines the resulting rankings with a training-free outer learner using head-adaptive mixing and surprise-gated token routing. The score is paired with adaptive per-head budgets and requires no training or LLM modification. Across RULER (4k--32k), LooGLE, LongBench, LongBench-E, InfiniteBench, and MMLU-Pro on Qwen3 and Llama-3.2 models, NestedKV is strongest when the retained cache is small. On Qwen3-4B, it improves over KeyDiff by up to 19.10 points on RULER and 19.29 on LongBench at r=0.75; at r=0.95, it retains 37.32 on LongBench versus 17.55 for KeyDiff.
Sources
- Expected Attention: KV Cache Compression by Estimating Attention from Future Queries Distribution
- The Llama 3 Herd of Models
- RULER: What's the Real Context Size of Your Long-Context Language Models?
- FlowKV: Enhancing Multi-Turn Conversational Coherence in LLMs via Isolated Key-Value Cache Management
- PyramidKV: Dynamic KV Cache Compression based on Pyramidal Information Funneling
- Semantic Integrity Matters: Benchmarking and Preserving High-Density Reasoning in KV Cache Compression
- SONIC: Segmented Optimized Nexus for Information Compression in Key-Value Caching
- Qwen3 Technical Report
- $\infty$Bench: Extending Long Context Evaluation Beyond 100K Tokens
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering