arXiv:2506.13284cs.CLcs.AI2025-06被引 66

通过SFT与RL协同优化,显著提升数学代码推理能力。

AceReason-Nemotron 1.1: Advancing Math and Code Reasoning through SFT and RL Synergy

  • 结合监督微调与强化学习,优化训练策略。
  • 在数学与代码基准上超越现有7B模型表现。
  • 适合追求高精度推理的开发者与研究者。

本文研究监督微调(SFT)与强化学习(RL)在构建强推理模型中的协同效应。通过两种扩展策略:增加收集的提示数量和每个提示生成的响应数量,均显著提升推理性能,其中增加提示数量带来的收益更明显。研究进一步探讨了两个关键问题:(i)更强的SFT模型是否能持续带来更好的最终性能?(ii)如何为特定的SFT初始化选择合适的采样温度以平衡探索与利用?结果表明,只要进行有效的RL训练,特别是将采样温度调整至使温度修正熵维持在0.3左右,该设置能良好平衡探索与利用,则问题(i)成立。值得注意的是,初始SFT模型间的性能差距在RL过程中显著缩小。基于强大的SFT基础及对SFT与RL协同机制的理解,我们的AceReason-Nemotron-1.1 7B模型显著优于AceReason-Nemotron-1.0,且在挑战性数学与代码基准上达到Qwen2.5-7B系列模型的新最优水平,验证了后训练方案的有效性。模型与数据已公开于https://huggingface.co/nvidia/AceReason-Nemotron-1.1-7B。

原文摘要 · Abstract (English)

In this work, we investigate the synergy between supervised fine-tuning (SFT) and reinforcement learning (RL) in developing strong reasoning models. We begin by curating the SFT training data through two scaling strategies: increasing the number of collected prompts and the number of generated responses per prompt. Both approaches yield notable improvements in reasoning performance, with scaling the number of prompts resulting in more substantial gains. We then explore the following questions regarding the synergy between SFT and RL: (i) Does a stronger SFT model consistently lead to better final performance after large-scale RL training? (ii) How can we determine an appropriate sampling temperature during RL training to effectively balance exploration and exploitation for a given SFT initialization? Our findings suggest that (i) holds true, provided effective RL training is conducted, particularly when the sampling temperature is carefully chosen to maintain the temperature-adjusted entropy around 0.3, a setting that strikes a good balance between exploration and exploitation. Notably, the performance gap between initial SFT models narrows significantly throughout the RL process. Leveraging a strong SFT foundation and insights into the synergistic interplay between SFT and RL, our AceReason-Nemotron-1.1 7B model significantly outperforms AceReason-Nemotron-1.0 and achieves new state-of-the-art performance among Qwen2.5-7B-based reasoning models on challenging math and code benchmarks, thereby demonstrating the effectiveness of our post-training recipe. We release the model and data at: https://huggingface.co/nvidia/AceReason-Nemotron-1.1-7B

推理模型强化学习代码生成数学推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。