mlsanityData to 14 Oct 2026

When Diffusion Breaks Constraints: Sequential Autoregressive Generation with RL and MCTS

arXiv 2512.01242 ↗First posted 1 Dec 2025cs.CV cs.AI cs.CLcited by 0

On the map: Diffusion models → Diffusion foundations and sampling

Abstract (arXiv)

Data-driven generative models excel in language and vision, but diffusion models often fail in constrained planning and design tasks, exhibiting severe constraint violations in engineering inverse design, molecular generation, multi-robot planning, and floorplan/scene synthesis even with projection or guidance. Such tasks combine hard-to-specify semantic goals with strict geometric or physical constraints (e.g., non-overlap, connectivity), yielding feasible solutions that lie on low-dimensional, small, and sometimes disconnected regions of the output space. This paper studies the failure mode through tangram generation from language, where seven fixed shapes must form a text-described silhouette while remaining connected and non-overlapping, and a simplified rectangle composition task with a learned bounding-box constraint. We find diffusion models struggle to satisfy constraints, consistent with difficulty generating samples near low-dimensional submanifolds. Motivated by locally feasible reparameterizations, we reformulate constrained generation as discrete autoregressive sequential generation. Reinforcement learning improves feasibility and task success, and Monte Carlo tree search quantifies the value of look-ahead when feasible regions shrink. Overall, the empirical, theoretical, and prior-work evidence points to a structural limitation of continuous density matching on this class of constrained-generation problems, and suggests sequential constraint-aware generation as a promising alternative.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 22 of its 34 references that are in the map. First posted on arXiv 1 Dec 2025.

Builds on

22 of 34 references.

Closest ancestors · 3

Same direction, newest first

  1. 2024Constrained Synthesis with Projected Diffusion Models5 Feb
  2. 2023Mirror Diffusion Models for Constrained and Watermarked Generation2 Oct
  3. 2023FreeDoM: Training-Free Energy-Guided Conditional Diffusion Model17 Mar

Other references · 19

  1. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  2. 2017Proximal Policy Optimization Algorithmscited by 15,215
  3. 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
  4. 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,739
  5. 2019Mastering Atari, Go, Chess and Shogi by Planning with a Learned Modelcited by 1,353
  6. 2019Improved Precision and Recall Metric for Assessing Generative Modelscited by 944
  7. 2019Stochastic Beams and Where to Find Them: The Gumbel-Top-k Trick for Sampling Sequences Without Replacementcited by 211
  8. 2023DiffuScene: Denoising Diffusion Models for Generative Indoor Scene Synthesiscited by 166
  9. 2023MiDi: Mixed Graph and 3D Denoising Diffusion for Molecule Generationcited by 98
  10. 2022HouseDiffusion: Vector Floorplan Generation via a Diffusion Model with Discrete and Continuous Denoisingcited by 73
  11. 2021SketchGen: Generating Constrained CAD Sketchescited by 68
  12. 2022Abstract Visual Reasoning with Tangram Shapescited by 43
7 more
  1. 2025Simultaneous Multi-Robot Motion Planning with Projected Diffusion Modelscited by 38
  2. 2025Monte Carlo Tree Diffusion for System 2 Planningcited by 33
  3. 2025Training-Free Constrained Generation With Stable Diffusion Modelscited by 20
  4. 2019Enforcing Deterministic Constraints on Generative Adversarial Networks for Emulating Physical Systemscited by 7
  5. 2024Constrained Layout Generation with Factor Graphscited by 7
  6. 2025EngiBench: A Framework for Data-Driven Engineering Design Researchcited by 6
  7. 2020Shape related constraints aware generation of Mechanical Designs through Deep Convolutional GANcited by 2

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Constrained Generative Modeling with Manually Bridged Diffusion Models
  2. 2026Discrete Diffusion Models Exploit Asymmetry to Solve Lookahead Planning Tasks
  3. 2025Projected Coupled Diffusion for Test-Time Constrained Joint Generation
  4. 2023Aligning Optimization Trajectories with Diffusion Models for Constrained Design Generation
  5. 2022Let us Build Bridges: Understanding and Extending Diffusion Generative Models
  6. 2024Constrained Diffusion with Trust Sampling
  7. 2024An Overview of Diffusion Models: Applications, Guided Generation, Statistical Rates and Optimization
  8. 2026Compositional Visual Planning via Inference-Time Diffusion Scaling