SAGE: Accelerating Vision-Language Models via Entropy-Guided Adaptive Speculative Decoding
On the map: Multimodal LLMs / vision-language models → Visual token pruning and efficiency
Abstract (arXiv)
Speculative decoding has emerged as a promising approach to accelerate inference in vision-language models (VLMs) by enabling parallel verification of multiple draft tokens. However, existing methods rely on static tree structures that remain fixed throughout the decoding process, failing to adapt to the varying prediction difficulty across generation steps. This leads to suboptimal acceptance lengths and limited speedup. In this paper, we propose SAGE, a novel framework that dynamically adjusts the speculation tree structure based on real-time prediction uncertainty. Our key insight is that output entropy serves as a natural confidence indicator with strong temporal correlation across decoding steps. SAGE constructs deeper-narrower trees for high-confidence predictions to maximize speculation depth, and shallower-wider trees for uncertain predictions to diversify exploration. SAGE improves acceptance lengths and achieves faster acceleration compared to static tree baselines. Experiments on multiple benchmarks demonstrate the effectiveness of SAGE: without any loss in output quality, it delivers up to 3.36× decoding speedup for LLaVA-OneVision-72B and 3.18× for Qwen2.5-VL-72B.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Placed in Visual token pruning and efficiency, one of 16 directions in Multimodal LLMs / vision-language models. Builds on 27 of its 31 references that are in the map. First posted on arXiv 31 Jan 2026.
Builds on
27 of 31 references.
Closest ancestors · 2
Same direction, newest first
- 2024FastVLM: Efficient Vision Encoding for Vision Language Models17 Dec
- 2024Fit and Prune: Fast and Training-free Visual Token Pruning for Multi-modal Large Language Models16 Sep
Other references · 25
- 2023BLIP-2: Bootstrapping Language-Image Pre-training with Frozen Image Encoders and Large Language Modelscited by 6,069
- 2025Qwen2.5-VL Technical Reportcited by 5,418
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2019The Curious Case of Neural Text Degenerationcited by 2,881
- 2024LLaVA-OneVision: Easy Visual Task Transfercited by 2,669
- 2025Qwen3-VL Technical Reportcited by 2,534
- 2019Towards VQA Models That Can Readcited by 1,696
- 2022Fast Inference from Transformers via Speculative Decodingcited by 1,615
- 2022ChartQA: A Benchmark for Question Answering about Charts with Visual and Logical Reasoningcited by 1,449
- 2023Video-LLaVA: Learning United Visual Representation by Alignment Before Projectioncited by 1,385
- 2023MVBench: A Comprehensive Multi-modal Video Understanding Benchmarkcited by 1,086
- 2023SEED-Bench: Benchmarking Multimodal LLMs with Generative Comprehensioncited by 929
13 more
- 2023Vision-Language Models for Vision Tasks: A Surveycited by 834
- 2024Medusa: Simple LLM Inference Acceleration Framework with Multiple Decoding Headscited by 821
- 2023Semantic Uncertainty: Linguistic Invariances for Uncertainty Estimation in Natural Language Generationcited by 795
- 2024LongVideoBench: A Benchmark for Long-context Interleaved Video-Language Understandingcited by 644
- 2024EAGLE: Speculative Sampling Requires Rethinking Feature Uncertaintycited by 564
- 2024EAGLE-2: Faster Inference of Language Models with Dynamic Draft Treescited by 362
- 2023SpecInfer: Accelerating Generative Large Language Model Serving with Tree-based Speculative Inference and Verificationcited by 359
- 2022Confident Adaptive Language Modelingcited by 297
- 2022Speculative Decoding: Exploiting Speculative Execution for Accelerating Seq2seq Generationcited by 169
- 2023Speculative Decoding with Big Little Decodercited by 163
- 2023SpecTr: Fast Speculative Decoding via Optimal Transportcited by 129
- 2025SpecVLM: Enhancing Speculative Decoding of Video LLMs via Verifier-Guided Token Pruningcited by 19
- 2025Robust Machine Unlearning for Quantized Neural Networks via Adaptive Gradient Reweighting with Similar Labelscited by 12
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026Does Compression Preserve Uncertainty? A Unified Benchmark for Quantized and Sparse LLMs via Conformal Predictioncited by 2
- 2026Is Multimodal Speculative Decoding Ready for Diffusion-Based Parallel Drafting? A Survey and Empirical Diagnosiscited by 0
- 2026Recovering Off-Policy Supervision for Speculative Decodingcited by 0
- 2026Recovering Off-Policy Supervision for Speculative Decoding30 Sep 2026
- 2026Is Multimodal Speculative Decoding Ready for Diffusion-Based Parallel Drafting? A Survey and Empirical Diagnosis21 Aug 2026
- 2026Does Compression Preserve Uncertainty? A Unified Benchmark for Quantized and Sparse LLMs via Conformal Prediction1 Jun 2026
Similar, not cited
Close in wording, no citation either way
- 2025Spec-LLaVA: Accelerating Vision-Language Models with Dynamic Tree-Based Speculative Decoding
- 2026MMSpec: Benchmarking Speculative Decoding for Vision-Language Models
- 2025SpecVLM: Fast Speculative Decoding in Vision-Language Models
- 2025ViSpec: Accelerating Vision-Language Models with Vision-Aware Speculative Decoding
- 2026Adapting Tree-Structured Speculative Decoding to DeepSeek-V4 for Efficient Inference
- 2025MASSV: Multimodal Adaptation and Self-Data Distillation for Speculative Decoding of Vision-Language Models
- 2025AdaSD: Adaptive Speculative Decoding for Efficient Language Model Inference
- 2025DREAM: Drafting with Refined Target Features and Entropy-Adaptive Cross-Attention Fusion for Multimodal Speculative Decoding