GPG: A Simple and Strong Reinforcement Learning Baseline for Model Reasoning
About
Reinforcement Learning (RL) can directly enhance the reasoning capabilities of large language models without extensive reliance on Supervised Fine-Tuning (SFT). In this work, we revisit the traditional Policy Gradient (PG) mechanism and propose a minimalist RL approach termed Group Policy Gradient (GPG). Unlike conventional methods, GPG directly optimize the original RL objective, thus obviating the need for surrogate loss functions. By eliminating the critic and reference models, avoiding KL divergence constraints, and addressing the advantage and gradient estimation bias, our approach significantly simplifies the training process compared to Group Relative Policy Optimization (GRPO). Our approach achieves superior performance without relying on auxiliary techniques or adjustments. As illustrated in Figure 1, extensive experiments demonstrate that our method not only reduces computational costs but also consistently outperforms GRPO across various unimodal and multimodal tasks. Our code is available at https://github.com/AMAP-ML/GPG.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Mathematical Reasoning | AMC | Accuracy65 | 151 | |
| Mathematical Reasoning | Minerva | Pass@127.21 | 138 | |
| Mathematical Reasoning | AIME 24 | Accuracy33.3 | 113 | |
| Mathematical Reasoning | MATH 500 | MATH 500 Accuracy80 | 106 | |
| Mathematical Reasoning | OlympiadBench | Accuracy0.424 | 34 | |
| Mathematical Reasoning | Olympiad | Accuracy (%)42.4 | 21 | |
| Visual Question Answering | SGG-VQA Anatomy | Avg@541.09 | 20 | |
| Visual Question Answering | OmniMedVQA MI | Avg@580.89 | 18 | |
| Mathematical Reasoning | Mathematical Reasoning Aggregate | Average Score37.93 | 18 | |
| Mathematical Reasoning | Olympiad | Score37.67 | 17 |