Search papers, labs, and topics across Lattice.
This paper introduces Dynamic Verification Relaxed Speculative Decoding (DIVERSED), a novel approach to speculative decoding that relaxes the strict verification step by learning an ensemble-based verifier to blend draft and target model distributions. By dynamically adjusting the verification stringency based on context and task, DIVERSED significantly increases the acceptance rate of drafted tokens, leading to improved inference efficiency. Experiments demonstrate that DIVERSED achieves substantially higher inference speedup compared to standard speculative decoding methods while maintaining generation quality.
Speculative decoding's speed boost just got a whole lot bigger: DIVERSED dynamically loosens the verification constraints, letting more good tokens through and accelerating inference.
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.
Amazon Science