Constrained Synthesis with Projected Diffusion Models
About
This paper introduces an approach to endow generative diffusion processes the ability to satisfy and certify compliance with constraints and physical principles. The proposed method recast the traditional sampling process of generative diffusion models as a constrained optimization problem, steering the generated data distribution to remain within a specified region to ensure adherence to the given constraints. These capabilities are validated on applications featuring both convex and challenging, non-convex, constraints as well as ordinary differential equations, in domains spanning from synthesizing new materials with precise morphometric properties, generating physics-informed motion, optimizing paths in planning scenarios, and human motion synthesis.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Constrained Generation | l2-ball constrained set (d=8) 1.0 (synthetic) | Constraint Violation Rate0.00e+0 | 16 | |
| Synthetic generation under constraints | Box | SWD Score0.1268 | 14 | |
| Robotic Manipulation | Robotic Manipulation Task (test) | Safety Rate46 | 10 | |
| Maze Navigation | Maze navigation task | Safety Rate22 | 10 | |
| PDE control | PDE control (test) | Rsample0.00e+0 | 10 | |
| Synthetic generation under constraints | 2 boxes | SWD0.5661 | 10 | |
| Synthetic generation under constraints | Subspace | SWD0.3079 | 10 | |
| Constrained Generation | l2-ball constrained set d=20 1.0 (synthetic) | Constraint Violation Rate0.00e+0 | 9 | |
| Constrained Generative Modeling | Earth & Climate | JSD (Volcano)0.156 | 9 | |
| Constrained Generative Modeling | Mesh Data | JSD (Bunny, 50)0.049 | 9 |