Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Making Every Verified Token Count: Adaptive Verification for MoE Speculative Decoding

About

Tree-based speculative decoding accelerates autoregressive generation by verifying multiple draft candidates in parallel, but this advantage weakens for sparse Mixture-of-Experts (MoE) models. As the draft tree grows, different branches activate different experts, expanding the union of activated experts and substantially increasing target-side verification cost. We propose EVICT, a training-free, hyperparameter-free, and lossless adaptive verification method for MoE speculative decoding. EVICT makes every verified token count by truncating the draft tree before target verification and retaining only the cost-effective prefix. It leverages fine-grained drafter signals to estimate candidate benefit, combines them with offline-profiled verification cost, and remains highly compatible with the high-performance graph-based serving framework SGLang. Extensive experiments on diverse MoE backbones and benchmarks show that EVICT achieves up to 2.35x speedup over autoregressive decoding and an average 1.21x speedup over the state-of-the-art baseline EAGLE-3, while significantly reducing unnecessary expert activations during verification.

Lehan Pan, Ziyang Tao, Ruoyu Pang, Xiao Wang, Jianjun Zhao, Yanyong Zhang• 2026

Related benchmarks

TaskDatasetResultRank
Instruction FollowingAlpaca--
173
Question AnsweringQA--
47
SummarizationCNN/DM--
32
Code GenerationHumanEval
TPS (Tokens/s)351.1
25
Multi-turn dialogueMT-Bench
Tokens/s271
20
Text GenerationAlpaca--
15
Text GenerationMT-Bench--
14
Code GenerationHumanEval
Decoding Speed (tokens/s)265.5
10
Mathematical ReasoningGSM8K
Decoding Speed (tokens/s)258.4
10
Question AnsweringNatural Questions QA
Decoding Speed (tokens/s)162.5
10
Showing 10 of 15 rows

Other info

Follow for update