Saved in:
| Main Authors: | , , , , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2026
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2602.05494 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866917256575844352 |
|---|---|
| author | Wu, Qingyuan Wang, Yuhui Zhan, Simon Sinong Dai, Yanning Deng, Shilong Habchi, Sarra Zhu, Qi Gallé, Matthias Huang, Chao |
| author_facet | Wu, Qingyuan Wang, Yuhui Zhan, Simon Sinong Dai, Yanning Deng, Shilong Habchi, Sarra Zhu, Qi Gallé, Matthias Huang, Chao |
| contents | Reinforcement Learning with Verified Reward (RLVR) has emerged as a critical paradigm for advancing the reasoning capabilities of Large Language Models (LLMs). Most existing RLVR methods, such as GRPO and its variants, ensure stable updates by constraining policy divergence through clipping likelihood ratios. This paper introduces a unified clipping framework that characterizes existing methods via a general notion of policy divergence, encompassing both likelihood ratios and Kullback-Leibler (KL) divergences and extending to alternative measures. The framework provides a principled foundation for systematically analyzing how different policy divergence measures affect exploration and performance. We further identify the KL3 estimator, a variance-reduced Monte Carlo estimator of the KL divergence, as a key policy divergence constraint. We theoretically demonstrate that the KL3-based constraint is mathematically equivalent to an asymmetric ratio-based clipping that reallocates probability mass toward high-confidence actions, promoting stronger exploration while retaining the simplicity of GRPO-style methods. Empirical results on mathematical reasoning benchmarks demonstrate that incorporating the KL3 estimator into GRPO improves both training stability and final performance, highlighting the importance of principled policy divergence constraints in policy optimization. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2602_05494 |
| institution | arXiv |
| publishDate | 2026 |
| record_format | arxiv |
| spellingShingle | A Unified Framework for Rethinking Policy Divergence Measures in GRPO Wu, Qingyuan Wang, Yuhui Zhan, Simon Sinong Dai, Yanning Deng, Shilong Habchi, Sarra Zhu, Qi Gallé, Matthias Huang, Chao Machine Learning Artificial Intelligence Reinforcement Learning with Verified Reward (RLVR) has emerged as a critical paradigm for advancing the reasoning capabilities of Large Language Models (LLMs). Most existing RLVR methods, such as GRPO and its variants, ensure stable updates by constraining policy divergence through clipping likelihood ratios. This paper introduces a unified clipping framework that characterizes existing methods via a general notion of policy divergence, encompassing both likelihood ratios and Kullback-Leibler (KL) divergences and extending to alternative measures. The framework provides a principled foundation for systematically analyzing how different policy divergence measures affect exploration and performance. We further identify the KL3 estimator, a variance-reduced Monte Carlo estimator of the KL divergence, as a key policy divergence constraint. We theoretically demonstrate that the KL3-based constraint is mathematically equivalent to an asymmetric ratio-based clipping that reallocates probability mass toward high-confidence actions, promoting stronger exploration while retaining the simplicity of GRPO-style methods. Empirical results on mathematical reasoning benchmarks demonstrate that incorporating the KL3 estimator into GRPO improves both training stability and final performance, highlighting the importance of principled policy divergence constraints in policy optimization. |
| title | A Unified Framework for Rethinking Policy Divergence Measures in GRPO |
| topic | Machine Learning Artificial Intelligence |
| url | https://arxiv.org/abs/2602.05494 |