arXiv Machine Learning
Sep 16

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

The paper introduces Window-Diffusion, a method that accelerates diffusion language model inference by pruning and caching tokens within a sliding window. It categorizes undecoded tokens into active, buffer, and far-field groups, computing only the first two while discarding the rest. Experiments on LLaDA and Dream demonstrate up to 99× speedup with minimal loss in generation quality.

By Fengrui Zuo, Zhiwei Ke, Yiming Liu, Wenqi Lou, Chao Wang, Xuehai Zhou
arXiv AI
Oct 9

DynaTE: Accelerating Diffusion LLMs via Dynamic Token Execution

DynaTE is a hardware–software co‑design that accelerates diffusion‑based large language models (dLLMs) by dynamically adapting execution to token states. It skips low‑utility token computations, uses a dimension‑reconfigurable processing array, refines locally dependent tokens with FLDD, and streams vocabulary to handle irregular token patterns. These techniques yield 2.05–2.78× speedup and 2.99–3.93× higher energy efficiency compared to existing dLLM accelerators, and outperform Jetson AGX Orin by 2.55× speedup and 6.07× energy efficiency.

By Minghan Jiang, Jiayi Wang, Shuaiting Li, Haibin Shen, Kejie Huang
arXiv Machine Learning
Oct 12

Rethinking the Roles of Diffusion and Decoding: Predictive-State Flows for Parallel Language Generation

The paper proposes Predictive-State Flows (PSF), a new architecture for continuous diffusion language models that separates the diffusion prior’s role to generate high‑level predictive states and the decoder’s role to resolve token‑level choices. PSF partitions sequences into blocks, uses a rectified‑flow prior to produce predictive states, and then a block‑causal decoder generates all blocks in parallel, reducing token‑level sequential depth. Experiments on OpenWebText show that a 197M‑parameter PSF model with eight predictive states achieves a generative perplexity of 22.8 and a MAUVE@512 of 0.93 while using far fewer FLOPs than a comparable ELF‑B model.

By Jinjin He, Xinsong Feng, Lucheng Fu, Sinan Wang, Bo Zhu, Jiayun Wang
arXiv Computation and Language
Aug 25

Accelerating Diffusion Language Models via Structured Suffix Modeling

The paper introduces a training‑free structured suffix modeling technique to accelerate Diffusion Language Models (DLMs). It partitions the suffix into local, middle, and tail regions, retaining varying numbers of tokens per region and incorporating previous decoding results into current token representations. Experiments on three DLMs show significant speedups—up to 72.81× in long‑sequence inference—while often improving performance, and the method is compatible with existing acceleration strategies.

By Zifeng Cheng, Keda Li, Zhiwei Jiang, Cong Wang, Fei Shen, Qing Gu