atlas

Research map / Large language models / Instruction and fine-tuning

OpenMathInstruct-1: A 1.8 Million Math Instruction Tuning Dataset

NeurIPS 2024 · Large language models › Instruction and fine-tuning · approach: Calibration and knowledge in LLM tuning · 3 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Builds OpenMathInstruct-1, a permissively licensed 1.8M-pair math instruction dataset by synthesizing code-interpreter solutions for GSM8K and MATH with Mixtral and masked-text-solution prompting, then fine-tunes CodeLlama/Mistral/Llama-2 models on it.

Problem

Math instruction-tuning datasets such as MetaMathQA and MAmmoTH rely on closed-source GPT models with restrictive licenses, and open-source LLMs have lower training-set coverage of correct solutions than GPT-4.

Approach

Use reference text solutions converted into masked text solutions as few-shot prompts (preventing shortcut answers) to raise coverage of correct code-interpreter solutions from an open model (Mixtral), plus fair round-robin downsampling and code-preferred selection.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use fair rather than naive downsampling (ablation)OpenMath fine-tuning data · GSM8K validation subsetaccuracy: 74.3 -> 75.3Helped
use fair rather than naive downsampling (ablation)OpenMath fine-tuning data · MATH validation subsetaccuracy: 35.0 -> 37.0Helped
use masked-text prompting rather than default prompting (ablation)OpenMath data synthesis · GSM8K validation subsetaccuracy: 73.8 -> 77.7Helped
use masked-text prompting rather than default prompting (ablation)OpenMath data synthesis · MATH validation subsetaccuracy: 36.9 -> 37.4Helped
increase fine-tuning dataset size from 128K to 512K (ablation)Mistral-7B OpenMath model · GSM8K validation subsetaccuracy: 75.3 -> 81.0Helped
increase fine-tuning dataset size from 128K to 512K (ablation)Mistral-7B OpenMath model · MATH validation subsetaccuracy: 37.0 -> 41.6Helped
use subject-specific rather than default prompts (ablation)OpenMath MATH data synthesis · MATH training set coveragecoverage: 80.1 -> 80.1No clear effect
use subject-specific rather than default prompts (ablation)OpenMath MATH fine-tuning data · MATH validation subset, greedy decodingaccuracyHurt
use subject-specific rather than default prompts (ablation)OpenMath MATH fine-tuning data · MATH validation subset, self-consistency with four samplesaccuracyHelped
use code-preferential solution selection (ablation)OpenMath MATH fine-tuning data · MATH validation subset, greedy decodingaccuracyHelped
use code-preferential solution selection (ablation)OpenMath MATH fine-tuning data · MATH validation subset, self-consistency decodingaccuracyHurt
use masked-text rather than unmasked reference solutions in prompts (ablation)Mixtral solution synthesis · MATH training set coveragecoverage: 80.1% -> 85.9%Helped

Surprising findings

Limitations

Experimental setup

Datasets: GSM8K MATH

Benchmarks: GSM8K MATH GSM-Hard SVAMP TabMWP ASDiv MAWPS

Metrics: accuracy training set coverage (pass@k)

Baselines: GPT-4 (Code Interpreter) WizardMath MetaMath MAmmoTH ToRA

Builds on: MetaMathQA MAmmoTH ToRA Chain-of-thought

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy