cs.CLNov 6, 2026

DUET: Dual-Sparse Inference for Block Diffusion Language Models

Authors: Siyuan Yan, Shen Gu, Xiaoxing Ma, Chun Cao, Jingwei Xu

Organizations: State Key Laboratory for Novel Software Technology, Nanjing University, China

Abstract

Diffusion language models (DLMs) enable parallel token generation through iterative denoising, yet their inference remains costly due to repeated attention over a growing historical prefix and repeated processing of the entire current generation block. Historical attention becomes increasingly expensive as generation progresses, while all current-block positions are propagated through the Transformer at every denoising step even though only a small fraction are typically resolved at that step. In this paper, we introduce Duet, a training-free dual-sparse inference framework that addresses these challenges through prefix Key-Value (KV) sparsity and early query sparsity, respectively. Specifically, Duet exploits the concentration and cross-step stability of historical attention within each generation block to construct a compact historical KV subset that is reused across denoising steps. For query sparsity, Duet predicts update positions from shallow representations and propagates only the corresponding queries through the suffix Transformer layers, avoiding deep computation at positions unlikely to be updated in the current step. Experiments show that Duet maintains competitive accuracy while achieving 1.34×\times/2.83×\times end-to-end/per-step speedups on LLaDA2.1-mini and 2.41×\times/4.72×\times on SDAR-8B-Chat-b32 at a 32K context length.

Figures & tables

Appendix figures & tables2 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Retrofitting Linear Attention into Diffusion Language Models

    Aug 6, 2026Jinha Kim, Younghun Roh, Jaeyeon KimBlockwise Diffusion Language ModelsDiffusion Language Model Inference

  2. Window-Diffusion: Accelerating Diffusion Language Model Inference with Windowed Token Pruning and Caching

    Jan 28, 2026Fengrui Zuo, Zhiwei Ke, Yiming Liu +3Diffusion Language Model InferenceToken Pruning