cs.LGSep 30, 2026

Manifold-Constrained Initial Noise Optimization for Efficient Generative Model Alignment

Authors: Jinho Chang, Jong Chul Ye

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

Appendix figures & tables6 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Gradient-Free Noise Optimization for Reward Alignment in Generative Models

    May 12, 2026Jeongsol Kim, Hongeun Kim, Jian Wang +1Reward GradientsDiffusion Alignment

  2. Trust-Region Noise Search for Black-Box Alignment of Diffusion and Flow Models

    Mar 15, 2026Niklas Schweiger, Daniel Cremers, Karnik RamDiffusion AlignmentGenerative Flow Networks

  3. Beyond Random Couplings: Contrastive Noise Alignment in Generative Flows

    Sep 16, 2026Lennart Wittke, Vinicius AzevedoDiffusion AlignmentGenerative Flow Networks