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

DIVERSED: Relaxed Speculative Decoding via Dynamic Ensemble Verification

About

Speculative decoding is an effective technique for accelerating large language model inference by drafting multiple tokens in parallel. In practice, its speedup is often bottlenecked by a rigid verification step that strictly enforces the accepted token distribution to exactly match the target model. This constraint leads to the rejection of many plausible tokens, lowering the acceptance rate and limiting overall time speedup. To overcome this limitation, we propose Dynamic Verification Relaxed Speculative Decoding (DIVERSED), a relaxed verification framework that improves time efficiency while preserving generation quality. DIVERSED learns an ensemble-based verifier that blends the draft and target model distributions with a task-dependent and context-dependent weight. We provide theoretical justification for our approach and demonstrate empirically that DIVERSED achieves substantially higher inference efficiency compared to standard speculative decoding methods. Code is available at: https://github.com/comeusr/diversed.

Ziyi Wang, Siva Rajesh Kasa, Ankith M S, Santhosh Kumar Kasa, Jiaru Zou, Sumit Negi, Ruqi Zhang, Nan Jiang, Qifan Song• 2026

Related benchmarks

TaskDatasetResultRank
Code GenerationMBPP (test)
Pass@159
298
SummarizationXSum (test)
ROUGE-25.01
246
Mathematical ReasoningGSM8K
Accuracy92
66
SummarizationCNNDM
ROUGE-212.64
62
SummarizationXsum
ROUGE-27.66
46
Text SummarizationXsum
ROUGE-27.22
10
Code GenerationMBPP
pass@162
6
Mathematical ReasoningGSM8K (test)
Accuracy88
6
Mathematical ReasoningGSM8K
Accuracy76
6
Abstractive SummarizationXsum
ROUGE-27.84
6
Showing 10 of 10 rows

Other info

Follow for update