arXiv AI

HaloBlock: Faster and Better Decoding for Diffusion Large Language Models

HaloBlock introduces a difficulty‑aware adaptive halo mechanism for diffusion large language models (dLLMs) to address two inference challenges: decoding difficulty dominated by right‑side tokens and boundary contextual discontinuity in block‑wise decoding. By using bidirectional cumulative entropy to assess block difficulty, HaloBlock allocates halo regions on both sides of each block, smoothing cross‑block context propagation and reducing difficulty for hard tokens. Experiments on LLaDA‑8B‑Instruct, Dream‑7B‑Instruct, and LLaDA‑1.5 show consistent accuracy gains and faster convergence on mathematical reasoning and code generation tasks.

arXiv AI
Jun 15

Residual Context Diffusion Language Models

arXiv:2601. 22954v2 Announce Type: replace-cross Abstract: Diffusion Large Language Models (dLLMs) have emerged as a promising alternative to purely autoregressive language models because they can decode multiple tokens in parallel.

By Yuezhou Hu, Harman Singh, Monishwaran Maheswaran, Haocheng Xi, Coleman Hooper, Jintao Zhang, Aditya Tomar, Michael W. Mahoney, Sewon Min, Mehrdad Farajtabar, Kurt Keutzer, Amir Gholami, Chenfeng Xu
arXiv Computation and Language
Sep 21

Ripple-Pivot Search: Active Parallel Decoding for Diffusion Large Language Models

Ripple-Pivot Search (RPS) is a training‑free decoding method for Diffusion Large Language Models that identifies mid‑entropy pivot positions to reduce uncertainty across remaining masked tokens. By proactively committing these pivots and evaluating token assignments via lookahead, RPS enables more tokens to be unmasked in parallel, speeding up decoding. Experiments on three dLLMs and four reasoning/code‑generation benchmarks show 4–10× wall‑clock speedup over standard decoding, up to 18× with KV caching, while maintaining or improving generation quality.

By Yushi Ye, Xu Chen, Haoyun Jiang, Jinsong Lan, Haihong Tang, Xiangtao Li, Mingming Gong, Ivor Tsang, Yanfeng Wang, Jiangchao Yao
arXiv AI
Jun 2

SimSD: Simple Speculative Decoding in Diffusion Language Models

arXiv:2606. 02544v1 Announce Type: cross Abstract: Diffusion large language models (dLLMs) have recently emerged as a promising alternative to autoregressive (AR) LLMs, offering faster inference through parallel or blockwise decoding.

By Junxia Cui, Haotian Ye, Runchu Tian, Hongcan Guo, Jinya Jiang, Haoru Li, Chaojie Ren, Yiming Huang, Kaijie Zhu, Zhongkai Yu, Kun Zhou, Jingbo Shang
arXiv Machine Learning
2d ago

PBPO: Prefix-Block Policy Optimization for Diffusion Language Models

The paper introduces Prefix-Block Policy Optimization (PBPO), a method that adapts block-level policy optimization to Diffusion Large Language Models (dLLMs). PBPO constructs a prefix-block state and uses a single model evaluation to create a factorized one‑step surrogate for block likelihood, allowing surrogate importance weighting and clipping to be applied separately to each token. Experiments on LLaDA-family models across six benchmarks and three generation lengths show that PBPO achieves competitive performance compared to existing dLLM reinforcement learning methods.

By Katherine Tieu, Dongqi Fu, Si Zhang, Hanqing Zeng, Tianxin Wei, Chenglin Wei, Yinglong Xia, Hong Li, Jingrui He