Are we really tilting? The mechanics of reward guidance in flow and diffusion models
On the map: Diffusion models → Diffusion and flow policies for robotics
also in 1 more
Abstract (arXiv)
Reward guidance algorithms steer a learned generative process toward the reward-tilted measure at inference time. While empirically powerful, these methods are prone to reward hacking: the guided model over-optimizes the reward at the cost of fidelity to the learned distribution. Prior work has attributed this to the complexity of neural reward functions or implicit biases in diffusion training, but its fundamental origins remain poorly understood. We show that reward hacking arises from an approximation made in most practical implementations of reward-guided diffusion -- finite-particle plug-in estimation of the Doob h-function -- even in the simplest non-trivial settings of Gaussian and Gaussian mixture targets with quadratic rewards. In closed form, we isolate two distinct failure modes of the plug-in estimator: it leads to reward hacking within each mode and it cannot select high-reward modes. We propose a closed-form reward damping schedule that corrects the within-mode bias with no additional compute, and clarify the role of best-of-n sampling in compensating for the mode selection failure. Experiments on Gaussian mixture targets, a 2D checkerboard, and FLUX.1 text-to-image generation confirm that our theoretical insights carry over to practical settings.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Placed in Diffusion and flow policies for robotics, one of 14 directions in Diffusion models. Builds on 41 of its 53 references that are in the map. First posted on arXiv 1 Jun 2026.
Builds on
41 of 53 references.
Closest ancestors · 7
Same direction, newest first
- 2026The tractability landscape of diffusion alignment: regularization, rewards, and computational primitives12 May
- 2026Diamond Maps: Efficient Reward Alignment via Stochastic Flow Maps5 Feb
- 2024Adjoint Matching: Fine-tuning Flow and Diffusion Generative Models with Memoryless Stochastic Optimal Control13 Sep
- 2024Fine-Tuning of Continuous-Time Diffusion Models as Entropy-Regularized Control23 Feb
- 2024Confronting Reward Overoptimization for Diffusion Models: A Perspective of Inductive and Primacy Biases13 Feb
- 2023Diffusion Model Alignment Using Direct Preference Optimization21 Nov
- 2023Training Diffusion Models with Reinforcement Learning22 May
Other references · 34
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,322
- 2022Classifier-Free Diffusion Guidancecited by 5,717
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2025Qwen2.5-VL Technical Reportcited by 5,418
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
- 2022Scaling Laws for Reward Model Overoptimizationcited by 1,056
- 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,010
- 2023Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generationcited by 950
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 897
22 more
- 2023Human Preference Score v2: A Solid Benchmark for Evaluating Human Preferences of Text-to-Image Synthesiscited by 847
- 2022Equivariant Diffusion for Molecule Generation in 3Dcited by 666
- 2022GeoDiff: a Geometric Diffusion Model for Molecular Conformation Generationcited by 511
- 2023SE(3) diffusion model with application to protein backbone generationcited by 214
- 2022Structure-based Drug Design with Equivariant Diffusion Modelscited by 213
- 2023Helping or Herding? Reward Model Ensembles Mitigate but do not Eliminate Reward Hackingcited by 195
- 20233D Equivariant Diffusion for Target-Aware Molecule Generation and Affinity Predictioncited by 176
- 2023Protein Design with Guided Discrete Diffusioncited by 154
- 2024Scaling Laws for Reward Model Overoptimization in Direct Alignment Algorithmscited by 139
- 2022Protein structure generation via folding diffusioncited by 113
- 2025How to build a consistency model: Learning flow maps via self-distillationcited by 96
- 2024DecompDiff: Diffusion Models with Decomposed Priors for Structure-Based Drug Designcited by 87
- 2025Inference-Time Alignment in Diffusion Models with Reward-Guided Generation: Tutorial and Reviewcited by 82
- 2024Analysis of Classifier-Free Guidance Weight Schedulerscited by 64
- 2025RewardDance: Reward Scaling in Visual Generationcited by 63
- 2024Theoretical Insights for Diffusion Guidance: A Case Study for Gaussian Mixture Modelscited by 46
- 2024What does guidance do? A fine-grained analysis in a simple settingcited by 45
- 2026Meta Flow Maps enable scalable reward alignmentcited by 32
- 2025Classifier-Free Guidance: From High-Dimensional Analysis to Generalized Guidance Formscited by 9
- 2024General Binding Affinity Guidance for Diffusion Models in Structure-Based Drug Designcited by 5
- 2026Steering diffusion models with quadratic rewards: a fine-grained analysiscited by 5
- 2026Emergence of Distortions in High-Dimensional Guided Diffusion Modelscited by 3
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2025PIRF: Physics-Informed Reward Fine-Tuning for Diffusion Models
- 2025Towards a unified framework for guided diffusion models
- 2026On-Policy Self-Distillation in Diffusion Models
- 2026A unified perspective on fine-tuning and sampling with diffusion and flow models
- 2026ReCAST: Reward Credit Assignment across Timesteps for Online Diffusion Reinforcement
- 2026Scaling Reinforcement Learning for Diffusion Models via Velocity Matching
- 2026A Unified Density Operator View of Flow Control and Merging
- 2023Reward-Directed Conditional Diffusion: Provable Distribution Estimation and Reward Improvement