arXiv:2512.02581cs.LG2025-12中稿 · ICML被引 3

用分阶段优化提升生成式策略的稳定性与表现力。

Training Diffusion Policies via Prior-Mapping Co-Evolution

  • 将策略优化与动作生成解耦,先在隐空间优化再映射到动作。
  • 在HopperStand任务上收益超870,是最强基线的3倍以上。
  • 适合高维复杂控制任务,尤其擅长处理多模态动作分布。

强化学习面临核心矛盾:易于优化的策略(如高斯分布)表达能力弱,难以刻画复杂控制所需多模态动作分布;而表达能力强的生成式策略(如扩散模型、流匹配)在在线强化学习中因不可计算的似然和长链梯度传播难以优化。本文提出一种结构化解决思路:将优化与生成解耦。基于此,引入算法无关的GoRL(Generative Online Reinforcement Learning)框架,通过将策略优化限制在可处理的隐空间,同时将动作生成交由条件生成解码器完成,从零训练表达性强的策略。该过程可视为先验-映射协同进化:每阶段先通过强化学习改进可处理的隐先验,再将其行为固化为更丰富的先验到动作映射。这一双时间尺度调度机制,以固定先验解码器精炼为锚点,实现稳定优化的同时持续增强表达能力。实验证明,GoRL在多种连续控制任务中持续优于单峰及生成式基线。特别地,在HopperStand任务上收益超过870,是最强基线的3倍以上;在高维人形机器人任务中,性能超越最强非GoRL基线一个数量级以上。

原文摘要 · Abstract (English)

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 -- can be difficult to optimize 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. Viewed as prior-mapping co-evolution, each stage first improves a tractable latent prior through RL and then consolidates the resulting behavior into a more expressive prior-to-action mapping. This two-timescale schedule, anchored by fixed-prior decoder refinement, enables stable optimization while continuously expanding expressiveness. Empirically, \textsc{GoRL} consistently outperforms unimodal and generative baselines across diverse continuous-control tasks. Notably, GoRL achieves returns exceeding 870 on HopperStand, more than 3* the strongest baseline; on high-dimensional humanoid tasks, it further outperforms the strongest non-GoRL baseline by over an order of magnitude.

强化学习生成策略扩散模型连续控制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。