Rethinking Policy Diversity in Ensemble Policy Gradient in Large-Scale Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
Scaling reinforcement learning to tens of thousands of parallel environments requires overcoming the limited exploration capacity of a single policy. Ensemble-based policy gradient methods, which employ multiple policies to collect diverse samples, have recently been proposed to promote exploration. However, merely broadening the exploration space does not always enhance learning capability, since excessive exploration can reduce exploration quality or compromise training stability. In this work, we theoretically analyze the impact of inter-policy diversity on learning efficiency in policy ensembles, and propose Coupled Policy Optimization which regulates diversity through KL constraints between policies. The proposed method enables effective exploration and outperforms strong baselines such as SAPG, PBT, and PPO across multiple tasks, including challenging dexterous manipulation, in terms of both sample efficiency and final performance. Furthermore, analysis of policy diversity and effective sample size during training reveals that follower policies naturally distribute around the leader, demonstrating the emergence of structured and efficient exploratory behavior. Our results indicate that diverse exploration under appropriate regulation is key to achieving stable and sample-efficient learning in ensemble policy gradient methods. Project page at https://naoki04.github.io/paper-cpo/ .
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 19 of its 26 references that are in the map. First posted on arXiv 2 Mar 2026.
Builds on
19 of 26 references.
References · 19
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2015Trust Region Policy Optimizationcited by 2,503
- 2017Curiosity-driven Exploration by Self-supervised Predictioncited by 1,173
- 2018Exploration by Random Network Distillationcited by 966
- 2021A Minimalist Approach to Offline Reinforcement Learningcited by 882
- 2021Isaac Gym: High Performance GPU-Based Physics Simulation For Robot Learningcited by 882
- 2018IMPALA: Scalable Distributed Deep-RL with Importance Weighted Actor-Learner Architecturescited by 866
- 2020AWAC: Accelerating Online Reinforcement Learning with Offline Datasetscited by 647
- 2021Learning to Walk in Minutes Using Massively Parallel Deep Reinforcement Learningcited by 395
- 2018Maximum a Posteriori Policy Optimisationcited by 318
- 2018Distributed Prioritized Experience Replaycited by 260
- 2022DeXtreme: Transfer of Agile In-hand Manipulation from Simulation to Realitycited by 136
7 more
- 2020Effective Diversity in Population Based Reinforcement Learningcited by 116
- 2023Dual RL: Unification and New Methods for Reinforcement and Imitation Learningcited by 48
- 2024Simple Policy Optimizationcited by 31
- 2023DexPBT: Scaling up Dexterous Manipulation for Hand-Arm Systems with Population Based Trainingcited by 24
- 2024SAPG: Split and Aggregate Policy Gradientscited by 17
- 2023Policy Space Diversity for Non-Transitive Gamescited by 15
- 2023Keep Various Trajectories: Promoting Exploration of Ensemble Policies in Continuous Controlcited by 5
Built on by
Not cited yet.