Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Recursive Scaling in Masked Diffusion Models

About

Masked diffusion models (MDMs) have recently emerged as a promising paradigm for sequence generation. Scaling MDMs is conventionally achieved by increasing the parameter count or the number of denoising steps. We introduce Recursive Masked Diffusion Models (R-MDMs), which add recursive depth as a third scaling axis by repeatedly applying the same denoising transformer within each diffusion step. Recursion enables iterative refinement of the output through parameter reuse, increasing effective model depth without increasing parameter count. Across structured generation tasks, including Sudoku and Countdown, we show that R-MDMs achieve substantially improved parameter efficiency: a model with $L$ recursive iterations often matches the performance of non-recursive baselines with roughly $L\times$ more parameters. Moreover, recursive refinement can partially substitute for additional denoising steps, allowing recursive models to reach the same generation quality with fewer forward passes at inference time. These results suggest that recursive depth is a practically useful scaling mechanism for MDMs, improving both parameter efficiency and the allocation of test-time compute.

Alba Carballo-Castro, Julianna Piskorz, Paulius Rauba, Mihaela van der Schaar, Pascal Frossard• 2026

Related benchmarks

TaskDatasetResultRank
CountdownCountdown-5
RTR (%)56.4
24
CountdownCountdown 4
RTR (%)86.2
20
Sudoku SolvingSudoku 25x25 70% masking
VPR (%)100
18
Sudoku SolvingSudoku 25x25 80% masking
VPR (%)100
18
Sudoku SolvingSudoku 25x25 (90% masking)
VPR (%)52.2
18
CountdownCountdown 3
RTR (%)98.4
15
Showing 6 of 6 rows

Other info

Follow for update