huggingface / huggingface/trl

[GPG][new trainer] Add support to new `GPG` method

Open
#3,472 1 comment 0 reactions 0 assignees View on GitHub
✨ enhancement
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

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.