Proximal Policy Optimization Algorithms
About
We propose a new family of policy gradient methods for reinforcement learning, which alternate between sampling data through interaction with the environment, and optimizing a "surrogate" objective function using stochastic gradient ascent. Whereas standard policy gradient methods perform one gradient update per data sample, we propose a novel objective function that enables multiple epochs of minibatch updates. The new methods, which we call proximal policy optimization (PPO), have some of the benefits of trust region policy optimization (TRPO), but they are much simpler to implement, more general, and have better sample complexity (empirically). Our experiments test PPO on a collection of benchmark tasks, including simulated robotic locomotion and Atari game playing, and we show that PPO outperforms other online policy gradient methods, and overall strikes a favorable balance between sample complexity, simplicity, and wall-time.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Code Generation | HumanEval (test) | Pass@17.8 | 444 | |
| Multi-turn Dialogue Evaluation | MT-Bench | Overall Score7.21 | 331 | |
| Mathematical Reasoning | AIME 2024 | Accuracy26.7 | 251 | |
| Mathematical Reasoning | MATH | Accuracy59.7 | 162 | |
| Mathematical Reasoning | CollegeMATH | Accuracy41.1 | 161 | |
| Mathematical Reasoning | MATH (test) | Pass@173.52 | 151 | |
| Mathematical Reasoning | AMC | Accuracy52.5 | 151 | |
| Multi-hop Question Answering | Bamboogle | Exact Match24 | 97 | |
| Mathematical Reasoning | GSM8K (val) | Accuracy87 | 67 | |
| Interactive Decision-making | ALFWorld (test) | Success Rate80.46 | 67 |