BitNest: Bit-Nested Speculative Decoding for Memory-Efficient LLM Inference Acceleration
cs.AI
Submitted: 2026-10-02
Updated: 2026-10-06
License: http://creativecommons.org/licenses/by/4.0/
The gist: Speculative decoding accelerates autoregressive generation by using a lightweight draft to propose multiple tokens for parallel verification.
Terminology
Abstract
Speculative decoding accelerates autoregressive generation by using a lightweight draft to propose multiple tokens for parallel verification. However, existing methods often require an additional draft model or weight representation, introducing non-negligible memory overhead on resource-constrained devices. Self-speculative approaches reduce this overhead, yet still face trade-offs between draft quality, target quality, and storage efficiency. We propose BitNest, a bit-nested speculative decoding framework that embeds a low-precision draft directly into the higher-precision target representation. Instead of deriving a draft from a predefined target, BitNest first constructs a strong low-precision base and then recovers the higher-precision target through residual refinement, enabling both models to share a single physical weight representation. BitNest further extends this progressive-precision design to the KV cache for long-context inference. Across multiple 7B--8B edge-friendly LLMs and diverse workloads, BitNest achieves an average speculative acceptance rate of 95.2% while closely preserving higher-precision model quality, and delivers 1.48--1.61x end-to-end speedup over FP16 autoregressive decoding. On the LLaMA models supported by all representative self-speculative baselines, BitNest also achieves consistently competitive or higher decoding speedup.
Sources
- QuaRot: Outlier-Free 4-Bit Inference in Rotated LLMs
- Program Synthesis with Large Language Models
- LongBench: A Bilingual, Multitask Benchmark for Long Context Understanding
- Medusa: Simple LLM Inference Acceleration Framework with Multiple Decoding Heads
- Accelerating Large Language Model Decoding with Speculative Sampling
- DFlash: Block Diffusion for Flash Speculative Decoding
- Evaluating Large Language Models Trained on Code
- The Llama 3 Herd of Models
- Efficient Attentions for Long Document Summarization
- Fine-Tuning Vision-Language-Action Models: Optimizing Speed and Success
- Fast Inference from Transformers via Speculative Decoding
- EAGLE-2: Faster Inference of Language Models with Dynamic Draft Trees
- EAGLE: Speculative Sampling Requires Rethinking Feature Uncertainty
- DuQuant: Distributing Outliers via Dual Transformation Makes Stronger Quantized LLMs
- AWQ: Activation-aware Weight Quantization for LLM Compression and Acceleration
- TiDAR: Think in Diffusion, Talk in Autoregression
- SpinQuant: LLM quantization with learned rotations
- Your LLM Knows the Future: Uncovering Its Multi-Token Prediction Potential
- SpecDiff-2: Scaling Diffusion Drafter Alignment For Faster Speculative Decoding
- FlatQuant: Flatness Matters for LLM Quantization
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- Agents' Last Exam
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems