DiffusionRVGE: Reinforcement Learning via Regression-Based Value Gradient Estimation

Jiayuan Wang, Tongyao Pang

Reinforcement learning improves diffusion models using rewards on generated images, but repeated denoising and gradient estimation make training expensive. We introduce \method, an…