RRPO: Reference-Relative Policy Optimization with Stratified Conditional Rollouts
Summary
Reference-Relative Policy Optimization (RRPO) is a new method that generalizes Group Relative Policy Optimization (GRPO) for reinforcement learning, particularly in settings where a single correctness criterion is insufficient. RRPO addresses this by replacing direct correctness-based advantage construction with reference-relative contrastive comparisons. The process involves using stratified conditional rollouts to create positive and negative anchor sets, followed by training a metric projection head with a set-contrastive objective to compare candidate rollouts against these anchors. These alignment scores then directly define contrastive advantages, which are centered within each rollout group during policy optimization while the projection head is frozen. Evaluated across verifiable reasoning, open-ended generation, and post-supervised fine-tuning (SFT) tasks, RRPO demonstrates competitiveness with verifier-based optimization, outperforms weakly supervised baselines, and offers additional performance gains after SFT.
Key takeaway
For AI Scientists developing reinforcement learning agents in complex domains lacking single correctness criteria, RRPO offers a robust alternative to verifier-based methods. You should consider integrating RRPO's reference-relative contrastive comparisons and stratified conditional rollouts to define policy advantages. This approach can enhance performance in open-ended generation and post-SFT settings, providing competitive results without relying on explicit ground-truth verifiers. Evaluate its applicability to your specific tasks requiring nuanced feedback.
Key insights
RRPO generalizes GRPO by using reference-relative contrastive comparisons and stratified conditional rollouts for policy optimization.
Principles
- Success in many tasks requires more than a single correctness criterion.
- Reference-relative contrastive comparisons can define advantages without ground-truth verifiers.
- Stratified conditional rollouts enable robust anchor set construction.
Method
RRPO constructs positive/negative anchor sets via stratified conditional rollouts, trains a metric projection head with a set-contrastive objective, then uses resulting alignment scores as centered contrastive advantages for policy optimization.
In practice
- Apply RRPO in open-ended generation tasks.
- Use RRPO for post-SFT performance gains.
- Implement RRPO for verifiable reasoning tasks.
Topics
- Reference-Relative Policy Optimization
- Reinforcement Learning
- Contrastive Learning
- Policy Optimization
- Stratified Conditional Rollouts
- Verifiable Feedback
Best for: Research Scientist, AI Scientist
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.