Balanced Aggregation: Understanding and Fixing Aggregation Bias in GRPO

Zhiyuan Zeng, Jiameng Huang, Zhangyue Yin, Bingrui Li, Jiashuo Liu, Yuhao Wu, Yining Zheng, Ge Zhang, Wenhao Huang, Xipeng Qiu, Ziniu Li


Abstract
Reinforcement learning with verifiable rewards (RLVR) has become a central paradigm for improving reasoning and code generation in large language models, and GRPO-style training is widely adopted for its simplicity and effectiveness. However, an important design choice remains underexplored: how token-level policy gradient terms are aggregated within each sampled group. Standard GRPO uses sequence aggregation, while recent work has advocated token aggregation as a better alternative. We show that these two rules induce different optimization biases: token aggregation introduces sign-length coupling, while sequence aggregation implicitly downweights longer responses through sequence-level equal weighting. To address this tension, we propose Balanced Aggregation (BA), a simple drop-in replacement that computes token-level means separately within the positive and negative subsets and then combines them with sequence-count-based weights. Experiments with Qwen2.5-Math-7B and Qwen3-1.7B on DAPO-17k and Polaris, evaluated on six reasoning and coding benchmarks, show that BA consistently improves training stability and final performance over standard token and sequence aggregation. Our analysis further shows that the relative effectiveness of token and sequence aggregation is largely governed by response-length variation and the positive-negative length gap, highlighting aggregation as a critical design dimension in GRPO-style RLVR.
Anthology ID:
2026.findings-emnlp.2120
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
45195–45207
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.2120/
DOI:
Bibkey:
Cite (ACL):
Zhiyuan Zeng, Jiameng Huang, Zhangyue Yin, Bingrui Li, Jiashuo Liu, Yuhao Wu, Yining Zheng, Ge Zhang, Wenhao Huang, Xipeng Qiu, and Ziniu Li. 2026. Balanced Aggregation: Understanding and Fixing Aggregation Bias in GRPO. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 45195–45207, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Balanced Aggregation: Understanding and Fixing Aggregation Bias in GRPO (Zeng et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.2120.pdf