Feature: Dr. GRPO Support

Open
#364 1 comment 3 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Assessment

Difficulty
4/5
Estimated time
3-5 days
Newbie friendliness
35/100
Issue type
Feature
Clarity
Needs clarification
Activity status
Stale
Tech stack
python

Research direction

No ART files or tests are named. Start by locating ART's GRPO training entry point and comparing its loss configuration with trl's GRPOTrainer support for loss_type="dr_grpo" in GRPOConfig, then read the linked Dr. GRPO paper. Done means ART can select Dr. GRPO and its behavior is covered by appropriate tests.

Written by the indexing model from the issue text.

Description

enhancement

It would be nice if art would support the Dr. GRPO optimizer.
Basically, Dr. GRPO is supposed to remove the response length bias which is present in the original GRPO, while retaining the performance benefits of GRPO. See Dr. GRPO Paper for more info.

Important Note: trl's GRPOTrainer already supports this by setting loss_type="dr_grpo" in GRPOConfig.

I believe this technique might be beneficial, especially for multi-turn RL scenarios, in which token length increase is more significant.

Dominant language
Python
Stars
10.8k
Forks
989
Avg merge
6h 25m
Merged PRs (30d)
86

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

More from OpenPipe/ART

All issues in OpenPipe/ART

Similar issues

More Python issues

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.