AIAny
Icon for item

Random Attention: Rethinking KV Cache Eviction for Efficient Reasoning

Compresses KV cache for long-chain reasoning by keeping prompt tokens and evicting remaining entries uniformly at random per attention head; across four models and six reasoning tasks it matches the strongest prior evictor while delivering 32–43% higher vLLM throughput. Relies on prompt protection and redundancy across heads/text to retain reasoning traces; suitable when static memory budgets and higher serving throughput are priorities.

Introduction

Long-chain chains-of-thought create a KV-cache memory bottleneck for LLM inference; counterintuitively, the paper shows that most selection signals add little value once the prompt is preserved. The core insight is that protecting the prompt plus relying on redundancy across attention heads and text restatement lets a head-wise uniform random eviction retain enough of the reasoning trace to match sophisticated selectors while improving serving throughput.

Key Findings
  • Random per-head eviction (no scoring) matches the strongest prior deterministic evictor on average across four models and six reasoning tasks, contradicting the assumption that fine-grained importance scoring is necessary for reasoning accuracy. This parity holds when the prompt is preserved.
  • Throughput gains of 32–43% in vLLM deployment are reported versus the strongest prior evictor, because random eviction avoids scoring overhead and simplifies memory movement.
  • Two redundancy mechanisms explain robustness: the model restates what it needs in the token stream (textual redundancy), and multiple attention heads independently preserve copies of the reasoning trace (cross-head redundancy). Together these make random retention effective without per-token importance estimates.
  • Empirical ablations show the prompt is the fragile part of the cache; many selector differences trace to whether they incidentally preserved prompt content rather than to superior scoring.
Who It's For and Tradeoffs

Great fit if you deploy LLMs for long-context reasoning under strict static memory budgets and want higher inference throughput with minimal engineering: Random Attention is simple to implement, removes scoring computation, and scales well in vLLM-like serving.
Look elsewhere if your workload critically depends on preserving very specific, non-redundant token signals that do not appear repeatedly in text or across heads, or if you need formal guarantees about worst-case attention error—randomized eviction trades deterministic selection for empirical robustness and throughput.

Where It Fits

This paper reframes cache-compression alternatives: selection-based sparse attention keeps the whole KV cache and reduces compute but not memory; deterministic eviction scores and prunes but can be brittle; Random Attention shows a low-cost, high-throughput eviction recipe that attains similar average accuracy in reasoning tasks by leveraging model-level redundancy rather than per-token importance scoring. Code is published by the authors for reproducibility and deployment tests.

Information

  • Websitearxiv.org
  • OrganizationsSalesforce AI Research, University of Illinois Urbana-Champaign
  • AuthorsHeng Wang, Jielin Qiu, Wenting Zhao, Cheng Qian, Liangwei Yang, Jiawei Han, Heng Ji, Silvio Savarese, Shelby Heinecke, Huan Wang
  • Published date2026/09/03

More Items

Studies on-policy distillation (OPD) at the data-minimal limit by training on a single query, measuring state coverage and alignment dynamics, and showing OPD is often data-overfed but algorithm-starved.

Shows that fully quantizing all 496 linear layers—including the recurrent Gated DeltaNet—of a hybrid 27B LLM to 4-bit NVFP4 W4A4 preserves benchmark accuracy while reducing model size to 17.5 GiB and improving prefill speed; includes a calibrated NVFP4 checkpoint.

Compresses conversational histories and long documents into short sequences of continuous soft memory tokens that a frozen decoder can read directly without text reconstruction. Uses a small reader-matched writer that trains only a tiny adapter, achieving 4–16× compression and much faster write/read latencies.