[GPG][new trainer] Add support to new `GPG` method
- Dominant language
- Python
- Stars
- 19.3k
- Forks
- 3k
- Avg merge
- 1d 20h
- Merged PRs (30d)
- 194
Description
### Method description
- **short description**: 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 optimizes 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. Extensive experiments demonstrate that our method not only reduces computational costs but also consistently outperforms GRPO across various unimodal and multimodal tasks.
- **paper link**: https://arxiv.org/abs/2504.02546
### Open source status
- [x] The method implementation is available
- [x] The model weights are available
- [x] The training datasets are available
### Provide useful links for the implementation
- source-code link: https://github.com/AMAP-ML/GPG
- model weights link:
- https://huggingface.co/GD-ML/Qwen2.5-Math-7B-GPG
- authors:
- GitHub: @cxxgtxy @lerogo @undyingjoker
- Scholar: [Xiangxiang Chu](https://scholar.google.com/citations?user=jn21pUsAAAAJ), [Hailang Huang](https://scholar.google.com/citations?user=X0o0Ib8AAAAJ), [Xiao Zhang](https://github.com/undyingjoker), [Fei Wei](https://scholar.google.com/citations?user=xqrPe6gAAAAJ), [Yong Wang](https://www.semanticscholar.org/author/Yong-Wang/1683878) (AMAP, Alibaba Group)
Contributor guide
Assessment
This issue has not been assessed yet.