Step-level Denoising-time Diffusion Alignment with Multiple Objectives
On the map: Diffusion models → Diffusion and flow policies for robotics
also in 1 more
Abstract (arXiv)
Reinforcement learning (RL) has emerged as a powerful tool for aligning diffusion models with human preferences, typically by optimizing a single reward function under a KL regularization constraint. In practice, however, human preferences are inherently pluralistic, and aligned models must balance multiple downstream objectives, such as aesthetic quality and text-image consistency. Existing multi-objective approaches either rely on costly multi-objective RL fine-tuning or on fusing separately aligned models at denoising time, but they generally require access to reward values (or their gradients) and/or introduce approximation error in the resulting denoising objectives. In this paper, we revisit the problem of RL fine-tuning for diffusion models and address the intractability of identifying the optimal policy by introducing a step-level RL formulation. Building on this, we further propose Multi-objective Step-level Denoising-time Diffusion Alignment (MSDDA), a retraining-free framework for aligning diffusion models with multiple objectives, obtaining the optimal reverse denoising distribution in closed form, with mean and variance expressed directly in terms of single-objective base models. We prove that this denoising-time objective is exactly equivalent to the step-level RL fine-tuning, introducing no approximation error. Moreover, we provide numerical results, which indicate our method outperforms existing denoising-time approaches.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Diffusion and flow policies for robotics, one of 14 directions in Diffusion models. Builds on 37 of its 42 references that are in the map. First posted on arXiv 15 Apr 2026.
Builds on
37 of 42 references.
Closest ancestors · 8
Same direction, newest first
- 2026DeRaDiff: Denoising Time Realignment of Diffusion Models28 Jan
- 2025Diffusion Blend: Inference-Time Multi-Preference Alignment for Diffusion Models24 May
- 2024Understanding Reinforcement Learning-Based Fine-Tuning of Diffusion Models: A Tutorial and Review18 Jul
- 2023Diffusion Model Alignment Using Direct Preference Optimization21 Nov
- 2023Directly Fine-Tuning Diffusion Models on Differentiable Rewards29 Sep
- 2023DPOK: Reinforcement Learning for Fine-tuning Text-to-Image Diffusion Models25 May
- 2023Training Diffusion Models with Reinforcement Learning22 May
- 2023Human Preference Score: Better Aligning Text-to-Image Models with Human Preference25 Mar
Other references · 29
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,585
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,355
- 2015Trust Region Policy Optimizationcited by 2,580
- 2021Variational Diffusion Modelscited by 1,153
- 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,045
- 2023Universal Guidance for Diffusion Modelscited by 437
- 2023Aligning Text-to-Image Models using Human Feedbackcited by 422
17 more
- 2023FreeDoM: Training-Free Energy-Guided Conditional Diffusion Modelcited by 294
- 2023Rewarded soups: towards Pareto-optimal alignment by interpolating weights fine-tuned on diverse rewardscited by 243
- 2014A Survey of Multi-Objective Sequential Decision-Makingcited by 242
- 2022Optimizing Prompts for Text-to-Image Generationcited by 207
- 2019A Generalized Algorithm for Multi-Objective Reinforcement Learning and Policy Adaptationcited by 177
- 2024Derivative-Free Guidance in Continuous and Discrete Diffusion Models with Soft Value-Based Decodingcited by 136
- 2024Token-level Direct Preference Optimizationcited by 135
- 2024TFG: Unified Training-Free Guidance for Diffusion Modelscited by 107
- 2024Decoding-Time Language Model Alignment with Multiple Objectivescited by 94
- 2025Test-time Alignment of Diffusion Models without Reward Over-optimizationcited by 93
- 2024Decoding-time Realignment of Language Modelscited by 67
- 2023VILA: Learning Image Aesthetics from User Comments with Vision-Language Pretrainingcited by 65
- 2023Three-Way Trade-Off in Multi-Objective Learning: Optimization, Generalization and Conflict-Avoidancecited by 36
- 2023Direction-oriented Multi-objective Learning: Simple and Provable Stochastic Algorithmscited by 34
- 2022Anchor-Changing Regularized Natural Policy Gradient for Multi-Objective Reinforcement Learningcited by 15
- 2025CoDe: Blockwise Control for Denoising Diffusion Modelscited by 14
- 2024MGDA Converges under Generalized Smoothness, Provablycited by 7
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025CoCA: Step-level Reward for Free in RL-based T2I Diffusion Model Fine-tuning
- 2025Towards Better Alignment: Training Diffusion Models with Reinforcement Learning Against Sparse Rewards
- 2026Do Less, Achieve More: Do We Need Every-Step Optimization for RL Fine-tuning of Diffusion Models?
- 2025Inference-Time Alignment Control for Diffusion Models with Reinforcement Learning Guidance
- 2026Explore or Converge? Stage-Guided Per-Step Optimization for Diffusion Models
- 2025Directly Aligning the Full Diffusion Trajectory with Fine-Grained Human Preference
- 2026Diffusion Alignment Beyond KL: Variance Minimisation as Effective Policy Optimiser
- 2024Training-free Diffusion Model Alignment with Sampling Demons