mlsanityData to 14 Oct 2026

Step-level Denoising-time Diffusion Alignment with Multiple Objectives

arXiv 2604.14379 ↗First posted 15 Apr 2026cs.LG cs.AI cs.CVcited by 0

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

  1. 2026DeRaDiff: Denoising Time Realignment of Diffusion Models28 Jan
  2. 2025Diffusion Blend: Inference-Time Multi-Preference Alignment for Diffusion Models24 May
  3. 2024Understanding Reinforcement Learning-Based Fine-Tuning of Diffusion Models: A Tutorial and Review18 Jul
  4. 2023Diffusion Model Alignment Using Direct Preference Optimization21 Nov
  5. 2023Directly Fine-Tuning Diffusion Models on Differentiable Rewards29 Sep
  6. 2023DPOK: Reinforcement Learning for Fine-tuning Text-to-Image Diffusion Models25 May
  7. 2023Training Diffusion Models with Reinforcement Learning22 May
  8. 2023Human Preference Score: Better Aligning Text-to-Image Models with Human Preference25 Mar

Other references · 29

  1. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  2. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
  3. 2022Training language models to follow instructions with human feedbackcited by 15,971
  4. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
  5. 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
  6. 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,585
  7. 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,355
  8. 2015Trust Region Policy Optimizationcited by 2,580
  9. 2021Variational Diffusion Modelscited by 1,153
  10. 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,045
  11. 2023Universal Guidance for Diffusion Modelscited by 437
  12. 2023Aligning Text-to-Image Models using Human Feedbackcited by 422
17 more
  1. 2023FreeDoM: Training-Free Energy-Guided Conditional Diffusion Modelcited by 294
  2. 2023Rewarded soups: towards Pareto-optimal alignment by interpolating weights fine-tuned on diverse rewardscited by 243
  3. 2014A Survey of Multi-Objective Sequential Decision-Makingcited by 242
  4. 2022Optimizing Prompts for Text-to-Image Generationcited by 207
  5. 2019A Generalized Algorithm for Multi-Objective Reinforcement Learning and Policy Adaptationcited by 177
  6. 2024Derivative-Free Guidance in Continuous and Discrete Diffusion Models with Soft Value-Based Decodingcited by 136
  7. 2024Token-level Direct Preference Optimizationcited by 135
  8. 2024TFG: Unified Training-Free Guidance for Diffusion Modelscited by 107
  9. 2024Decoding-Time Language Model Alignment with Multiple Objectivescited by 94
  10. 2025Test-time Alignment of Diffusion Models without Reward Over-optimizationcited by 93
  11. 2024Decoding-time Realignment of Language Modelscited by 67
  12. 2023VILA: Learning Image Aesthetics from User Comments with Vision-Language Pretrainingcited by 65
  13. 2023Three-Way Trade-Off in Multi-Objective Learning: Optimization, Generalization and Conflict-Avoidancecited by 36
  14. 2023Direction-oriented Multi-objective Learning: Simple and Provable Stochastic Algorithmscited by 34
  15. 2022Anchor-Changing Regularized Natural Policy Gradient for Multi-Objective Reinforcement Learningcited by 15
  16. 2025CoDe: Blockwise Control for Denoising Diffusion Modelscited by 14
  17. 2024MGDA Converges under Generalized Smoothness, Provablycited by 7

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025CoCA: Step-level Reward for Free in RL-based T2I Diffusion Model Fine-tuning
  2. 2025Towards Better Alignment: Training Diffusion Models with Reinforcement Learning Against Sparse Rewards
  3. 2026Do Less, Achieve More: Do We Need Every-Step Optimization for RL Fine-tuning of Diffusion Models?
  4. 2025Inference-Time Alignment Control for Diffusion Models with Reinforcement Learning Guidance
  5. 2026Explore or Converge? Stage-Guided Per-Step Optimization for Diffusion Models
  6. 2025Directly Aligning the Full Diffusion Trajectory with Fine-Grained Human Preference
  7. 2026Diffusion Alignment Beyond KL: Variance Minimisation as Effective Policy Optimiser
  8. 2024Training-free Diffusion Model Alignment with Sampling Demons