arXiv:2601.19404cs.AIcs.LG2026-01被引 3

只优化推理路径后半段,训练速度提升90%以上

RPO:Reinforcement Fine-Tuning with Partial Reasoning Optimization

  • 用缓存经验只生成推理后缀,省去前半段计算
  • 1.5B模型训练时间减少90%,7B模型减少72%
  • 可插拔集成现有算法,提速不降性能

在大语言模型领域,强化学习微调算法需从输入查询开始生成完整的推理路径,导致训练阶段的滚动生成过程计算开销巨大。为解决此问题,我们分析了推理路径不同片段对最终结果正确性的影响,并据此提出一种即插即用的强化学习微调算法——部分推理优化(RPO)。与传统方法生成完整推理路径不同,RPO通过经验缓存仅生成推理路径的后缀进行训练。在训练的滚动生成阶段,RPO将令牌生成量降低约95%,显著降低理论时间开销。相比全路径强化学习微调算法,RPO使1.5B模型的训练时间减少90%,7B模型减少72%。同时,RPO可与GRPO、DAPO等典型算法集成,使其在保持原有性能的前提下实现训练加速。代码已开源:https://github.com/yhz5613813/RPO。

原文摘要 · Abstract (English)

Within the domain of large language models, reinforcement fine-tuning algorithms necessitate the generation of a complete reasoning trajectory beginning from the input query, which incurs significant computational overhead during the rollout phase of training. To address this issue, we analyze the impact of different segments of the reasoning path on the correctness of the final result and, based on these insights, propose Reinforcement Fine-Tuning with Partial Reasoning Optimization (RPO), a plug-and-play reinforcement fine-tuning algorithm. Unlike traditional reinforcement fine-tuning algorithms that generate full reasoning paths, RPO trains the model by generating suffixes of the reasoning path using experience cache. During the rollout phase of training, RPO reduces token generation in this phase by approximately 95%, greatly lowering the theoretical time overhead. Compared with full-path reinforcement fine-tuning algorithms, RPO reduces the training time of the 1.5B model by 90% and the 7B model by 72%. At the same time, it can be integrated with typical algorithms such as GRPO and DAPO, enabling them to achieve training acceleration while maintaining performance comparable to the original algorithms. Our code is open-sourced at https://github.com/yhz5613813/RPO.

强化学习微调加速推理优化

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