FIPO通过未来KL影响优化,让大模型推理更深入更长。
FIPO: Eliciting Deep Reasoning with Future-KL Influenced Policy Optimization
- 用未来KL散度动态重加权每个词的奖励,精准识别关键推理节点。
- 在Qwen2.5-32B上将思维链长度从4000提升至超10000token,准确率升至58.0%。
- 适合研究大模型推理增强、强化学习训练机制的开发者与研究者。
我们提出未来KL影响策略优化(FIPO),一种用于突破大语言模型推理瓶颈的强化学习算法。尽管类似GRPO的训练方式可扩展性强,但通常依赖基于结果的奖励(ORM),将全局优势均匀分配给轨迹中每个词元,这种粗粒度信用分配限制了性能上限,无法区分关键逻辑转折点与冗余词元。FIPO通过引入折扣未来KL散度,构建密集优势表达式,根据词元对后续轨迹行为的影响程度进行重加权。实验表明,FIPO使模型突破标准基线的长度停滞。在Qwen2.5-32B上,平均思维链长度从约4000提升至超过10000词元,AIME 2024 Pass@1准确率从50.0%提升至峰值58.0%(收敛于约56.0%),优于DeepSeek-R1-Zero-Math-32B(约47.0%)和o1-mini(约56.0%)。结果表明,建立密集优势形式是演化基于ORM算法、释放基础模型全部推理潜力的关键路径。我们已开源基于verl框架的训练系统。
原文摘要 · Abstract (English)
We present Future-KL Influenced Policy Optimization (FIPO), a reinforcement learning algorithm designed to overcome reasoning bottlenecks in large language models. While GRPO style training scales effectively, it typically relies on outcome-based rewards (ORM) that distribute a global advantage uniformly across every token in a trajectory. We argue that this coarse-grained credit assignment imposes a performance ceiling by failing to distinguish critical logical pivots from trivial tokens. FIPO addresses this by incorporating discounted future-KL divergence into the policy update, creating a dense advantage formulation that re-weights tokens based on their influence on subsequent trajectory behavior. Empirically, FIPO enables models to break through the length stagnation seen in standard baselines. Evaluated on Qwen2.5-32B, FIPO extends the average chain-of-thought length from roughly 4,000 to over 10,000 tokens and increases AIME 2024 Pass@1 accuracy from 50.0% to a peak of 58.0% (converging at approximately 56.0\%). This outperforms both DeepSeek-R1-Zero-Math-32B (around 47.0%) and o1-mini (approximately 56.0%). Our results suggest that establishing dense advantage formulations is a vital path for evolving ORM-based algorithms to unlock the full reasoning potential of base models. We open-source our training system, built on the verl framework.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。