Full Attention Strikes Back: Transferring Full Attention into Sparse within Hundred Training Steps
On the map: Efficient LLM inference and serving → KV cache optimization
Abstract (arXiv)
Long-context inference in large language models is bottlenecked by the quadratic cost of full attention. Existing efficient alternatives often rely either on native sparse training or on heuristic token eviction, creating an undesirable trade-off among efficiency, training cost, and accuracy. In this work, we show that full-attention LLMs are already intrinsically sparse and can be transformed into highly sparse models with only minimal adaptation. Our approach is built on three observations: (1) only a small subset of attention heads truly requires full long-context processing; (2) long-range retrieval is governed primarily by a low-dimensional subspace, allowing relevant tokens to be retrieved efficiently with a 16-dimensional indexer; and (3) the useful token budget is strongly query-dependent, making dynamic top-p selection more suitable than fixed top-k sparsification. Based on these insights, we propose RTPurbo, which retains the full KV cache only for retrieval heads and introduces a lightweight token indexer for sparse attention. By exploiting the model's intrinsic sparsity, RTPurbo achieves sparsification with only a few hundred training steps. Experiments on long-context benchmarks and reasoning tasks show that RTPurbo preserves near-lossless accuracy while delivering substantial efficiency gains, including up to a 9.36× prefill speedup at 1M context and about a 2.01× decode speedup. These results suggest that strong sparse inference can be obtained from standard full-attention training without expensive native sparse pretraining.
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Placed in KV cache optimization, one of 12 directions in Efficient LLM inference and serving. Builds on 21 of its 23 references that are in the map. First posted on arXiv 16 May 2026.
Builds on
21 of 23 references.
Closest ancestors · 5
Same direction, newest first
- 2026FASA: Frequency-aware Sparse Attention3 Feb
- 2024DuoAttention: Efficient Long-Context LLM Inference with Retrieval and Streaming Heads14 Oct
- 2024RazorAttention: Efficient KV Cache Compression Through Retrieval Heads22 Jul
- 2024Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference16 Jun
- 2024SnapKV: LLM Knows What You are Looking for Before Generation22 Apr
Other references · 16
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,796
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
- 2023Efficient Streaming Language Models with Attention Sinkscited by 2,186
- 2024MMLU-Pro: A More Robust and Challenging Multi-Task Language Understanding Benchmarkcited by 1,848
- 2023LongBench: A Bilingual, Multitask Benchmark for Long Context Understandingcited by 1,417
- 2024RULER: What's the Real Context Size of Your Long-Context Language Models?cited by 1,346
- 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,160
- 2024MInference 1.0: Accelerating Pre-filling for Long-Context LLMs via Dynamic Sparse Attentioncited by 383
- 2025MathArena: Evaluating LLMs on Uncontaminated Math Competitionscited by 297
- 2024Patchscopes: A Unifying Framework for Inspecting Hidden Representations of Language Modelscited by 235
- 2025MoBA: Mixture of Block Attention for Long-Context LLMscited by 198
4 more
- 2025FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inferencecited by 123
- 2025SpargeAttention: Accurate and Training-free Sparse Attention Accelerating Any Model Inferencecited by 120
- 2025The emergence of sparse attention: impact of data distribution and benefits of repetitioncited by 25
- 2025BLASST: Dynamic BLocked Attention Sparsity via Softmax Thresholdingcited by 13
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Text Template Tokens Are Implicit Semantic Registers in Diffusion Transformerscited by 4
- 2026CoSA: Accelerating Long-Context Inference via Proxy-Kernel Co-Designed Sparse Attentioncited by 1
- 2026SCOPE: Subspace Clustering with Online Per-Head Top-K Estimation for Sparse Video Attentioncited by 0
- 2026Competition Domains Matter: Selective Head Globalization for Training-Free Sparse Attention in Video Diffusion Transformerscited by 0
- 2026Competition Domains Matter: Selective Head Globalization for Training-Free Sparse Attention in Video Diffusion Transformers10 Oct 2026
- 2026SCOPE: Subspace Clustering with Online Per-Head Top-K Estimation for Sparse Video Attention13 Aug 2026
- 2026CoSA: Accelerating Long-Context Inference via Proxy-Kernel Co-Designed Sparse Attention28 Jul 2026
- 2026Text Template Tokens Are Implicit Semantic Registers in Diffusion Transformers21 Jul 2026
Similar, not cited
Close in wording, no citation either way
- 2025Training-free Context-adaptive Attention for Efficient Long Context Modeling
- 2026Double-P: Hierarchical Top-P Sparse Attention for Long-Context LLMs
- 2025Adamas: Hadamard Sparse Attention for Efficient Long-Context Inference
- 2025SALE : Low-bit Estimation for Efficient Sparse Attention in Long-context LLM Prefilling
- 2025Every Token Counts: Generalizing 16M Ultra-Long Context in Large Language Models
- 2026Simplified Sparse Attention via Gist Tokens
- 2025Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attention
- 2026AB-Sparse: Sparse Attention with Adaptive Block Size for Accurate and Efficient Long-Context Inference