The Impact of On-Policy Parallelized Data Collection on Deep Reinforcement Learning Networks
This work provides incremental insights for RL practitioners by optimizing data collection in on-policy algorithms to enhance agent performance.
The study analyzed how parallel data collection strategies, such as the number of environments and rollout lengths, affect bias-variance trade-offs and optimization stability in PPO, a popular RL algorithm, finding that larger datasets improve performance and scaling parallel environments is more effective than increasing rollout lengths.
The use of parallel actors for data collection has been an effective technique used in reinforcement learning (RL) algorithms. The manner in which data is collected in these algorithms, controlled via the number of parallel environments and the rollout length, induces a form of bias-variance trade-off; the number of training passes over the collected data, on the other hand, must strike a balance between sample efficiency and overfitting. We conduct an empirical analysis of these trade-offs on PPO, one of the most popular RL algorithms that uses parallel actors, and establish connections to network plasticity and, more generally, optimization stability. We examine its impact on network architectures, as well as the hyper-parameter sensitivity when scaling data. Our analyses indicate that larger dataset sizes can increase final performance across a variety of settings, and that scaling parallel environments is more effective than increasing rollout lengths. These findings highlight the critical role of data collection strategies in improving agent performance.