Autonomy-of-Heads: Data-Free Sparse Attention from Frozen Query-Key Geometry
On the map: Efficient LLM inference and serving → KV cache optimization
Abstract (arXiv)
Long-context LLM inference is bottlenecked by quadratic attention computation and growing KV-cache costs. Existing sparse attention and KV-compression methods typically decide which tokens or heads to preserve from runtime attention scores, observation windows, calibration prompts, or learned gates, making head diagnosis input-dependent and costly to deploy. We propose Autonomy-of-Heads (AoH), a data-free method that identifies retrieval and streaming heads from the spectral geometry of query-key projections. AoH defines the kernel attention operator Mh = WKhtopWQh and uses its effective-rank as a weight-space measure of head function: concentrated spectra indicate a small number of dominant query-key matching directions and are associated with retrieval heads, whereas diffuse spectra indicate the absence of a dominant global matching direction and are associated with streaming heads. We further derive an efficient dhead-dimensional computation that avoids constructing the full dmodel× dmodel matrix. We conducted extensive experiments across models demonstrating that at 50% sparsity, AoH retains 96.5% of Full Attention performance on average while reducing prefill and decode latency by up to 41.4% and 66.0%, respectively, and KV-cache memory by 50.0% at 256K tokens.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in KV cache optimization, one of 12 directions in Efficient LLM inference and serving. Builds on 23 of its 34 references that are in the map. First posted on arXiv 7 Aug 2026.
Builds on
23 of 34 references.
Closest ancestors · 7
Same direction, newest first
- 2026LycheeDecode: Accelerating Long-Context LLM Inference via Hybrid-Head Sparse Decoding4 Feb
- 2024NACL: A General and Effective KV Cache Eviction Framework for LLMs at Inference Time7 Aug
- 2024RazorAttention: Efficient KV Cache Compression Through Retrieval Heads22 Jul
- 2024Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference16 Jun
- 2024Reducing Transformer Key-Value Cache Size with Cross-Layer Attention21 May
- 2024SnapKV: LLM Knows What You are Looking for Before Generation22 Apr
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttention12 Sep
Other references · 16
- 2017Attention Is All You Needcited by 59,932
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
- 2020Longformer: The Long-Document Transformercited by 3,362
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
- 2023Efficient Streaming Language Models with Attention Sinkscited by 2,115
- 2020Big Bird: Transformers for Longer Sequencescited by 1,673
- 2019Root Mean Square Layer Normalizationcited by 1,425
- 2023LongBench: A Bilingual, Multitask Benchmark for Long Context Understandingcited by 1,378
- 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,311
- 2023SARATHI: Efficient LLM Inference by Piggybacking Decodes with Chunked Prefillscited by 187
4 more
- 2024You Only Cache Once: Decoder-Decoder Architectures for Language Modelscited by 135
- 2026IndexCache: Accelerating Sparse Attention via Cross-Layer Index Reusecited by 27
- 2025Kascade: A Practical Sparse Attention Method for Long-Context LLM Inferencecited by 9
- 2026Linear Predictability of Attention Heads in Large Language Modelscited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Sparse Attention as a Range Searching Problem: Towards an Inference-Efficient Index for KV Cache
- 2025Tactic: Adaptive Sparse Attention with Clustering and Distribution Fitting for Long-Context LLMs
- 2026Flux Attention: Context-Aware Hybrid Attention for Efficient LLMs Inference
- 2026From Rigid to Dynamic: Entropy-Guided Adaptive Inference for Long-Context LLMs
- 2024Loki: Low-rank Keys for Efficient Sparse Attention
- 2026HeadRank: Decoding-Free Passage Reranking via Preference-Aligned Attention Heads
- 2026Self-Indexing Attention for Compression-Compatible Sparse Long-Context LLM Inference
- 2026ATFlash: Per-RoPE-Wavelength Attention Windows for Compute/Memory-Efficient LLM Inference