cs.LGApr 6, 2026

DualDiffusion: A Speculative Decoding Strategy for Masked Diffusion Models

Authors: Satyam Goyal, Kushal Patel, Tanush Mittal, Arjun Laxman

Organizations: Department of Computer Science and Engineering, University of Michigan, Ann Arbor, MI, USA.

Abstract

Masked Diffusion Models (MDMs) offer a promising alternative to autoregressive language models by enabling parallel token generation and bidirectional context modeling. However, their inference speed is significantly limited by the inability to cache key-value pairs due to bidirectional attention, requiring O(N2)O(N^2) computations at each generation step. While recent methods like FastDLLM and DkvCache improve inference speed through attention approximations and caching strategies, they achieve speedups at the cost of generation quality. We propose DualDiffusion, a speculative decoding framework for MDMs that combines fast drafter models (using efficient approximations) with slower, more accurate verifier models. By running multiple steps of a lightweight drafter followed by a single verification step, DualDiffusion achieves a superior Pareto frontier between generation steps and accuracy compared to existing approaches. We evaluate our method on MMLU and GSM8K, demonstrating that DualDiffusion maintains high accuracy while reducing the number of generation steps required, effectively pushing the quality-efficiency trade-off curve for masked diffusion language models.

Figures & tables

Explore similar work

CardsList
  1. Understanding and Accelerating the Training of Masked Diffusion Language Models

    May 13, 2026Chunsan Hong, Sanghyun Lee, Chieh-Hsin Lai +5Masked Diffusion ModelsDiffusion Language Model Training

  2. Edit-Based Refinement for Parallel Masked Diffusion Language Models

    May 10, 2026Houxing Ren, Mingjie Zhan, Zimu Lu +5Masked Diffusion ModelsParallel Decoding

  3. SimSD: Simple Speculative Decoding in Diffusion Language Models

    Jun 1, 2026Junxia Cui, Haotian Ye, Runchu Tian +9Masked Diffusion ModelsSpeculative Decoding for Diffusion Language Models