用格式和长度做奖励信号,让大模型解数学题不用标准答案
Surrogate Signals from Format and Length: Reinforcement Learning for Solving Mathematical Problems without Ground Truth Answers
- 用输出格式和长度作为替代奖励信号指导强化学习
- 70亿参数模型在AIME2024上达40.0%准确率,媲美有标准答案的训练
- 适合资源有限时提升大模型推理能力,尤其关注高效训练策略
大型语言模型在自然语言处理中表现卓越,强化学习在适配其特定应用中起关键作用。然而,在数学问题求解中,依赖真实答案带来高昂收集成本与数据稀缺。本文探索使用简单的替代信号——格式与长度——来引导强化学习训练。研究发现,早期训练主要由格式学习主导,仅结构反馈即带来大部分性能提升。引入长度奖励可进一步优化输出,避免过长或过短的回答,使基于格式-长度信号的GRPO方法在某些情况下超越甚至逼近基于真实答案的优化效果。例如,该方法在70亿参数基模型上于AIME2024达到40.0%准确率,并对不同模型规模和题目系列具有良好泛化性。除了实际效率优势,这些发现揭示:强化学习并非传授新知识,而是激活预训练模型中已存在的推理潜能。这表明轻量、低标签依赖的策略可有效结合预训练,释放大模型在高推理强度任务中的潜在能力。
原文摘要 · Abstract (English)
Large Language Models (LLMs) have achieved remarkable success in natural language processing tasks, with Reinforcement Learning (RL) playing a key role in adapting them to specific applications. In mathematical problem solving, however, the reliance on ground truth answers poses significant challenges due to their high collection cost and limited availability. This work explores the use of simple surrogate signals, format and length, to guide RL training. We find that early training is dominated by format learning, where structural feedback alone accounts for most performance gains. Incorporating length-based rewards further refines outputs by discouraging overly long or short responses, enabling a GRPO approach with format-length signals to approximate, and in some cases surpass, ground-truth-based optimization. For example, our method achieves 40.0% accuracy on AIME2024 with a 7B base model, and generalizes across different model sizes and series. Beyond practical efficiency, these findings provide an inspirational perspective on RL: rather than imparting new knowledge, RL primarily activates reasoning capabilities already embedded in pre-trained models. This insight suggests that lightweight, label-efficient strategies can complement pre-training to unlock LLMs' latent potential in reasoning-intensive tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。