Spend Where It Matters: Particle Search with MH Refinement for Budget-Constrained Audio-Visual Alignment
cs.CV, cs.SD, eess.AS
Submitted: 2026-10-08
Updated: 2026-10-08
Code: https://github.com/LAION-AI/aesthetic-predictor
License: http://creativecommons.org/licenses/by-nc-sa/4.0/
The gist: Reward-guided inference-time alignment can improve joint audio-visual generation, but optimizing multiple alignment objectives requires many expensive model evaluations, making existing methods
Terminology
Abstract
Reward-guided inference-time alignment can improve joint audio-visual generation, but optimizing multiple alignment objectives requires many expensive model evaluations, making existing methods costly under limited compute. Particle-based search enables global exploration but incurs estimation error with a finite population, while Metropolis-Hastings (MH) can refine candidates locally but benefits from a strong initialization. We introduce Spend Where It Matters, a two-stage, training-free, reward-agnostic framework that combines particle search with MH refinement under a fixed compute budget. In the first stage, we adaptively allocate particles across denoising checkpoints using online variance estimates, minimizing estimation variance while concentrating compute where it is most useful. This search provides a promising candidate for MH refinement, which uses local noise-and-denoise proposals on completed outputs to further improve reward. By starting from a strong candidate, MH can recover additional reward without requiring a larger particle population or costly global exploration. We show that, at matched compute, combining particle search with MH refinement achieves higher rewards than either approach alone. On JavisBench-Mini, at closely matched compute, achieves mean relative gains across seven metrics of 30.7% over BoN and 14.1% over EvoSearch with JavisDiT++, and 9.2% over BoN and 17.0% over EvoSearch with LTX-2. Beyond joint audio-visual generation, we also demonstrate the benefits of adaptive particle allocation on SD1.5 image generation, improving AestheticScore by 2.2% and ImageReward by 6.0% over uniform allocation.
Sources
- Training Diffusion Models with Reinforcement Learning
- Stable Video Diffusion: Scaling Latent Video Diffusion Models to Large Datasets
- SyncDPO: Enhancing Temporal Synchronization in Video-Audio Joint Generation via Preference Learning
- Directly Fine-Tuning Diffusion Models on Differentiable Rewards
- Scaling Rectified Flow Transformers for High-Resolution Image Synthesis
- Fast Timing-Conditioned Latent Audio Diffusion
- ImageBind: One Embedding Space To Bind Them All
- LTX-2: Efficient Joint Audio-Visual Foundation Model
- Denoising Diffusion Probabilistic Models
- Inference-Time Scaling for Joint Audio-Video Generation
- SpongeBob: Sync-Aware Harmonious Audio-Visual Generative Editing
- Flow Matching for Generative Modeling
- AudioLDM: Text-to-Audio Generation with Latent Diffusion Models
- Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flow
- Large Language Diffusion Models
- Simple and Effective Masked Diffusion Language Models
- ROCM: RLHF on consistency models
- SEE-DPO: Self Entropy Enhanced Direct Preference Optimization
- VASR: Variance-Aware Systematic Resampling for Reward-Guided Diffusion
- A General Framework for Inference-time Scaling and Steering of Diffusion Models
Related papers
- Loss Knows Best: Detecting Annotation Errors in Videos via Loss Trajectories
- AnchorWeave: World-Consistent Video Generation with Retrieved Local Spatial Memories
- Benchmarking the Robustness of Foundation Models for Mammography under Domain Shift
- MambaX-Net: Dual-Input Mamba-Enhanced Cross-Attention Network for Longitudinal MRI Segmentation
- TeleOCR: Navigating Document Parsing Across Digital and Camera-Captured Documents
- A Survey on Efficient Vision-Language-Action Models