47 min
Efficient Reinforcement Learning for Diffusion Models
Reinforcement Learning, Machine Learning and seminar by Yongxin Chen, Georgia Institute of Technology
Monday 14:00 Los Angeles (GMT-7)
Recording available
Berkeley, United States
Recording
Abstract
Applying language-model reinforcement-learning recipes directly to diffusion trajectories is costly in both computation and memory. This talk examines objectives, likelihood estimation and rollout design for diffusion post-training. An estimator based on a forward-process variational bound separates training from rollout and permits fast ODE sampling, producing substantial speed improvements. A related first-order stochastic-control approach is also considered.
Topics
language-model reinforcement-learningdiffusion trajectoriesdiffusion post-trainingobjectiveslikelihood estimationrollout designforward-process variational boundODE samplingfirst-order stochastic-control approach