通过百万并行环境提升PPO训练,打破性能停滞瓶颈。
Preventing Learning Stagnation in PPO by Scaling to 1 Million Parallel Environments
- 将PPO外循环建模为随机优化,揭示步长过大导致学习停滞。
- 实验证明增加并行环境数可使性能持续提升至一万亿步。
- 适合追求高效强化学习训练的算法研究者和工程师。
深度在线策略强化学习中,智能体性能停滞于次优水平是常见问题。聚焦于PPO,我们发现某些场景下的性能平台并非源于探索、容量或优化挑战,而是因为样本估计的损失逐渐偏离真实目标。PPO在训练中交替进行在线多环境采样(外循环)与基于离线数据集的多次小批量SGD更新(内循环)。本文抽象掉内循环,将外循环视为标准随机优化过程:步长受策略正则化强度控制,梯度噪声由两次更新间采样数量决定。该框架预测,若外循环步长相对于噪声过大,更新将失去信息量,导致政策在局部最优附近震荡。据此,解决该问题有两种途径:减小步长或增加采样数量。实验验证了该模型预测,并表明增加并行环境数量是同时调节两因素的简单方法。将PPO扩展至超过100万并行环境,实现从一万亿过渡的单调性能提升,在复杂开放域中显著优于以往基线。
原文摘要 · Abstract (English)
An agent's performance stagnating at a suboptimal level is a common problem in deep on-policy RL. Focusing on PPO, we show that plateaus in certain regimes arise not because of known exploration, capacity, or optimisation challenges, but because sample-based estimates of the loss eventually become poor proxies for the true objective over the course of training. Looking deeper, PPO alternates between sampling rollouts from several parallel environments online using the current policy (which we call the "outer loop") and performing repeated minibatch SGD steps against this offline dataset (the "inner loop"). In our work, we abstract away the inner loop, and conceptually model the outer loop as standard stochastic optimisation. The step size is then controlled by the regularisation strength towards the previous policy and the gradient noise by the number of samples collected between policy update steps. This framing predicts that, much like in SGD, if the outer step size is too large relative to the noise, updates become uninformative and lead to the policy thrashing around a local optimum instead of converging. Recasting PPO in this light makes it clear that there are two ways to address this particular type of learning stagnation: either reduce the step size or increase the number of samples collected between updates. We validate the predictions of our model and conclude that increasing the number of parallel environments is a simple way to avoid these plateaus by simultaneously altering both these factors. Applying our analysis and scaling PPO to more than 1M parallel environments enables monotonic performance improvement up to one trillion transitions and leads to vastly superior performance compared to prior baselines in a complex open-ended domain.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。