用最近轨迹重放降低PPO方差,提升样本效率。
Enhancing PPO with Trajectory-Aware Hybrid Policies
- 引入先进先出的轨迹缓冲区,只保留最新策略生成的数据。
- 每轮更新用最优轨迹加随机采样轨迹,实测方差显著下降。
- 适合追求稳定训练的连续控制任务研究者使用。
近端策略优化(PPO)是现代强化学习中最流行的在策略算法之一,广泛应用于多个领域。尽管其具备稳定的性能和理论上的策略改进保证,但高方差与高样本复杂度仍是其核心挑战。为此,我们提出混合策略近端策略优化(HP3O),利用轨迹重放缓冲区高效复用近期策略生成的轨迹。该缓冲区采用先进先出(FIFO)策略,仅保留最新轨迹,以缓解数据分布漂移问题。每轮更新使用包含最优回报轨迹和从缓冲区随机采样的其他轨迹组成的批次来更新策略网络。该机制使智能体基于最新最佳表现持续提升,并在实验中有效降低方差。我们理论上构建了该算法的策略改进保证。HP3O在多个连续控制环境上与多种基线算法进行了验证与对比,结果表明其性能更优。代码已公开。
原文摘要 · Abstract (English)
Proximal policy optimization (PPO) is one of the most popular state-of-the-art on-policy algorithms that has become a standard baseline in modern reinforcement learning with applications in numerous fields. Though it delivers stable performance with theoretical policy improvement guarantees, high variance, and high sample complexity still remain critical challenges in on-policy algorithms. To alleviate these issues, we propose Hybrid-Policy Proximal Policy Optimization (HP3O), which utilizes a trajectory replay buffer to make efficient use of trajectories generated by recent policies. Particularly, the buffer applies the "first in, first out" (FIFO) strategy so as to keep only the recent trajectories to attenuate the data distribution drift. A batch consisting of the trajectory with the best return and other randomly sampled ones from the buffer is used for updating the policy networks. The strategy helps the agent to improve its capability on top of the most recent best performance and in turn reduce variance empirically. We theoretically construct the policy improvement guarantees for the proposed algorithm. HP3O is validated and compared against several baseline algorithms using multiple continuous control environments. Our code is available here.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。