Low-Confidence Remasking Traps Flexibility: Realizing Arbitrary-Order Potential for Diverse Rollouts in Diffusion LLMs
On the map: Diffusion models → Discrete and text diffusion
Abstract (arXiv)
Masked diffusion language models support arbitrary-order generation, suggesting a natural way to produce diverse outputs. However, recent work argues that this flexibility reduces diversity by delaying high-uncertainty tokens that can lead to different generation paths. We trace this diversity loss not to arbitrary-order generation itself, but largely to low-confidence remasking (LCR), a widely used decoding rule. At each step, LCR samples a token at every masked position but commits only the sampled token with the highest probability, filtering out the rest. We show that this mechanism can exponentially suppress lower-probability tokens as more positions compete, and observe the same suppression in LLaDA. In contrast, top-probability position selection (TPP), which has often been conflated with LCR under the shared label confidence-based decoding, avoids this diversity loss. TPP first selects the position whose most likely token has the highest probability, then samples directly from that position's distribution. Replacing LCR with TPP restores diversity and yields Pass@k comparable to left-to-right decoding, suggesting that the reported diversity loss stems largely from LCR's filtering rather than from generating high-confidence positions first. To further exploit order flexibility, we introduce Entropy-Guided Initialization (EGI), which samples the first token at the highest-entropy position and then follows TPP. This simple modification further improves rollout diversity and solution coverage beyond left-to-right decoding, with gains extending to downstream policy optimization, highlighting the potential of arbitrary-order generation for diverse rollouts.
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 41 of its 44 references that are in the map. First posted on arXiv 28 Sep 2026.
Builds on
41 of 44 references.
Closest ancestors · 24
Same direction, newest first
- 2026A Tale of Two Temperatures: Simple, Efficient, and Diverse Sampling from Diffusion Language Models10 Apr
- 2026Differences in Text Generated by Diffusion and Autoregressive Language Models4 Apr
- 2026Locally Confident, Globally Stuck: The Quality-Exploration Dilemma in Diffusion Language Models1 Apr
- 2026The Flexibility Trap: Rethinking the Value of Arbitrary Order in Diffusion Language Models21 Jan
- 2025KLASS: KL-Guided Fast Inference in Masked Diffusion Models7 Nov
- 2025SPG: Sandwiched Policy Gradient for Masked Diffusion Language Models10 Oct
- 2025Demystifying MaskGIT Sampler and Beyond: Adaptive Order Selection in Masked Diffusion6 Oct
- 2025Simple Policy Gradients for Reasoning with Diffusion Language Models5 Oct
- 2025d2: Improving Reasoning in Diffusion Language Models via Trajectory Likelihood Estimation25 Sep
- 2025Revolutionizing Reinforcement Learning Framework for Diffusion Large Language Models8 Sep
- 2025Dream 7B: Diffusion Large Language Models21 Aug
- 2025wd1: Weighted Policy Optimization for Reasoning in Diffusion Language Models7 Jul
12 more
- 2025DiffuCoder: Understanding and Improving Masked Diffusion Models for Code Generation25 Jun
- 2025Accelerated Sampling from Masked Diffusion Models via Entropy Bounded Unmasking30 May
- 2025Dimple: Discrete Diffusion Multimodal Large Language Model with Parallel Decoding22 May
- 2025Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models12 Mar
- 2025Remasking Discrete Diffusion Models with Inference-Time Scaling1 Mar
- 2025Large Language Diffusion Models14 Feb
- 2025Train for the Worst, Plan for the Best: Understanding Token Ordering in Masked Diffusions10 Feb
- 2024Simple and Effective Masked Diffusion Language Models11 Jun
- 2024Simplified and Generalized Masked Diffusion for Discrete Data6 Jun
- 2024Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Data6 Jun
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution25 Oct
- 2023A Reparameterized Discrete Diffusion Model for Text Generation11 Feb
Other references · 17
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2021Program Synthesis with Large Language Modelscited by 3,509
- 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,665
- 2022MaskGIT: Masked Generative Image Transformercited by 1,082
- 2024Large Language Monkeys: Scaling Inference Compute with Repeated Samplingcited by 946
- 2021Argmax Flows and Multinomial Diffusion: Learning Categorical Distributionscited by 566
- 2018Texygen: A Benchmarking Platform for Text Generation Modelscited by 484
- 2025Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decodingcited by 404
- 2025LLaDA 1.5: Variance-Reduced Preference Optimization for Large Language Diffusion Modelscited by 273
5 more
- 2025d1: Scaling Reasoning in Diffusion Large Language Models via Reinforcement Learningcited by 178
- 2022On Decoding Strategies for Neural Text Generatorscited by 89
- 2025ACECODER: Acing Coder RL via Automated Test-Case Synthesiscited by 89
- 2025Diversity-Aware Policy Optimization for Large Language Model Reasoningcited by 39
- 2025Principled RL for Diffusion LLMs Emerges from a Sequence-Level Perspectivecited by 30
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Generation Order and Parallel Decoding in Masked Diffusion Models: An Information-Theoretic Perspective
- 2025Lookahead Unmasking Elicits Accurate Decoding in Diffusion Language Models
- 2026CoDR: Training-Free Confidence-Drift Remasking for Diffusion Language Models
- 2026Re-evaluating Confidence Remasking in Masked Diffusion Language Models
- 2026Rethinking the Generation Order of Block Diffusion Language Models
- 2026NAVIRA: Decoupled Stochastic Remasking for Masked Diffusion Language Models
- 2025STaRR: Spatial-Temporal Token-Dynamics-Aware Responsive Remasking for Diffusion Language Models
- 2026Fork-dLLM: Avoiding the Flexibility Trap in Diffusion Language Models