arXiv:2412.14135cs.AIcs.LG2024-12被引 59

从强化学习视角拆解o1成功关键,给出复现路线图。

Scaling of Search and Learning: A Roadmap to Reproduce o1 from Reinforcement Learning Perspective

  • 以强化学习为框架,拆解o1的策略初始化、奖励设计、搜索与学习四要素。
  • 通过搜索生成高质量解,结合更多参数与数据提升模型性能。
  • 适合关注大模型推理能力提升与开源复现的研究者参考。

OpenAI o1 在多项需强推理能力的挑战性任务中达到专家水平,是人工智能的重要里程碑。尽管其核心技术被归因于强化学习,现有基于知识蒸馏等方法的复现尝试受限于教师模型的能力上限。本文从强化学习视角出发,提出实现o1的路线图,聚焦四个核心组件:策略初始化使模型具备类人推理行为,有效探索复杂问题空间;奖励设计通过奖励塑形或建模提供密集有效的信号,指导搜索与学习;搜索在训练和测试阶段生成高质量解,计算资源越多效果越好;学习利用搜索生成的数据优化策略,参数越多、数据越丰富,性能越优。现有开源复现项目可视为该路线图的部分或变体。这些组件共同揭示了学习与搜索如何驱动o1的突破,对大模型发展具有重要启示。

原文摘要 · Abstract (English)

OpenAI o1 represents a significant milestone in Artificial Inteiligence, which achieves expert-level performances on many challanging tasks that require strong reasoning ability.OpenAI has claimed that the main techinique behinds o1 is the reinforcement learining. Recent works use alternative approaches like knowledge distillation to imitate o1's reasoning style, but their effectiveness is limited by the capability ceiling of the teacher model. Therefore, this paper analyzes the roadmap to achieving o1 from the perspective of reinforcement learning, focusing on four key components: policy initialization, reward design, search, and learning. Policy initialization enables models to develop human-like reasoning behaviors, equipping them with the ability to effectively explore solution spaces for complex problems. Reward design provides dense and effective signals via reward shaping or reward modeling, which is the guidance for both search and learning. Search plays a crucial role in generating high-quality solutions during both training and testing phases, which can produce better solutions with more computation. Learning utilizes the data generated by search for improving policy, which can achieve the better performance with more parameters and more searched data. Existing open-source projects that attempt to reproduce o1 can be seem as a part or a variant of our roadmap. Collectively, these components underscore how learning and search drive o1's advancement, making meaningful contributions to the development of LLM.

强化学习大模型推理搜索机制

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