MEGA Hub

Off-Context GRPO: Learning to Reason on Hard Problems using Privileged Information

Authors

Do you know Priyank Agrawal?You can claim authorship or link another user.Do you know Ankur Samanta?You can claim authorship or link another user.Do you know Shervin Ghasemlou?You can claim authorship or link another user.Do you know Jalaj Bhandari?You can claim authorship or link another user.Do you know Kavosh Asadi?You can claim authorship or link another user.Do you know Daniel Jiang?You can claim authorship or link another user.Do you know Aditya Modi?You can claim authorship or link another user.

Abstract

Reinforcement learning with verifiable rewards (RLVR) improves reasoning in large language models. Yet, typical RLVR approaches fail on difficult problems: when a model cannot generate any correct solutions, it receives \textit{zero} learning signal. Providing privileged guidance during training, such as solution prefixes, can help overcome this learning cliff by steering the model towards {correct solutions with non-zero reward}. {We call these rollouts \textit{off-context}: they are generated from a training prompt that contains privileged guidance, while the target objective is defined by the original prompt without that guidance.} {We introduce} Off-Context GRPO (OC-GRPO), a minimally modified variant of GRPO that uses guided rollouts but applies an importance-corrected objective to steer the update back toward the original unguided objective, avoiding the mismatch that destabilizes uncorrected guided training. Empirically, our algorithm achieves a 3.9\% absolute improvement (13.8\% relative gain) over vanilla GRPO on average across standard mathematical reasoning benchmarks with negligible additional cost.

Community

00

Publication notes

Author note
24 Pages