mlsanityData to 14 Oct 2026

Recursive Scaling in Masked Diffusion Models

arXiv 2606.18022 ↗First posted 16 Jun 2026cs.LGcited by 2

On the map: Diffusion models → Discrete and text diffusion

Abstract (arXiv)

Masked diffusion models (MDMs) generate sequences by iteratively refining a partially masked state and committing tokens in parallel. We introduce recursion in MDMs and propose new Recursive Masked Diffusion Models (R-MDMs), which apply a shared denoising transformer L times within each denoising step, adding recursive depth as an additional compute axis without increasing parameter count. Across structured generation tasks, recursive depth improves quality at fixed parameter budget, matches substantially larger non-recursive models at matched FLOPs, and can reduce the number of denoising steps needed to reach a target quality. We interpret these gains with a dependence--fidelity decomposition of parallel decoding error: recursion refines model marginals at a fixed masked state, whereas denoising steps change that state by committing tokens. Building on this analysis, we propose to treat decoding as a two-axis decision (how many loops to run and which tokens to commit) and show that entropy-guided adaptive rules improve the quality--compute frontier over fixed schedules, transferring across various tasks on Sudoku, Countdown, RNA, and executable math generation. Together, these results establish recursive depth as a practical, complementary test-time scaling mechanism for MDMs.

Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 42 of its 48 references that are in the map. First posted on arXiv 16 Jun 2026.

Builds on

42 of 48 references.

Closest ancestors · 12

Same direction, newest first

  1. 2026Flow Map Language Models: One-step Language Modeling via Continuous Denoising18 Feb
  2. 2026Scaling Beyond Masked Diffusion Language Models16 Feb
  3. 2026Reasoning with Latent Tokens in Diffusion Language Models3 Feb
  4. 2026Towards Latent Diffusion Suitable For Text7 Jan
  5. 2025On the Reasoning Abilities of Masked Diffusion Language Models15 Oct
  6. 2025Coevolutionary Continuous Discrete Diffusion: Make Your Diffusion Language Model a Latent Reasoner3 Oct
  7. 2025Beyond Masked and Unmasked: Discrete Diffusion Models via Partial Masking24 May
  8. 2025Large Language Diffusion Models14 Feb
  9. 2024Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning18 Oct
  10. 2024Promises, Outlooks and Challenges of Diffusion Language Modeling17 Jun
  11. 2024Simple and Effective Masked Diffusion Language Models11 Jun
  12. 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution25 Oct

Other references · 30

  1. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  2. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
  3. 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
  4. 2020Scaling Laws for Neural Language Modelscited by 6,586
  5. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  6. 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,211
  7. 2019ALBERT: A Lite BERT for Self-supervised Learning of Language Representationscited by 2,790
  8. 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,719
  9. 2019Root Mean Square Layer Normalizationcited by 1,480
  10. 2018Universal Transformerscited by 647
  11. 2019Deep Equilibrium Modelscited by 615
  12. 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 415
18 more
  1. 2014Deeply-Supervised Netscited by 367
  2. 2024Generative Flows on Discrete State-Spaces: Enabling Multimodal Flows with Applications to Protein Co-Designcited by 305
  3. 2025Reasoning with Latent Thoughts: On the Power of Looped Transformerscited by 240
  4. 2023Looped Transformers as Programmable Computerscited by 237
  5. 2025Hierarchical Reasoning Modelcited by 177
  6. 2023Looped Transformers are Better at Learning Learning Algorithmscited by 149
  7. 2024Stream of Search (SoS): Learning to Search in Languagecited by 145
  8. 2024Looped Transformers for Length Generalizationcited by 99
  9. 2024Relaxed Recursive Transformers: Effective Parameter Sharing with Layer-wise LoRAcited by 78
  10. 2023One-Step Diffusion Distillation via Deep Equilibrium Modelscited by 61
  11. 2026LoopFormer: Elastic-Depth Looped Transformers for Latent Reasoning via Shortcut Modulationcited by 61
  12. 2026Parcae: Scaling Laws For Stable Looped Language Modelscited by 55
  13. 2024Causal Language Modeling Can Elicit Search and Reasoning Capabilities on Logic Puzzlescited by 38
  14. 2025Enhancing Auto-regressive Chain-of-Thought through Loop-Aligned Reasoningcited by 19
  15. 2024Fixed Point Diffusion Modelscited by 16
  16. 2025A Formal Comparison Between Chain of Thought and Latent Thoughtcited by 15
  17. 2026SpiralFormer: Looped Transformers Can Learn Hierarchical Dependencies via Multi-Resolution Recursioncited by 12
  18. 2026Looping Back to Move Forward: Recursive Transformers for Efficient and Flexible Large Multimodal Modelscited by 5

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Remasking Discrete Diffusion Models with Inference-Time Scaling
  2. 2025Any-Order Flexible Length Masked Diffusion
  3. 2026VGB for Masked Diffusion Model: Efficient Test-time Scaling for Reward Satisfaction and Sample Editing
  4. 2026Learn from Your Mistakes: Self-Correcting Masked Diffusion Models
  5. 2026Not All Denoising Steps Are Equal: Model Scheduling for Faster Masked Diffusion Language Models
  6. 2026Transformer-Based Masked Diffusion Models Provably Learn Distributions with Latent Structure
  7. 2024Simplified and Generalized Masked Diffusion for Discrete Data
  8. 2025Error Bounds and Optimal Schedules for Masked Diffusions with Factorized Approximations