Your Group-Relative Advantage Is Biased
Fengkai Yang, Zherui Chen, Xiaohan Wang, Xiaodong Lu, Jiajun Chai, Guojun Yin, Wei Lin, Shuai Ma, Fuzhen Zhuang, Deqing Wang, Yaodong Yang, Jianxin Li, Yikun Ban
TL;DR
Group-relative RLVR exhibits a systematic bias in the estimated advantage: hard prompts tend to be undervalued and easy prompts overvalued due to the empirical group baseline. The authors formalize this bias, prove its dependence on prompt difficulty, and introduce History-Aware Adaptive Difficulty Weighting (HA-DW), which uses an evolving difficulty anchor and history signals to reweight advantages. Theoretical results show HA-DW can reduce bias in expectation, and extensive experiments on five mathematical reasoning benchmarks across multiple model families demonstrate consistent performance gains with GRPO and related algorithms. This work highlights a crucial bias in current RLVR practice and offers a practical, plug-in method to enhance robustness and sample efficiency for reasoning-focused LLMs.
Abstract
Reinforcement Learning from Verifier Rewards (RLVR) has emerged as a widely used approach for post-training large language models on reasoning tasks, with group-based methods such as GRPO and its variants gaining broad adoption. These methods rely on group-relative advantage estimation to avoid learned critics, yet its theoretical properties remain poorly understood. In this work, we uncover a fundamental issue of group-based RL: the group-relative advantage estimator is inherently biased relative to the true (expected) advantage. We provide the first theoretical analysis showing that it systematically underestimates advantages for hard prompts and overestimates them for easy prompts, leading to imbalanced exploration and exploitation. To address this issue, we propose History-Aware Adaptive Difficulty Weighting (HA-DW), an adaptive reweighting scheme that adjusts advantage estimates based on an evolving difficulty anchor and training dynamics. Both theoretical analysis and experiments on five mathematical reasoning benchmarks demonstrate that HA-DW consistently improves performance when integrated into GRPO and its variants. Our results suggest that correcting biased advantage estimation is critical for robust and efficient RLVR training.
