Model Releases
ALoDLM: Adaptively Looped Diffusion Language Models
arXiv:2610.04198v1 Announce Type: cross Abstract: Diffusion language models (DLMs) enable fast generation by predicting multiple tokens in parallel, but their practical adoption remains limited by a p
arXiv:2610.04198v1 Announce Type: cross Abstract: Diffusion language models (DLMs) enable fast generation by predicting multiple tokens in parallel, but their practical adoption remains limited by a persistent quality gap relative to comparably sized autoregressive (AR) models. We attribute this gap to a computation-difficulty mismatch: within a partially observed sequence, some unknown tokens are easy to predict, while others require substantially more computation. Existing DLMs nevertheless apply uniform computational depth to all unknown positions at each denoising step. We introduce ALoDLM, which replaces uniform computation with token-adaptive latent recurrence. At each denoising step, ALoDLM iteratively refines latent representations and allocates computation according to token difficulty. Tokens ready to commit are fed back as discrete context, while unresolved tokens retain and further refine their latent states through additional recurrent passes. To learn token prediction and computation allocation jointly, we formulate token-wise computation schedules as latent variables and derive a conditional negative evidence lower bound (NELBO). We train ALoDLM at 1.7B and 8B parameter scales. Across eleven benchmarks, ALoDLM outperforms all evaluated DLMs and the corresponding AR baselines in average benchmark score at both scales. ALoDLM also retains fast parallel decoding, yielding a strong quality-efficiency trade-off among evaluated autoregressive and diffusion models under optimized inference engines.
Related
- Visual-Redundancy-Controlled Parallel Decoding for Diffusion-Based Multimodal Large Language Models
- Beyond the Bidirectional Promise: Re-evaluating the Robustness of Diffusion Language Models
- Rethinking Soft Tokens for Parallel Decoding in Diffusion Language Models
- Prefix-Denoising Consistency: Test-Time Verification for Diffusion Language Models
- Retrofitting Linear Attention into Diffusion Language Models
Source: arXiv cs.LG | 2026-10-06