Skip to content

feat(length-reward): support global reasoning profile bands - #3802

Closed
ehosseiniasl wants to merge 1 commit into
adithyare/gdpo-lenpen-no-truncation-mainfrom
ehsan/adi-async-length-wiring
Closed

feat(length-reward): support global reasoning profile bands#3802
ehosseiniasl wants to merge 1 commit into
adithyare/gdpo-lenpen-no-truncation-mainfrom
ehsan/adi-async-length-wiring

Conversation

@ehosseiniasl

Copy link
Copy Markdown
Contributor

Summary

  • Allow a global reasoning profile band under grpo.length_bonus.default when dataset rows do not provide profile metadata.
  • Preserve row-level and agent-level overrides.
  • Compute reasoning length from generated assistant token IDs only, split at the configured reasoning-end token.
  • Clamp the post-band multiplier to its configured floor.
  • Add focused utility and async propagation coverage.

Motivation

Video async GRPO uses one shared reasoning-length profile across the cached dataset. Requiring every row to carry identical profile metadata is unnecessary and makes Arushi-style video recipes harder to configure safely.

Validation

  • 13 focused length-adjustment tests pass.
  • Async collector regression confirms length_adjustment_config reaches Gym rollouts.
  • Ruff checks pass for the changed implementation and tests.

Full 2-node synchronous and 16-node asynchronous video validation is pending; this PR remains draft until those runs complete.

Dependency

The Arushi-aligned video recipes are being validated on the separate integration branch ehsan/adi-video-profile-integration because the current #3604 branch predates the merged video recipes from #3648.

- clamp profile-band multipliers at the configured floor after the upper bound
- derive reasoning length from generated assistant token IDs without prompt history
- support global profile bands with row and agent overrides
- cover async configuration propagation and profile-band behavior

Signed-off-by: Ehsan Hosseini Asl <ehsan.hosseiniasl@gmail.com>
@copy-pr-bot

copy-pr-bot Bot commented Aug 24, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@ehosseiniasl

Copy link
Copy Markdown
Contributor Author

Closing for now while runtime validation and upstream dependency work continue. I will reopen or create the review PR only when it is ready.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant