Manifold-Constrained Initial Noise Optimization for Efficient Generative Model Alignment
Organizations: Kim Jaechul Graduate School of AI KAIST Seoul, South Korea
Abstract
Recent advances in distillation and flow-map models have enabled deterministic one- or few-step generation for high-quality data, facilitating a new branch of reward alignment approaches that directly optimize the initial noise from a Gaussian distribution. However, most existing initial-noise optimization methods rely on first-order gradient information, which is either inapplicable or suffers from instability and inefficiency in black-box reward scenarios. Here, we introduce ZeNOVA, a stable and efficient initial noise alignment method in a gradient-free manner. Specifically, we address existing algorithms' major challenge in black-box scenarios through annealed soft-value guidance, manifold-constrained hyperspherical Langevin dynamics, and Metropolis-Hastings jumping. Extensive experiments on image and video generative models show that ZeNOVA outperforms all evaluated zeroth-order baselines by optimizing the initial noise toward higher rewards substantially more stably while exploiting the geometry of the Gaussian prior, demonstrating its practical applicability to various black-box reward alignment.
Figures & tables
| ImageReward | PickScore | GenEval | Counting | OCR | ||||||||
| Method | Reward | Aes. | Val. | Reward | Aes. | Val. | Reward | Aes. | Reward | Aes. | Reward | Aes. |
| SDXL-Turbo | ||||||||||||
| base model | 1.0505 | 6.0282 | 22.8809 | 21.5378 | 5.4020 | 0.5321 | 0.5375 | 5.3853 | 28.375 | 5.6965 | 0.1368 | 5.4256 |
| Best-of- | 1.6932 | 6.1148 | 23.2829 | 23.1925 | 5.5747 | 0.8949 | 0.7232 | 5.3717 | 2.9 | 5.7096 | 0.4784 | 5.3303 |
| ReNO | 1.7378 | 6.0395 | 22.9585 | 25.1031 | 5.5721 | 0.7755 | - | - | - | - | - | - |
| ORIGEN | 1.8198 | 5.1640 | 21.1789 | 24.2321 | 5.5748 | 0.9286 | - | - | - | - | - | - |
| VideoAlign | ||||
| Method | VQ | MQ | TA | All |
| base model | 3.4854 | 1.3684 | 2.1731 | 6.1997 |
| ZeNOVA | 6.4464 | 3.3124 | 3.9088 | 10.8146 |
| Variants | ImageReward | PickScore | OCR |
| ZeNOVA | 1.8078 | 23.4564 | 0.5114 |
| - MH jumping | 1.7262 | 23.1867 | 0.4412 |
| - Value Annealing | 1.6984 | 23.2960 | 0.3756 |
| - Hyperspherical update | 1.3968 | 22.2734 | 0.3823 |
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.