arXiv AI By Zhaohui Zheng, Chenhang He, Jinrui Zhang, Ming-Ming Cheng, Lei Zhang

HaloBlock: Faster and Better Decoding for Diffusion Large Language Models

Read the original on arXiv AI →

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.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv AI.

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