arXiv Computer Vision By Shivanshu Shekhar, Raghavendra Addanki, Tong Zhang, Uttaran Bhattacharya

Spend Where It Matters: Particle Search with MH Refinement for Budget-Constrained Audio-Visual Alignment

Read the original on arXiv Computer Vision →

The paper introduces “Spend Where It Matters”, a two‑stage, training‑free framework that combines particle search with Metropolis‑Hastings refinement for budget‑constrained audio‑visual alignment. In the first stage, particles are adaptively allocated across denoising checkpoints based on online variance estimates, reducing estimation variance while focusing compute where it matters most. The second stage uses MH refinement with local noise‑and‑denoise proposals to further improve reward, starting from the strong candidate produced by the particle search. Experiments on “JavisBench‑Mini” show that this combined approach yields higher rewards than either particle search or MH alone, achieving significant relative gains over baseline methods. The method also demonstrates benefits for image generation on SD1.5, improving aesthetic and image reward scores compared to uniform particle allocation.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Computer Vision.

arXiv Machine Learning
Oct 16

Noise-Space Reward Alignment for Few-Step Generative Models

The paper introduces Noise Policy Optimization (NPO), a method for aligning few-step generative models with black-box reward signals. NPO learns an amortized flow policy that transforms initial Gaussian noise into a reward-tilted distribution, allowing the pretrained generator to produce higher-reward outputs without fine-tuning. The approach achieves superior reward-quality trade-offs across image, language, and protein generation tasks while drastically reducing inference cost compared to test-time search methods.

By Woocheol Shin, Taeyoung Yun, Jiwon Sung, Hyeongyu Kang, Jeongmin Park, Jaewoo Lee, Kyuil Sim, Dohyun Kim, Yutong He, Jinkyoo Park
arXiv Machine Learning
Aug 28

GRAS: Guided Reduced-Variance Proposals and Adaptive Selection for Training-Free Reward Alignment in Discrete Diffusion

The paper introduces GRAS, a method that improves training‑free reward alignment for discrete diffusion models by reducing variance in guided proposals and adapting the resampling temperature during search. It achieves this without adding denoiser cost, using Rao‑Blackwellized estimates for differentiable rewards and a leave‑one‑out baseline for non‑differentiable ones. Experiments on regulatory DNA and protein design show GRAS outperforms existing training‑free techniques and rivals reward‑fine‑tuned models.

By Kwanyoung Kim