Vegas: Self-Speculative Decoding with Verification-Guided Sparse Attention
On the map: Efficient LLM inference and serving → Speculative decoding
Abstract (arXiv)
Long-context large language model (LLM) inference has become the norm for today's AI applications. However, it is severely bottlenecked by the increasing memory demands of its KV cache. Previous works have shown that self-speculative decoding with sparse attention, where tokens are drafted using a subset of the KV cache and verified in parallel against the full KV cache, speeds up inference in a lossless manner. However, they rely on a standalone KV selection algorithm to select the KV entries used for drafting and overlook the fact that the criticality of each KV entry is inherently computed during verification. In this paper, we propose Vegas, a self-speculative decoding method with verification-guided sparse attention. Vegas identifies critical KV cache entries as a byproduct of verification and computes attention only over these entries when drafting subsequent tokens. This not only improves the draft token acceptance rate but also incurs low KV selection overhead, thereby improving decoding throughput. Vegas achieves a 1.25×-2.81× speedup in decoding throughput over default vLLM and a 1.15×-1.29× speedup over state-of-the-art sparse attention-based self-speculative decoding methods. Our code is available at https://github.com/platformxlab/vegas.
Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Placed in Speculative decoding, one of 12 directions in Efficient LLM inference and serving. Builds on 44 of its 53 references that are in the map. First posted on arXiv 6 Feb 2026.
Builds on
44 of 53 references.
Closest ancestors · 13
Same direction, newest first
- 2025Accelerating Large-Scale Reasoning Model Inference with Sparse Self-Speculative Decoding1 Dec
- 2025SpecExtend: A Drop-in Enhancement for Speculative Decoding of Long Sequences27 May
- 2025EAGLE-3: Scaling up Inference Acceleration of Large Language Models via Training-Time Test3 Mar
- 2025RAPID: Long-Context Inference with Retrieval-Augmented Speculative Decoding27 Feb
- 2025QuantSpec: Self-Speculative Decoding with Hierarchical Quantized KV Cache5 Feb
- 2024SWIFT: On-the-Fly Self-Speculative Decoding for LLM Inference Acceleration9 Oct
- 2024MagicDec: Breaking the Latency-Throughput Tradeoff for Long Context Generation with Speculative Decoding20 Aug
- 2024TriForce: Lossless Acceleration of Long Sequence Generation with Hierarchical Speculative Decoding18 Apr
- 2024Break the Sequential Dependency of LLM Inference Using Lookahead Decoding3 Feb
- 2023DistillSpec: Improving Speculative Decoding via Knowledge Distillation12 Oct
- 2023Accelerating Large Language Model Decoding with Speculative Sampling2 Feb
- 2022Fast Inference from Transformers via Speculative Decoding30 Nov
1 more
Other references · 31
- 2017Attention Is All You Needcited by 60,361
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2025Qwen3 Technical Reportcited by 8,820
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2020Longformer: The Long-Document Transformercited by 3,413
- 2023Efficient Streaming Language Models with Attention Sinkscited by 2,186
- 2024SnapKV: LLM Knows What You are Looking for Before Generationcited by 880
- 2024DistServe: Disaggregating Prefill and Decoding for Goodput-optimized Large Language Model Servingcited by 617
- 2024FlashAttention-3: Fast and Accurate Attention with Asynchrony and Low-precisioncited by 519
- 2024Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inferencecited by 501
- 2023Model Tells You What to Discard: Adaptive KV Cache Compression for LLMscited by 474
19 more
- 2025Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attentioncited by 407
- 2024LongBench v2: Towards Deeper Understanding and Reasoning on Realistic Long-context Multitaskscited by 379
- 2024LayerSkip: Enabling Early Exit Inference and Self-Speculative Decodingcited by 307
- 2024DuoAttention: Efficient Long-Context LLM Inference with Retrieval and Streaming Headscited by 288
- 2025FlashInfer: Efficient and Customizable Attention Engine for LLM Inference Servingcited by 258
- 2024InfLLM: Training-Free Long-Context Extrapolation for LLMs with an Efficient Context Memorycited by 197
- 2024Transformers are Multi-State RNNscited by 120
- 2025A Comprehensive Survey on Long Context Language Modelingcited by 117
- 2025CodeElo: Benchmarking Competition-level Code Generation of LLMs with Human-comparable Elo Ratingscited by 115
- 2024MagicPIG: LSH Sampling for Efficient LLM Generationcited by 111
- 2024PQCache: Product Quantization-based KVCache for Long Context LLM Inferencecited by 95
- 2024ClusterKV: Manipulating LLM KV Cache in Semantic Space for Recallable Compressioncited by 65
- 2024SCBench: A KV Cache-Centric Analysis of Long-Context Methodscited by 60
- 2025Twilight: Adaptive Attention Sparsity with Hierarchical Top-p Pruningcited by 49
- 2024SampleAttention: Near-Lossless Acceleration of Long Context LLM Inference with Adaptive Structured Sparse Attentioncited by 39
- 2024TokenSelect: Efficient Long-Context Inference and Length Extrapolation for LLMs via Dynamic Token-Level KV Cache Selectioncited by 39
- 2025SpecVLM: Enhancing Speculative Decoding of Video LLMs via Verifier-Guided Token Pruningcited by 20
- 2024How Sparse Attention Approximates Exact Attention? Your Attention is Naturally nC-Sparsecited by 17
- 2025RetroInfer: A Vector Storage Engine for Scalable Long-Context LLM Inferencecited by 13
Built on by
Citing papers per year: 2026: 7.
7 papers.
- 2026A Sparse Glimpse of the Whole: Train-Free Self-Speculative Decodingcited by 1
- 2026ASPIRE: Asynchronous Batched Self-Speculative Decoding for Long-Context LLM Inferencecited by 1
- 2026Spectral Origins of the Self-Correction Blind Spot in Autoregressive Generationcited by 0
- 2026Self-Speculation for Faster Reasoning Modelscited by 0
- 2026Where Activation Sparsity and KV-Cache Sparsity Cross in LLM Decodingcited by 0
- 2026S3R: Fast and Faithful RL Rollouts via Self-Sparse Speculative Decodingcited by 0
- 2026FOLD: Position-Free Far Memory with Local Order for Long-Context Speculative Decodingcited by 0
- 2026FOLD: Position-Free Far Memory with Local Order for Long-Context Speculative Decoding14 Oct 2026
- 2026S3R: Fast and Faithful RL Rollouts via Self-Sparse Speculative Decoding9 Oct 2026
- 2026Where Activation Sparsity and KV-Cache Sparsity Cross in LLM Decoding27 Sep 2026
- 2026ASPIRE: Asynchronous Batched Self-Speculative Decoding for Long-Context LLM Inference16 Sep 2026
- 2026A Sparse Glimpse of the Whole: Train-Free Self-Speculative Decoding30 Jul 2026
- 2026Spectral Origins of the Self-Correction Blind Spot in Autoregressive Generation9 Jul 2026
- 2026Self-Speculation for Faster Reasoning Models17 Jun 2026
Similar, not cited
Close in wording, no citation either way
- 2026Dustin: Draft-Augmented Sparse Verification for Efficient Long-Context Generation with Speculative Decoding
- 2025SPIRe: Boosting LLM Inference Throughput with Speculative Decoding
- 2025SpecAttn: Speculating Sparse Attention
- 2026KnapSpec: Self-Speculative Decoding via Adaptive Layer Selection as a Knapsack Problem
- 2023Draft & Verify: Lossless Large Language Model Acceleration via Self-Speculative Decoding
- 2026OasisKV: Scaling In-Decode KV Cache Beyond HBM with Lookahead Sparse Prefetching
- 2026Lynx: Progressive Speculative Quantization for accelerating KV Transfer in Long-Context Inference
- 2026BudgetDraft: Acceptance-Aware Multi-View Training for Sparse-KV Speculative Decoding