Enforcing Constraints in Generative Sampling via Adaptive Correction Scheduling
Not on the map. Why?
Abstract (arXiv)
Hard constraints in generative sampling are typically enforced by projection, applied either once at the end of sampling or after every update. This binary framing overlooks a fundamental issue: projection changes the distribution of states which future updates depend on. As a result, delayed projection can produce samples that are feasible but inconsistent with the intended sampling dynamics, even after final projection. We formalize constraint enforcement as a correction scheduling problem over the generative rollout. Using one-step constraint defect as a local signal of geometric mismatch, we introduce adaptive correction scheduling, a state-dependent policy that allocates projection budget to the steps that most strongly perturb the trajectory. Terminal and stepwise projection arise as limiting cases of this family. Across controlled manifold rollouts and a learned projected diffusion sampler, adaptive scheduling improves the cost-accuracy frontier at matched projection budgets, recovering 71.2% of full stepwise benefit with 75% fewer corrections. These results show that constraint timing is a first-class design variable in generative sampling, and that enforcing feasibility alone is insufficient to preserve the intended constrained sampling dynamics.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 15 of its 23 references that are in the map. First posted on arXiv 11 May 2026.
Builds on
15 of 23 references.
References · 15
- 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusioncited by 2,838
- 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,061
- 2024Constrained Synthesis with Projected Diffusion Modelscited by 89
- 2023SafeDiffuser: Safe Planning with Diffusion Probabilistic Modelscited by 75
- 2025Physics-Constrained Flow Matching: Sampling Generative Models with Hard Constraintscited by 52
- 2024Gradient-Free Generation for Hard-Constrained Systemscited by 38
- 2025Simultaneous Multi-Robot Motion Planning with Projected Diffusion Modelscited by 38
- 2024Schedule On the Fly: Diffusion Time Prediction for Faster and Better Image Generationcited by 30
- 2024DiffuSolve: Diffusion-based Solver for Non-convex Trajectory Optimizationcited by 20
- 2025Model Already Knows the Best Noise: Bayesian Active Noise Selection via Attention in Video Diffusion Modelcited by 10
- 2025Optimal Stepsize for Diffusion Samplingcited by 7
- 2026Verifier-Constrained Flow Expansion for Discovery Beyond the Datacited by 7
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Constraint-Aware Flow Matching: Decision Aligned End-to-End Training for Constrained Sampling
- 2026Integration Matters: Rollout-Based Training for Constrained Diffusion Models
- 2025HardFlow: Hard-Constrained Sampling for Flow-Matching Models via Trajectory Optimization
- 2025Chance-constrained Flow Matching for High-Fidelity Constraint-aware Generation
- 2026Formalizing the Sampling Design Space of Diffusion-Based Generative Models via Adaptive Solvers and Wasserstein-Bounded Timesteps
- 2024Constrained Diffusion with Trust Sampling
- 2026Manifold-Aware Perturbations for Constrained Generative Modeling
- 2025Designing Instance-Level Sampling Schedules via REINFORCE with James-Stein Shrinkage