arXiv:2411.16489cs.CLcs.AI2024-11被引 99

用简单蒸馏法复现O1模型,数学推理能力超越O1-preview。

O1 Replication Journey -- Part 2: Surpassing O1-preview through Simple Distillation, Big Progress or Bitter Lesson?

  • 从O1-API蒸馏数万条长思考链,再微调即可提升性能。
  • 在AIME测试中表现优于O1-preview,仅需少量技术投入。
  • 蒸馏模型泛化能力强,抗幻觉且减少谄媚倾向,适合研究者参考。

本文深入探讨复现OpenAI O1模型能力的当前方法,特别关注知识蒸馏技术的广泛使用但常未公开。通过实验发现,仅需对基础模型进行数十万样本的O1-API蒸馏与监督微调,即可在复杂数学推理任务上超越O1-preview。在美式邀请数学竞赛(AIME)测试中,该方法表现更优,技术复杂度极低。研究还扩展至多任务泛化能力:尽管训练数据仅限数学求解,模型在开放域问答、幻觉控制和安全特性上均表现优异,且微调后显著降低谄媚倾向。本文旨在推动AI研究透明化,批判当前技术声明模糊的风气。工作包括:(1)详细阐述蒸馏过程及其有效性;(2)建立评估与分类复现尝试的基准框架;(3)讨论过度依赖蒸馏的风险。最终得出关键教训:追求更强模型固然重要,但培养基于原理的科研思维更为根本。

原文摘要 · Abstract (English)

This paper presents a critical examination of current approaches to replicating OpenAI's O1 model capabilities, with particular focus on the widespread but often undisclosed use of knowledge distillation techniques. While our previous work explored the fundamental technical path to O1 replication, this study reveals how simple distillation from O1's API, combined with supervised fine-tuning, can achieve superior performance on complex mathematical reasoning tasks. Through extensive experiments, we show that a base model fine-tuned on simply tens of thousands of samples O1-distilled long-thought chains outperforms O1-preview on the American Invitational Mathematics Examination (AIME) with minimal technical complexity. Moreover, our investigation extends beyond mathematical reasoning to explore the generalization capabilities of O1-distilled models across diverse tasks: hallucination, safety and open-domain QA. Notably, despite training only on mathematical problem-solving data, our models demonstrated strong generalization to open-ended QA tasks and became significantly less susceptible to sycophancy after fine-tuning. We deliberately make this finding public to promote transparency in AI research and to challenge the current trend of obscured technical claims in the field. Our work includes: (1) A detailed technical exposition of the distillation process and its effectiveness, (2) A comprehensive benchmark framework for evaluating and categorizing O1 replication attempts based on their technical transparency and reproducibility, (3) A critical discussion of the limitations and potential risks of over-relying on distillation approaches, our analysis culminates in a crucial bitter lesson: while the pursuit of more capable AI systems is important, the development of researchers grounded in first-principles thinking is paramount.

模型蒸馏数学推理O1复现AI透明

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