arXiv:2501.12570cs.CL2025-01ACL被引 306

让复杂推理模型更快更准,通过智能压缩思考过程。

O1-Pruner: Length-Harmonizing Fine-Tuning for O1-Like Reasoning Pruning

  • 先采样评估基础性能,再用强化学习引导模型生成更短推理路径。
  • 在保持高准确率前提下,推理耗时显著降低,数学题答对率更高。
  • 适合需要高效推理的长思考类大模型应用,如智能解题系统。

近期,类似 OpenAI O1 的长思考推理大模型通过模拟人类深入思考复杂问题的方式,显著提升了问题求解能力并取得优异成果。然而,这种推理模式导致推理时间大幅增加,带来显著的计算开销。本文实验表明,此类模型难以根据问题难度合理分配令牌预算,且存在推理冗余。为此,我们提出长度调和微调方法(O1-Pruner),旨在降低推理开销的同时保持精度。该方法首先通过预采样估计模型基线性能,随后采用类似强化学习的微调策略,促使模型在满足准确率约束的前提下生成更短的推理过程。在多个数学推理基准上的实验显示,O1-Pruner 不仅显著减少推理延迟,还实现了更高的准确率,为长思考型大模型的高效部署提供了新思路。代码即将开源于 https://github.com/StarDewXXX/O1-Pruner。

原文摘要 · Abstract (English)

Recently, long-thought reasoning LLMs, such as OpenAI's O1, adopt extended reasoning processes similar to how humans ponder over complex problems. This reasoning paradigm significantly enhances the model's problem-solving abilities and has achieved promising results. However, long-thought reasoning process leads to a substantial increase in inference time. A pressing challenge is reducing the inference overhead of long-thought LLMs while ensuring accuracy. In this paper, we experimentally demonstrate that long-thought reasoning models struggle to effectively allocate token budgets based on problem difficulty and reasoning redundancies. To address this, we propose Length-Harmonizing Fine-Tuning (O1-Pruner), aiming at minimizing reasoning overhead while maintaining accuracy. This effective fine-tuning method first estimates the LLM's baseline performance through pre-sampling and then uses RL-style fine-tuning to encourage the model to generate shorter reasoning processes under accuracy constraints. This allows the model to achieve efficient reasoning with lower redundancy while maintaining accuracy. Experiments on various mathematical reasoning benchmarks show that O1-Pruner not only significantly reduces inference overhead but also achieves higher accuracy, providing a novel and promising solution to this challenge. Our code is coming soon at https://github.com/StarDewXXX/O1-Pruner

推理优化长思考强化学习大模型压缩

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