mlsanityData to 7 Oct 2026

Decoding in Order-Agnostic Language Models: Chain-Rule Deviation and Uniform Spreading

arXiv 2606.00997 ↗First posted 31 May 2026cs.CLcited by 0

On the map: Diffusion models → Discrete and text diffusion

Abstract (arXiv)

Order-agnostic language models (OALMs), including discrete diffusion language models (dLLMs), are trained to predict masked tokens under arbitrary conditioning sets, allowing sequences to be generated or scored under arbitrary reveal orders at inference time. In LLaDA-2.1, we report three findings. First, the learned conditionals are not exact factorizations of a coherent joint distribution: changing only the reveal order shifts target log-likelihood by up to 0.49 nats/token, so likelihood alone mixes content difficulty with path-dependent artifacts. Second, although confidence-first (CF) decoding is order-agnostic, its reveal orders are close to left-to-right (L2R) on content tokens. Third, we propose a complementary diagnostic based on the shape of the confidence trace. A uniform-spreading theorem shows that, at fixed total likelihood, target recoverability is maximized when per-step confidence is spread uniformly; the resulting deviation motivates Var(log qt) as a diagnostic for comparing decoding paths. Across C4 and four downstream benchmarks, low variance separates structured paths from random ordering, and variance is consistently associated with downstream correctness. These results support reporting mean confidence and confidence variance jointly when comparing OALM decoding paths.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 27 of its 34 references that are in the map. First posted on arXiv 31 May 2026.

Builds on

27 of 34 references.

Closest ancestors · 16

Same direction, newest first

  1. 2026Improving Sampling for Masked Diffusion Models via Information Gain20 Feb
  2. 2026Where-to-Unmask: Ground-Truth-Guided Unmasking Order Learning for Masked Diffusion Language Models10 Feb
  3. 2026LLaDA2.1: Speeding Up Text Diffusion via Token Editing9 Feb
  4. 2026Generation Order and Parallel Decoding in Masked Diffusion Models: An Information-Theoretic Perspective30 Jan
  5. 2026Improving Diffusion Language Model Decoding through Joint Search in Generation Order and Token Space28 Jan
  6. 2026Auto-Regressive Masked Diffusion Models23 Jan
  7. 2026Parallelism and Generation Order in Masked Diffusion Language Models: Limits Today, Potential Tomorrow22 Jan
  8. 2026Deferred Commitment Decoding for Diffusion Language Models5 Jan
  9. 2025Optimizing Decoding Paths in Masked Diffusion Models by Quantifying Uncertainty24 Dec
  10. 2025LLaDA2.0: Scaling Up Diffusion Language Models to 100B10 Dec
  11. 2025Remasking Discrete Diffusion Models with Inference-Time Scaling1 Mar
  12. 2025Large Language Diffusion Models14 Feb
4 more
  1. 2025Train for the Worst, Plan for the Best: Understanding Token Ordering in Masked Diffusions10 Feb
  2. 2024Simple and Effective Masked Diffusion Language Models11 Jun
  3. 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution25 Oct
  4. 2021Autoregressive Diffusion Models5 Oct

Other references · 11

  1. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
  2. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
  3. 2022Large Language Models are Zero-Shot Reasonerscited by 4,323
  4. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
  5. 2019XLNet: Generalized Autoregressive Pretraining for Language Understandingcited by 2,801
  6. 2024MMLU-Pro: A More Robust and Challenging Multi-Task Language Understanding Benchmarkcited by 1,758
  7. 2023Instruction-Following Evaluation for Large Language Modelscited by 1,143
  8. 2022MaskGIT: Masked Generative Image Transformercited by 1,082
  9. 2015A Diversity-Promoting Objective Function for Neural Conversation Modelscited by 1,068
  10. 2017Non-Autoregressive Neural Machine Translationcited by 457
  11. 2023CMATH: Can Your Language Model Pass Chinese Elementary School Math Test?cited by 89

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Who Should Lead Decoding Now? Tracking Reliable Trajectories for Ensembling Masked Diffusion Language Models
  2. 2026Confidence-Based Decoding is Provably Efficient for Diffusion Language Models
  3. 2025Lookahead Unmasking Elicits Accurate Decoding in Diffusion Language Models
  4. 2025Diffusion Language Models Know the Answer Before Decoding
  5. 2025Beyond Confidence: Adaptive and Coherent Decoding for Diffusion Language Models
  6. 2026Locally Confident, Globally Stuck: The Quality-Exploration Dilemma in Diffusion Language Models
  7. 2025From Bits to Rounds: Parallel Decoding with Exploration for Diffusion Language Models
  8. 2025Any-Order GPT as Masked Diffusion Model: Decoupling Formulation and Architecture