Boost Inference Performance up to 15x on NVIDIA Blackwell Using DFlash Speculative Decoding
DFlash 是一个开源 block-diffusion 草稿模型:草稿器用一次前向就生成一整块候选 token(而非逐 token),target 模型并行验证。三个关键技术——并行 block-diffusion 草稿、来自大模型的 target hidden-state 条件、KV 注入维持接受率。在 8 卡 NVIDIA DGX B300 上跑 gpt-oss-120b、高交互场景 (每用户 500–600 tokens/sec) 时,相比标准自回归解码吞吐提升最高 15x,比 EAGLE-3 投机解码高 1.5x;较小模型增益温和:Gemma 4 31B 提升 5.8x、Qwen3 8B 提升 5.1x。
INSIGHT投机解码的瓶颈在草稿阶段仍是串行;DFlash 用 block-diffusion 把草稿也并行化,正好吃满 Blackwell 双 die、640 Tensor Core 暴露出来的并行算力。如果社区搜索用大 LLM 做 query 理解或重排,DFlash 这类把'起草+验证'都并行的方案能把解码瓶颈搬走,且开源、跑在 TensorRT 栈上,迁移门槛低。注意 15x 是 120B 大模型高并发下的极值,8B 量级只有约 5x,选型要按自己的模型规模估收益。