Reinforcement Learning-driven Diffusion Models: Language, Vision, and Beyond

Zhiyuan Ma, Jiabao Wei, Tianyu Wang, Hanqing Wang, Jiadong Pan +19 more
10/15/2026
cs.AI

Official code: MAIR-Lab-HUST/Awesome-RL-for-Diffusion-Models (8 stars, last commit 2026-10-13).

Abstract

Diffusion models (DMs) have steadily demonstrated outstanding performance in high-fidelity, controllable generation across diverse scenarios, following the "pretraining-finetuning" paradigm established in LLMs. Despite these advances, conventional DMs still face critical limitations in sophisticated reasoning, autonomous exploration of generation pathways, and adaptability to complex, human-aligned objectives. Inspired by RLHF's success in LLMs, RL has recently emerged as a pivotal post-training mechanism to iteratively refine DMs through reward-driven trajectory optimization. This paper presents the first comprehensive review of RL-driven DMs and an empirical analysis of how RL reshapes the diffusion process towards experience-centric, goal-directed generation. We propose a novel technical panorama elucidating RL's multifaceted roles in diffusion-based language, multimodal, robotic, and scientific models, with a granular focus on diffusion trajectory optimization and reward engineering. Through extensive experiments on multiple task-specific benchmarks, we systematically evaluate representative methods to intuitively highlight the advantages of different RL post-training strategies. Furthermore, we reveal and dissect three crucial bottlenecks that currently hinder the real-world deployment of RL-driven DMs: reward design fragility, exploration-exploitation trade-off in high-dimensional generation spaces, and training instability under sparse rewards. Based on these perspectives, we provide a principled roadmap for the integration of RL into DMs and outline promising future research directions for robust, interpretable RL-diffusion systems. We also construct a public repository to continuously track the latest developments in this rapidly evolving field: https://github.com/MAIR-Lab-HUST/Awesome-RL-for-Diffusion-Models.

View on arXivView PDF

Code Implementations(1)

A curated list of resources and papers on Reinforcement Learning (RL) for Diffusion Models — including reward-driven training, preference optimization, RLHF, and controllable generation.

81Sep 20, 20251 weeks ago

Similar Papers

Cite this paper

@article{ma2026reinforcement,
  title  = {Reinforcement Learning-driven Diffusion Models: Language, Vision, and Beyond},
  author = {Zhiyuan Ma and Jiabao Wei and Tianyu Wang and Hanqing Wang and Jiadong Pan and Haowen Xu and Liangliang Zhao and Ruixun Liu and Yuhan Wang and Yuwei Zhang and Liang Feng and Junhao Xiao and Kaiyan Zhang and Xuekai Zhu and Daizong Liu and Biqing Qi and Yihao Liu and Ganqu Cui and Ning Ding and Linfeng Zhang and Jianjun Li and Kun He and Xiang Bai and Bowen Zhou},
  year   = {2026},
  eprint = {2610.21539},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2610.21539v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2610.21539.svg)](https://codeofpaper.com/paper/2610.21539)

Discussion