改进强化学习方法,让大模型推理更全面
ReCo: Reweighting GRPO Against Distributional Concentration

- 通过重加权缓解策略集中于高频回复的偏差
- 在数学推理任务上显著提升大k值时的通过率
- 适合需要全面推理路径的模型微调场景
组相对策略优化(GRPO)已成为后训练语言模型的标准强化学习方法。近期研究发现,GRPO会削弱基础模型的推理能力,在大k值的Pass@k指标上表现不佳,表明推理路径覆盖范围缩小。我们发现这种现象源于GRPO使响应集中在基础模型已高概率生成的内容上。这一集中现象由GRPO更新中的两个机制导致:在响应层面,高频响应因重复出现而主导组梯度;在词元层面,重要性比例放大梯度,进一步强化当前策略下更可能的词元。我们提出ReCo,一种重加权方法,分别应对这两种效应:响应贡献按其在采样组中的期望出现频率归一化,词元层面的重要性比例被基于方差的比例替代,该比例对尚未饱和的决策点(仍存在合理备选词元)赋予更大更新尺度。在Qwen2.5-Math-1.5B/7B和Llama-3.1-8B-Instruct五个数学推理基准上,ReCo在大k值下显著提升Pass@k,且在小k值下与GRPO表现相当。
原文摘要 · Abstract (English)
Group Relative Policy Optimization (GRPO) has become a standard reinforcement learning method for post-training language models. Recent work shows that GRPO can reduce the base model's reasoning capacity and underperform it in Pass@k when k is large, indicating reduced coverage of reasoning paths. We find that this reduction is associated with GRPO concentrating on responses that the base model already generates with high probability. We trace this concentration to two mechanisms in the GRPO update. At the response level, high-probability responses dominate the group gradient through repeated occurrence. At the token level, GRPO's importance ratio scales gradients, further reinforcing tokens that become more likely under the current policy. We propose ReCo, a reweighting method that addresses both effects. Response contributions are normalized by their expected occurrence within the rollout group, and the token-level importance ratio is replaced with a variance-based ratio that gives larger update scale to non-saturated decision points where alternative token choices remain plausible. Across Qwen2.5-Math-1.5B/7B and Llama-3.1-8B-Instruct on five mathematical reasoning benchmarks, ReCo improves Pass@k for large values of k and is comparable to GRPO for small values of k.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。