OpenMathInstruct-1: A 1.8 Million Math Instruction Tuning Dataset

Shubham Toshniwal,Ivan Moshkov,Sean Narenthiran,Daria Gitman,Fei Jia,Igor Gitman

2024-02-16

Abstract:Recent work has shown the immense potential of synthetically generated datasets for training large language models (LLMs), especially for acquiring targeted skills. Current large-scale math instruction tuning datasets such as MetaMathQA (Yu et al., 2024) and MAmmoTH (Yue et al., 2024) are constructed using outputs from closed-source LLMs with commercially restrictive licenses. A key reason limiting the use of open-source LLMs in these data generation pipelines has been the wide gap between the mathematical skills of the best closed-source LLMs, such as GPT-4, and the best open-source LLMs. Building on the recent progress in open-source LLMs, our proposed prompting novelty, and some brute-force scaling, we construct OpenMathInstruct-1, a math instruction tuning dataset with 1.8M problem-solution pairs. The dataset is constructed by synthesizing code-interpreter solutions for GSM8K and MATH, two popular math reasoning benchmarks, using the recently released and permissively licensed Mixtral model. Our best model, OpenMath-CodeLlama-70B, trained on a subset of OpenMathInstruct-1, achieves a score of 84.6% on GSM8K and 50.7% on MATH, which is competitive with the best gpt-distilled models. We release our code, models, and the OpenMathInstruct-1 dataset under a commercially permissive license.

Computation and Language,Artificial Intelligence,Machine Learning

What problem does this paper attempt to address?

This paper aims to address the problem of training data for mathematical reasoning models. Current large-scale mathematical instruction fine-tuning datasets, such as MetaMathQA and MAmmoTH, rely on data generated by closed-source language models with commercial license restrictions. Due to the significant gap in mathematical capabilities between open-source language models and the best closed-source models (such as GPT-4), their use in data generation pipelines is limited. In this paper, the authors construct a large-scale mathematical instruction fine-tuning dataset called OpenMathInstruct-1, consisting of 1.8 million question-answer pairs, using the recently released and license-friendly Mixtral model. They achieve performance on GSM8K and MATH benchmarks comparable to the best gpt-refined model with their OpenMath-CodeLlama-70B model trained on this dataset. Additionally, OpenMathInstruct-1 is the first large-scale dataset with a permissive license, allowing unrestricted use in future work. The paper also explores different prompt strategies to enhance data quality and model performance, and ultimately releases the code, models, and dataset.