PSD: Pushing the Pareto Frontier of Diffusion LLMs via Parallel Speculative Decoding
For practitioners using diffusion LLMs, PSD offers a training-free method to significantly speed up inference while maintaining generation quality.
Parallel Speculative Decoding (PSD) improves inference efficiency of diffusion LLMs by jointly unmasking multiple tokens per step and collapsing multiple denoising steps into one verification call, achieving up to 5.5× tokens per forward pass with accuracy comparable to greedy decoding.
Diffusion large language models (dLLMs) generate text by iteratively denoising masked token sequences. Although dLLMs can predict all masked positions in parallel within each step, the large number of denoising iterations still makes inference expensive. This cost can be reduced spatially by unmasking multiple tokens per step, or temporally by collapsing multiple denoising steps into one verification call. We propose Parallel Speculative Decoding (PSD), a training-free framework that jointly improves inference along both axes. Using the confidence scores from a single forward pass, PSD selects positions to unmask via a configurable, adaptive unmasking policy and constructs multi-depth speculative drafts without extra model calls. A final batched verification pass then applies hierarchical acceptance, keeping the deepest draft that remains consistent with the updated predictions. Experiments on three dLLMs across reasoning and code generation tasks show that PSD achieves favorable trade-offs between inference efficiency and generation quality, reaching up to $5.5\times$ tokens per forward pass with accuracy comparable to greedy decoding.