通过训练时多采样、筛选高效回答,让大模型推理更简洁
Sample More to Think Less: Group Filtered Policy Optimization for Concise Reasoning
- 训练时多采样并筛选短而高效的回答进行优化
- 在多个复杂任务上减少46%-85%的冗长输出,保持准确率
- 适合追求高效推理、降低计算成本的研究者和开发者
以可验证奖励训练的大语言模型常因追求准确率而过度延长响应长度,产生大量重复冗余内容。本文提出GFPO(组过滤策略优化),在训练中对每道题采样更多答案,并基于响应长度和每令牌奖励(奖励/令牌比)两个指标筛选训练样本,从而抑制长度膨胀。在Phi-4-reasoning模型上,相比基线方法GRPO,GFPO在AIME 24/25、GPQA、Omni-MATH、LiveCodeBench等挑战性STEM与编码基准上,将长度膨胀降低46%-71%;进一步优化奖励/令牌比,降幅可达71%-85%。我们还提出自适应难度GFPO,根据实时难度估计动态分配更多训练资源给难题,提升复杂问题上的效率与准确率平衡。结果表明,增加训练阶段算力可直接降低推理阶段算力需求,是一种简单而有效的高效推理策略。
原文摘要 · Abstract (English)
Large language models trained with reinforcement learning with verifiable rewards tend to trade accuracy for length--inflating response lengths to achieve gains in accuracy. While longer answers may be warranted for harder problems, many tokens are merely "filler": repetitive, verbose text that makes no real progress. We introduce GFPO (Group Filtered Policy Optimization), which curbs this length explosion by sampling larger groups per problem during training and filtering responses to train on based on two key metrics: (1) response length and (2) token efficiency: reward per token ratio. By sampling more at training time, we teach models to think less at inference time. On the Phi-4-reasoning model, GFPO cuts GRPO's length inflation by 46-71% across challenging STEM and coding benchmarks (AIME 24/25, GPQA, Omni-MATH, LiveCodeBench) while maintaining accuracy. Optimizing for reward per token further increases reductions in length inflation to 71-85%. We also propose Adaptive Difficulty GFPO, which dynamically allocates more training resources to harder problems based on real-time difficulty estimates, improving the balance between computational efficiency and accuracy especially on difficult questions. GFPO demonstrates that increased training-time compute directly translates to reduced test-time compute--a simple yet effective trade-off for efficient reasoning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。