arXiv:2410.01560cs.CLcs.AI2024-10ICLR被引 206

开源1400万数学题数据集,让大模型数学推理能力提升16%。

OpenMathInstruct-2: Accelerating AI for Math with Massive Open-Source Instruction Data

  • 用强教师生成数据,比弱学生更有效;简洁解法优于冗长描述。
  • 仅用60万独特题目,使模型在MATH测试中准确率提升至67.8%。
  • 适合想快速复现数学推理模型的开发者和研究者使用。

数学推理仍是大语言模型发展的关键挑战,但当前多数前沿进展因训练数据闭源而无法复现。为构建高质量数学推理微调数据集,我们基于新发布的Llama3.1系列模型开展系统消融实验。结果表明:(a) 解答格式影响显著,过度冗长反而降低微调性能;(b) 强教师生成的数据优于同等规模弱学生生成的数据;(c) 微调对低质量解答具有鲁棒性,可容忍粗粒度过滤;(d) 问题多样性是实现数据规模化增益的关键。基于这些发现,我们构建了OpenMathInstruct-2数据集,包含1400万题解对(约60万唯一问题),规模接近此前最大开源数学数据集的八倍。使用该数据集微调Llama-3.1-8B-Base模型,在MATH测试集上准确率从51.9%提升至67.8%,绝对提升15.9%。为推动开源发展,我们已发布代码、微调模型及数据集,采用商业友好型许可协议。

原文摘要 · Abstract (English)

Mathematical reasoning continues to be a critical challenge in large language model (LLM) development with significant interest. However, most of the cutting-edge progress in mathematical reasoning with LLMs has become \emph{closed-source} due to lack of access to training data. This lack of data access limits researchers from understanding the impact of different choices for synthesizing and utilizing the data. With the goal of creating a high-quality finetuning (SFT) dataset for math reasoning, we conduct careful ablation experiments on data synthesis using the recently released \texttt{Llama3.1} family of models. Our experiments show that: (a) solution format matters, with excessively verbose solutions proving detrimental to SFT performance, (b) data generated by a strong teacher outperforms equally-sized data generated by a weak student model, (c) SFT is robust to low-quality solutions, allowing for imprecise data filtering, and (d) question diversity is crucial for achieving data scaling gains. Based on these insights, we create the OpenMathInstruct-2 dataset, which consists of 14M question-solution pairs ($\approx$ 600K unique questions), making it nearly eight times larger than the previous largest open-source math reasoning dataset. Finetuning the \texttt{Llama-3.1-8B-Base} using OpenMathInstruct-2 outperforms \texttt{Llama3.1-8B-Instruct} on MATH by an absolute 15.9\% (51.9\% $\rightarrow$ 67.8\%). Finally, to accelerate the open-source efforts, we release the code, the finetuned models, and the OpenMathInstruct-2 dataset under a commercially permissive license.

数学推理开源数据大模型微调指令数据

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