Generative Online Reinforcement Learning
Beijing University of Posts and Telecommunications · National University of Singapore · Nanjing University · Nanyang Technological University · CMU, Carnegie Mellon University · Centre for Frontier AI Research, A*STAR, Singapore · A*STAR
PDF 由论文原始站点提供,PaperCompass 不保存论文文件。
摘要
Reinforcement learning (RL) faces a persistent tension: policies that are stable to optimize (e.g., Gaussians) are often too simple to represent the multimodal action distributions required for complex control. Conversely, expressive generative policies—such as diffusion and flow matching—are frequently unstable in online RL due to intractable likelihoods and gradients propagating through long sampling chains. We address this tension with a key structural principle: decoupling optimization from generation. Building on this, we introduce GoRL (Generative Online Reinforcement Learning), an algorithm-agnostic framework that trains expressive policies from scratch by confining policy optimization to a tractable latent space while delegating action synthesis to a conditional generative decoder. Using a two-timescale alternating schedule and anchoring decoder refinement to a fixed prior, GoRL enables stable optimization while continuously expanding expressiveness. Empirically, GoRL consistently outperforms unimodal and generative baselines across diverse continuous-control tasks. Notably, on the challenging HopperStand task, it achieves episodic returns exceeding 870—more than $3\times$ that of the strongest baseline—demonstrating a practical path to policies that are both stable and highly expressive.