让大模型学会多样解题路径,提升数学推理能力
DRA-GRPO: Your GRPO Needs to Know Diverse Reasoning Paths for Mathematical Reasoning
- 用语义密度调整奖励信号,鼓励不同解题思路
- 在5个数学数据集上平均准确率达58.2%,仅用7000样本
- 适合追求高效对齐与多样推理的模型训练者
通过强化学习微调大模型,尤其是组相对策略优化(GRPO),已成为提升数学推理能力的重要方法。然而,标准GRPO依赖标量正确性奖励,该奖励对语义内容不具单射性:不同推理路径可能获得相同奖励,导致多样性-质量不一致问题——策略收敛于少数主导模式,忽略结构新颖但同样有效的策略。为此,我们提出多样性感知奖励调节(DRA),一种理论驱动的框架,利用采样组的语义密度校准奖励信号。通过子模互信息(SMI)实现逆倾向评分(IPS)机制,有效降低梯度估计偏差,形成排斥冗余的驱动力,促进高奖励空间的更优覆盖。本方法可即插即用,与GRPO变体无缝集成。在五个数学基准上的实证评估显示,DRA-GRPO持续优于强基线,在DeepSeek-R1-Distill-Qwen-1.5B模型上仅用7,000训练样本和55美元成本,平均准确率达58.2%,凸显了多样性校准在数据高效对齐中的关键作用。
原文摘要 · Abstract (English)
Post-training LLMs with Reinforcement Learning, specifically Group Relative Policy Optimization (GRPO), has emerged as a paradigm for enhancing mathematical reasoning. However, standard GRPO relies on scalar correctness rewards that are often non-injective with respect to semantic content: distinct reasoning paths receive identical rewards. This leads to a Diversity-Quality Inconsistency, where the policy collapses into a narrow set of dominant modes while ignoring equally valid but structurally novel strategies. To bridge this gap, we propose Diversity-aware Reward Adjustment (DRA), a theoretically grounded framework that calibrates the reward signal using the semantic density of sampled groups. By leveraging Submodular Mutual Information (SMI), DRA implements an Inverse Propensity Scoring (IPS) mechanism that effectively de-biases the gradient estimation. This creates a repulsive force against redundancy, driving the policy to achieve better coverage of the high-reward landscape. Our method is plug-and-play and integrates seamlessly with GRPO variants. Empirical evaluations on five math benchmarks demonstrate that DRA-GRPO consistently outperforms strong baselines, achieving an average accuracy of 58.2% on DeepSeek-R1-Distill-Qwen-1.5B with only 7,000 training samples and $55 cost, highlighting the critical role of diversity calibration in data-efficient alignment. The code is available at https://github.com/xiwenc1/DRA-GRPO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。