SRPO:仅用10%训练步数,数学代码推理双突破

SRPO:仅用10%训练步数,数学代码推理双突破

【免费下载链接】SRPO-Qwen-32B 【免费下载链接】SRPO-Qwen-32B 项目地址: https://ai.gitcode.com/hf_mirrors/Kwaipilot/SRPO-Qwen-32B

导语

最新研究表明,名为SRPO的新型强化学习框架在数学推理和代码生成两大核心任务上实现突破,仅需传统方法十分之一的训练步数,性能便超越了DeepSeek-R1-Zero-32B等现有模型。

行业现状

随着大语言模型在专业领域应用的深入,数学推理与代码生成已成为衡量模型能力的关键指标。然而,当前主流模型普遍面临训练成本高昂、跨领域能力难以兼顾的挑战。据行业数据显示,顶尖模型的训练往往需要数千GPU小时,且在数学或代码单一领域表现优异的模型,在另一领域常出现性能折损。

产品/模型亮点

SRPO-Qwen-32B基于Qwen2.5-32B基座模型开发,其核心创新在于"两阶段历史重采样策略优化"框架。该框架通过两大技术突破实现效率跃升:

首先是两阶段跨域训练范式,第一阶段专注数学数据训练以培养深度推理能力,第二阶段引入代码数据实现技能融合。这种设计有效解决了数学长推理链与代码简洁表达之间的训练冲突。

更关键的是历史重采样(HR)技术,通过过滤"过于简单"的样本,保留提供有效梯度信号的"信息性"样本,使训练效率提升近10倍。实验数据显示,在AIME24数学 benchmark上,SRPO达到50.0%的Pass@1准确率,在LiveCodeBench代码任务上达到41.6%,双双超越DeepSeek-R1-Zero-32B。

图片为AIME24中不同模型(SRPO、DeepSeek-R1-Zero-Qwen-32B、Qwen2.5-32B-Base)的准确率随训练步数变化的折线图,SRPO模型准确率随训练步数增长显著提升,其他模型准确率相对稳定。 该图表清晰展示了SRPO在数学推理任务上的效率优势。当训练步数仅为DeepSeek的10%时,SRPO的准确率已实现反超,且持续攀升趋势明显,印证了其训练方法的高效性。

另一项突破性发现是模型展现出的类人认知行为。训练过程中,SRPO自发形成了自我反思、修正和回溯等高级推理模式,甚至会用代码验证数学解,实现跨领域技能的有机结合。

该图表由四个子图组成,展示了训练过程中不同类型‘顿悟时刻’(Aha Moments)相关关键词的频率趋势,横轴为训练步骤,纵轴为频率,涵盖Alternatives、Hesitations、Rechecks及Total四个类别。 这组趋势图记录了SRPO在训练中逐渐发展出的复杂推理行为。随着训练推进,模型使用"再检查"、"犹豫"等反思性词汇的频率显著增加,表明其正在形成类似人类解决复杂问题时的思维模式。

行业影响

SRPO技术路径的成功验证了"方法论创新优于资源堆砌"的可能性。对于企业而言,这意味着可以用更低的计算成本开发高性能模型,尤其利好算力资源有限的研究机构和中小企业。在应用层面,该技术有望推动AI在科学计算、工程开发等专业领域的普及,加速研发流程。

值得注意的是,SRPO在代码生成领域同样表现出色。在LiveCodeBench基准测试中,其性能曲线呈现独特的"第二阶段跃升"现象,表明两阶段训练策略对代码能力培养的有效性。

这是LiveCodeBench(2024.08-2025.01)的折线图,展示了SRPO、DeepSeek-R1-Zero-Qwen-32B、Qwen2.5-32B-Base三种模型在训练步骤下的准确率变化,其中SRPO在Stage 2阶段准确率显著提升。 该图揭示了SRPO在代码任务上的独特学习模式。在第二阶段引入代码训练数据后,模型准确率出现陡峭上升,显示出前期数学推理训练为代码能力打下了坚实基础,验证了跨领域能力迁移的可行性。

结论/前瞻

SRPO框架的提出为大模型训练提供了新范式,其"少即是多"的高效训练理念可能引领行业从"算力竞赛"转向"方法创新"。随着技术迭代,我们有理由期待更高效、更智能的AI系统出现,在保持高性能的同时大幅降低计算资源消耗。对于开发者和企业而言,关注这类方法论创新将成为获取竞争优势的关键。

【免费下载链接】SRPO-Qwen-32B 【免费下载链接】SRPO-Qwen-32B 项目地址: https://ai.gitcode.com/hf_mirrors/Kwaipilot/SRPO-Qwen-32B

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值