Component-Aware Self-Speculative Decoding in Hybrid Language Models
On the map: Efficient LLM inference and serving → Speculative decoding
Abstract (arXiv)
Speculative decoding accelerates autoregressive inference by drafting candidate tokens with a fast model and verifying them in parallel with the target. Self-speculative methods avoid the need for an external drafter but have been studied exclusively in homogeneous Transformer architectures. We introduce component-aware self-speculative decoding, the first method to exploit the internal architectural heterogeneity of hybrid language models, isolating the SSM/linear-attention subgraph as a zero-cost internal draft. We evaluate this on two architecturally distinct hybrid families: Falcon-H1 (parallel: Mamba-2 + attention per layer) and Qwen3.5 (sequential: interleaved linear and attention layers), with a pure Transformer control (Qwen2.5). Parallel hybrids achieve acceptance rates of alpha = 0.68 at draft length k=2 under greedy decoding, while sequential hybrids yield only alpha = 0.038 -- an 18x gap attributable to how each architecture integrates its components. The property is scale-invariant: Falcon-H1 at 3B reproduces the rates observed at 0.5B. We further show that perplexity degradation from a companion ablation study predicts speculative viability without running speculative decoding: a 3.15x ratio (Falcon) maps to alpha = 0.37 at k=4, while 81.96x (Qwen) maps to alpha = 0.019. For sequential hybrids, generic LayerSkip achieves 12x higher acceptance rates than the component-aware strategy. The composition pattern of hybrid models -- not merely the presence of alternative components -- determines whether component-level self-speculation is viable.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Speculative decoding, one of 12 directions in Efficient LLM inference and serving. Builds on 21 of its 24 references that are in the map. First posted on arXiv 1 May 2026.
Builds on
21 of 24 references.
Closest ancestors · 8
Same direction, newest first
- 2026ConfLayers: Adaptive Confidence-based Layer Skipping for Self-Speculative Decoding16 Apr
- 2026KnapSpec: Self-Speculative Decoding via Adaptive Layer Selection as a Knapsack Problem23 Feb
- 2025Mamba Drafters for Speculative Decoding1 Jun
- 2025CLaSp: In-Context Layer Skip for Self-Speculative Decoding30 May
- 2025STree: Speculative Tree Decoding for Hybrid State-Space Models20 May
- 2024SWIFT: On-the-Fly Self-Speculative Decoding for LLM Inference Acceleration9 Oct
- 2023Accelerating Large Language Model Decoding with Speculative Sampling2 Feb
- 2022Fast Inference from Transformers via Speculative Decoding30 Nov
Other references · 13
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2020Measuring Massive Multitask Language Understandingcited by 7,963
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
- 2016Pointer Sentinel Mixture Modelscited by 3,096
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,336
- 2023Gated Linear Attention Transformers with Hardware-Efficient Trainingcited by 514
- 2024Jamba: A Hybrid Transformer-Mamba Language Modelcited by 401
- 2024LayerSkip: Enabling Early Exit Inference and Self-Speculative Decodingcited by 307
- 2024Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modelingcited by 174
- 2025Falcon-H1: A Family of Hybrid-Head Language Models Redefining Efficiency and Performancecited by 57
- 2025RAD: Redundancy-Aware Distillation for Hybrid Models via Self-Speculative Decodingcited by 8
- 2024Exploring the Limitations of Mamba in COPY and CoT Reasoningcited by 6
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2025Self Speculative Decoding for Diffusion Large Language Models
- 2026SpecBound: Adaptive Bounded Self-Speculation with Layer-wise Confidence Calibration
- 2026LongSpark: Efficient speculative decoding with a fixed-cost parallel drafter
- 2025Pipeline Parallelism is All You Need for Optimized Early-Exit Based Self-Speculative Decoding
- 2025Draft, Verify, and Improve: Toward Training-Aware Speculative Decoding
- 2024Draft on the Fly: Adaptive Self-Speculative Decoding using Cosine Similarity
- 2024Unlocking Efficiency in Large Language Model Inference: A Comprehensive Survey of Speculative Decoding
- 2025Fast Inference via Hierarchical Speculative Decoding