Leveraging Procedural Generation to Benchmark Reinforcement Learning
Karl Cobbe, Christopher Hesse, Jacob Hilton, John Schulman
Code Available — Be the first to reproduce this paper.
ReproduceCode
- github.com/openai/procgenOfficialIn papernone★ 1,148
- github.com/openai/train-procgenOfficialIn papertf★ 0
- github.com/joonleesky/train-procgen-pytorchpytorch★ 0
- github.com/rgilman33/carlitanone★ 0
- github.com/ana-tudor/classyconditioningtf★ 0
- github.com/zxtan98/cprocgennone★ 0
Abstract
We introduce Procgen Benchmark, a suite of 16 procedurally generated game-like environments designed to benchmark both sample efficiency and generalization in reinforcement learning. We believe that the community will benefit from increased access to high quality training environments, and we provide detailed experimental protocols for using this benchmark. We empirically demonstrate that diverse environment distributions are essential to adequately train and evaluate RL agents, thereby motivating the extensive use of procedural content generation. We then use this benchmark to investigate the effects of scaling model size, finding that larger models significantly improve both sample efficiency and generalization.