Alignment and Safety of Diffusion Models via Reinforcement Learning and Reward Modeling: A Survey
On the map: Diffusion models → Diffusion and flow policies for robotics
also in 1 more
Abstract (arXiv)
Diffusion models have become a central paradigm for image and multimodal generation, yet their deployment raises persistent questions about alignment, safety, preference satisfaction, and robustness to misuse. This survey reviews recent progress on aligning text-to-image diffusion models through reinforcement learning, reward modeling, preference optimization, and safety-specific fine-tuning. We organize the literature along five axes: the source of feedback, the form of the reward or preference signal, the optimization mechanism, the treatment of distribution shift and reward overoptimization, and the extent to which safety is addressed as an explicit constraint rather than a generic preference. The review covers reinforcement learning from human feedback, KL-regularized policy optimization, direct preference optimization, binary utility optimization, differentiable reward fine-tuning, surrogate reward learning, region-aware fine-tuning, and safety-oriented DPO variants. To make the survey accessible, we include tutorial explanations of diffusion sampling, reward modeling, and preference optimization, and briefly connect image diffusion alignment to emerging text and masked language diffusion models. We also compare representative methods in terms of feedback requirements, computational cost, scalability, susceptibility to reward hacking, and suitability for safety-critical deployment. Finally, we synthesize the literature into a set of open challenges: multi-objective alignment, feedback-efficient preference learning, adversarially robust safety alignment, continual alignment under changing norms, and interpretable reward modeling. The goal of this survey is to provide a coherent technical map of the emerging area of diffusion model alignment and to identify the methodological gaps that must be addressed before aligned generative models can be reliably deployed.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in Diffusion and flow policies for robotics, one of 14 directions in Diffusion models. Builds on 44 of its 51 references that are in the map. First posted on arXiv 23 May 2025.
Builds on
44 of 51 references.
Closest ancestors · 15
Same direction, newest first
- 2025Flow-GRPO: Training Flow Matching Models via Online RL8 May
- 2025A Simple and Effective Reinforcement Learning Method for Text-to-Image Diffusion Fine-tuning2 Mar
- 2025Personalized Preference Fine-tuning of Diffusion Models11 Jan
- 2024Reward Fine-Tuning Two-Step Diffusion Models via Learning Differentiable Latent-Space Surrogate Reward22 Nov
- 2024Aligning Few-Step Diffusion Models with Dense Reward Difference Learning18 Nov
- 2024Aligning Diffusion Models by Optimizing Human Utility6 Apr
- 2024Feedback Efficient Online Fine-Tuning of Diffusion Models26 Feb
- 2024PRDP: Proximal Reward Difference Prediction for Large-Scale Reward Finetuning of Diffusion Models13 Feb
- 2024Confronting Reward Overoptimization for Diffusion Models: A Perspective of Inductive and Primacy Biases13 Feb
- 2024A Dense Reward View on Aligning Text-to-Image Diffusion with Preference13 Feb
- 2023Using Human Feedback to Fine-tune Diffusion Models without Any Reward Model22 Nov
- 2023Diffusion Model Alignment Using Direct Preference Optimization21 Nov
3 more
Other references · 29
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 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
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,585
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,355
- 2022BLIP: Bootstrapping Language-Image Pre-training for Unified Vision-Language Understanding and Generationcited by 4,322
- 2022LAION-5B: An open large-scale dataset for training next generation image-text modelscited by 4,152
- 2017Deep reinforcement learning from human preferencescited by 3,912
17 more
- 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,761
- 2021GLIDE: Towards Photorealistic Image Generation and Editing with Text-Guided Diffusion Modelscited by 3,614
- 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,719
- 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,045
- 2022Diffusion-LM Improves Controllable Text Generationcited by 1,027
- 2019An Introduction to Variational Autoencoderscited by 1,000
- 2023Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generationcited by 977
- 2025Large Language Diffusion Modelscited by 912
- 2023Human Preference Score v2: A Solid Benchmark for Evaluating Human Preferences of Text-to-Image Synthesiscited by 890
- 2024Simple and Effective Masked Diffusion Language Modelscited by 848
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distributioncited by 745
- 2023RLAIF vs. RLHF: Scaling Reinforcement Learning from Human Feedback with AI Feedbackcited by 625
- 2022DiffuSeq: Sequence to Sequence Text Generation with Diffusion Modelscited by 450
- 2023Aligning Text-to-Image Models using Human Feedbackcited by 422
- 2024Alignment of Diffusion Models: Fundamentals, Challenges, and Futurecited by 35
- 2024AlignGuard: Scalable Safety Alignment for Text-to-Image Generationcited by 20
- 2025Focus-N-Fix: Region-Aware Fine-Tuning for Text-to-Image Generationcited by 9
Built on by
Citing papers per year: 2025: 1; 2026: 2.
3 papers.
Similar, not cited
Close in wording, no citation either way
- 2024Large-scale Reinforcement Learning for Diffusion Models
- 2025Diffusion Blend: Inference-Time Multi-Preference Alignment for Diffusion Models
- 2026Beyond Pairwise Preferences: Listwise Reward-Aware Alignment for Diffusion Models
- 2025Preference Alignment on Diffusion Model: A Comprehensive Survey for Image Generation and Editing
- 2025Towards Better Alignment: Training Diffusion Models with Reinforcement Learning Against Sparse Rewards
- 2025Taming Preference Mode Collapse via Directional Decoupling Alignment in Diffusion Reinforcement Learning
- 2025Diffusion-SDPO: Safeguarded Direct Preference Optimization for Diffusion Models
- 2023Enhancing Diffusion Models with Text-Encoder Reinforcement Learning