atlas

Research map / Large language models / Language, safety and interpretability

MathPile: A Billion-Token-Scale Pretraining Corpus for Math

NeurIPS 2024 · Large language models › Language, safety and interpretability · approach: Controlled text generation · 2 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.

Introduces MathPile, a ~9.5B-token math-centric pretraining corpus built from textbooks, arXiv, Wikipedia, ProofWiki, StackExchange and Common Crawl with math-specific cleaning, deduplication and benchmark contamination removal, and shows continual pretraining of Mistral-7B gives gains on math benchmarks.

Problem

Open-source, high-quality, large-scale pretraining corpora for mathematical reasoning are scarce, since prominent math corpora (Minerva, MathMix) are not released and existing open ones are narrow or contaminated.

Approach

Assembles a diverse math-centric corpus from multiple curated sources with source-specific preprocessing, customized per-source language-ID thresholds, math-specific heuristic filters, MinHash deduplication, and early benchmark contamination detection, then validates it via continual pretraining of Mistral-7B under a "less is more" quality principle.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Continual-pre-train on MathPile TextbooksMistral-7B-v0.1 · SAT-MATHaccuracy: 47.27 -> 56.36Helped
Continual-pre-train on MathPile TextbooksMistral-7B-v0.1 · MATHaccuracy: 10.08 -> 12.10Helped
Continual-pre-train on MathPile WikipediaMistral-7B-v0.1 · GSM8Kaccuracy: 47.38 -> 49.96Helped
Continual-pre-train on MathPile WikipediaMistral-7B-v0.1 · AQuAaccuracy: 27.95 -> 35.43Helped
Continual-pre-train on MathPile StackExchangeMistral-7B-v0.1 · GSM8Kaccuracy: 47.38 -> 43.06Hurt
Continual-pre-train on MathPile arXivMistral-7B-v0.1 · MATHaccuracy: 10.08 -> 7.50Hurt
Continual-pre-train on MathPile arXivMistral-7B-v0.1 · GSM8Kaccuracy: 47.38 -> 47.91Helped
Continual-pre-train on MathPile Textbooks, Wikipedia, StackExchange, and Common CrawlMistral-7B-v0.1 · MATHaccuracy: 10.08 -> 11.70Helped
Continual-pre-train on synthetic AMPS dataMistral-7B-v0.1 · GSM8Kaccuracy: 47.38 -> 0.08Hurt
Continual-pre-train on synthetic DM-Mathematics dataMistral-7B-v0.1 · GSM8Kaccuracy: 47.38 -> 0.00Hurt
Continual-pre-train on sampled OpenWebMathMistral-7B-v0.1 · GSM8Kaccuracy: 47.38 -> 43.21Hurt
Skip global data processing and use sampled raw Wikipedia (ablation)Mistral-7B-v0.1 · GSM8Kaccuracy: 47.38 -> 41.92Hurt

Surprising findings

Limitations

Experimental setup

Datasets: MathPile OpenWebMath AMPS DM-Mathematics SlimPajama GSM8K MATH MMLU-STEM AGIEval-SAT-MATH MathQA AQuA

Benchmarks: GSM8K MATH AGIEval-SAT-MATH MMLU-Math MathQA AQuA

Metrics: accuracy

Baselines: OpenWebMath AMPS DM-Mathematics Mistral-7B-v0.1

Builds on: RedPajama SlimPajama OpenWebMath NaturalProofs

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