mlsanityData to 14 Oct 2026

A Minimalist Method for Fine-tuning Text-to-Image Diffusion Models

arXiv 2506.12036 ↗First posted 23 May 2025cs.LG cs.AIcited by 3

On the map: Diffusion models → Diffusion and flow policies for robotics

also in 1 more
Abstract (arXiv)

Recent work uses reinforcement learning (RL) to fine-tune text-to-image diffusion models, improving text-image alignment and sample quality. However, existing approaches introduce unnecessary complexity: they cache the full sampling trajectory, depend on differentiable reward models or large preference datasets, or require specialized guidance techniques. Motivated by the "golden noise" hypothesis -- that certain initial noise samples can consistently yield superior alignment -- we introduce Noise PPO, a minimalist RL algorithm that leaves the pre-trained diffusion model entirely frozen and learns a prompt-conditioned initial noise generator. Our approach requires no trajectory storage, reward backpropagation, or complex guidance tricks. Extensive experiments show that optimizing the initial noise distribution consistently improves alignment and sample quality over the original model, with the most significant gains at low inference steps. As the number of inference steps increases, the benefit of noise optimization diminishes but remains present. These findings clarify the scope and limitations of the golden noise hypothesis and reinforce the practical value of minimalist RL fine-tuning for diffusion models.

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 43 of its 46 references that are in the map. First posted on arXiv 23 May 2025.

Builds on

43 of 46 references.

Closest ancestors · 5

Same direction, newest first

  1. 2023Diffusion Model Alignment Using Direct Preference Optimization21 Nov
  2. 2023Directly Fine-Tuning Diffusion Models on Differentiable Rewards29 Sep
  3. 2023DPOK: Reinforcement Learning for Fine-tuning Text-to-Image Diffusion Models25 May
  4. 2023Training Diffusion Models with Reinforcement Learning22 May
  5. 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusion7 Mar

Other references · 38

  1. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  2. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
  3. 2023GPT-4 Technical Reportcited by 18,396
  4. 2017Proximal Policy Optimization Algorithmscited by 15,215
  5. 2020Denoising Diffusion Implicit Modelscited by 9,781
  6. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  7. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
  8. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
  9. 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
  10. 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,355
  11. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
  12. 2022Classifier-Free Diffusion Guidancecited by 5,847
26 more
  1. 2022Flow Matching for Generative Modelingcited by 5,702
  2. 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,356
  3. 2022LAION-5B: An open large-scale dataset for training next generation image-text modelscited by 4,152
  4. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,700
  5. 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 3,045
  6. 2022Prompt-to-Prompt Image Editing with Cross Attention Controlcited by 2,431
  7. 2022Video Diffusion Modelscited by 2,305
  8. 2023Consistency Modelscited by 1,822
  9. 2023Align your Latents: High-Resolution Video Synthesis with Latent Diffusion Modelscited by 1,476
  10. 2022Scaling Autoregressive Models for Content-Rich Text-to-Image Generationcited by 1,295
  11. 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,061
  12. 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,045
  13. 2023Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generationcited by 977
  14. 2023Human Preference Score v2: A Solid Benchmark for Evaluating Human Preferences of Text-to-Image Synthesiscited by 890
  15. 2023Adversarial Diffusion Distillationcited by 759
  16. 2022Equivariant Diffusion for Molecule Generation in 3Dcited by 677
  17. 2022GeoDiff: a Geometric Diffusion Model for Molecular Conformation Generationcited by 516
  18. 2024Guiding a Diffusion Model with a Bad Version of Itselfcited by 299
  19. 2024Applying Guidance in a Limited Interval Improves Sample and Distribution Quality in Diffusion Modelscited by 254
  20. 2022Structure-based Drug Design with Equivariant Diffusion Modelscited by 217
  21. 2023LCM-LoRA: A Universal Stable-Diffusion Acceleration Modulecited by 211
  22. 2024Not All Noises Are Created Equally:Diffusion Noise Selection and Optimizationcited by 71
  23. 2025Diffusion Models without Classifier-free Guidancecited by 33
  24. 2024TextCraftor: Your Text Encoder Can be Image Quality Controllercited by 27
  25. 2024ReNeg: Learning Negative Embedding with Reward Guidancecited by 9
  26. 2024Improving image synthesis with diffusion-negative samplingcited by 8

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Towards Better Alignment: Training Diffusion Models with Reinforcement Learning Against Sparse Rewards
  2. 2026PAST: Prompt-Adaptive Sampling Termination for Efficient Diffusion Model
  3. 2025A Simple and Effective Reinforcement Learning Method for Text-to-Image Diffusion Fine-tuning
  4. 2025CoCA: Step-level Reward for Free in RL-based T2I Diffusion Model Fine-tuning
  5. 2025Alignment and Safety of Diffusion Models via Reinforcement Learning and Reward Modeling: A Survey
  6. 2023Aligning Text-to-Image Diffusion Models with Reward Backpropagation
  7. 2026Step-level Denoising-time Diffusion Alignment with Multiple Objectives
  8. 2026Critic-Guided Reinforcement Unlearning in Text-to-Image Diffusion