Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

DFlash: Block Diffusion for Flash Speculative Decoding

About

Autoregressive large language models (LLMs) deliver strong performance but require inherently sequential decoding, leading to high inference latency and poor GPU utilization. Speculative decoding mitigates this bottleneck by using a fast draft model whose outputs are verified in parallel by the target LLM; however, existing methods still rely on autoregressive drafting, which remains sequential and limits practical speedups. Diffusion LLMs offer a promising alternative by enabling parallel generation, but current diffusion models typically underperform compared with autoregressive models. In this paper, we introduce DFlash, a speculative decoding framework that employs a lightweight block diffusion model for parallel drafting. By generating draft tokens in a single forward pass and conditioning the draft model on context features extracted from the target model, DFlash enables efficient drafting with high-quality outputs and higher acceptance rates. Experiments show that DFlash achieves over 6x lossless acceleration across a range of models and tasks, delivering up to 2.5x higher speedup than the state-of-the-art speculative decoding method EAGLE-3.

Jian Chen, Yesheng Liang, Zhijian Liu• 2026

Related benchmarks

TaskDatasetResultRank
Multi-turn dialogueMT-Bench
Speedup2.85
47
Code GenerationHumanEval
Average Tau (τ)5.52
45
Mathematical ReasoningMATH500
Throughput (tok/s)2.04e+4
20
Code GenerationMBPP
Speedup4.78
12
Code GenerationLiveCodeBench (LCB)
Speedup5.51
12
General PerformanceAggregate Across Math, Code, Chat
Speedup4.91
12
Math ReasoningGSM8K
Speedup5.15
12
Math ReasoningMATH 500
Speedup6.09
12
Math ReasoningAIME 25
Speedup5.68
12
Mathematical ReasoningMATH 500
Speedup4.64
4
Showing 10 of 13 rows

Other info

GitHub

Follow for update