atlas

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

DataComp-LM: In search of the next generation of training sets for language models

NeurIPS 2024 · Large language models › Language, safety and interpretability · approach: LLM pruning and scaling data · 10 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 DCLM, a controlled testbed for language-model training-data curation with a 240T-token Common Crawl pool, fixed OpenLM training recipes at five scales, and 53 evaluations, and uses it to build DCLM-baseline via fastText model-based filtering.

Problem

Comparisons of training datasets for language models are confounded by differing architectures, compute, and hyperparameters, and recent open-weight models do not disclose their training data, so it is unclear which data curation strategies work best.

Approach

A standardized benchmark where participants curate data from a fixed raw pool (DCLM-Pool) and are scored by training models with a fixed recipe at multiple compute scales; the baseline finds that a fastText classifier trained on instruction-formatted (OpenHermes 2.5 + ELI5) positives and RefinedWeb negatives, keeping the top 10%, gives the best data.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use fastText trained on OH-2.5 + ELI5 for quality filtering instead of the other tested filters1B-1x data curation · 1B-1xCore: 29.2 -> 30.2Helped
Extract text with resiliparse rather than using WET filesRefinedWeb heuristic-filtered data · 1B-1xCore: 20.7 -> 24.1Helped
Use OH-2.5 + ELI5 rather than Wikipedia as fastText positive examples (ablation)fastText quality filtering · 7B-1xCore: 35.7 -> 41.0Helped
Use OH-2.5 + ELI5 rather than OpenWebText2 as fastText positive examples (ablation)fastText quality filtering · 7B-1xCore: 34.7 -> 41.0Helped
Use OH-2.5 + ELI5 rather than GPT-3 Approx as fastText positive examples (ablation)fastText quality filtering · 7B-1xCore: 37.5 -> 41.0Helped
Keep the top 10% rather than top 15% of fastText-scored documents (ablation)fastText OH-2.5 + ELI5 filtering · 7B-1xCore: 39.8 -> 41.0Helped
Keep the top 10% rather than top 20% of fastText-scored documents (ablation)fastText OH-2.5 + ELI5 filtering · 7B-1xCore: 38.7 -> 41.0Helped
Mix RPJ extras into C4 (ablation)C4 · 1B-1xCore: +2.2Helped
Mix RPJ extras into RPJ CC only (ablation)RPJ CC only · 1B-1xCore: +1.7Helped
Mix RPJ extras into RefinedWeb (ablation)RefinedWeb · 1B-1xCore: +1.4Helped
Mix RPJ extras into DCLM-baseline (ablation)DCLM-baseline · 1B-1xCore: -1.2Hurt
Remove detected MMLU and Hellaswag overlaps (ablation)DCLM-baseline · 7B-2xMMLU: 51.8 -> 52.7No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: DCLM-Pool DCLM-baseline C4 RefinedWeb RedPajama Dolma-V1 OpenHermes 2.5 ELI5 StarCoder ProofPile2 Wikipedia OpenWebText2

Benchmarks: MMLU 5-shot CORE (22 tasks) EXTENDED (53 tasks) HellaSwag ARC-E AlpacaEval 2.0

Metrics: accuracy centered accuracy win-rate (LC) Pearson correlation

Baselines: C4 RefinedWeb RedPajama Dolma-V1.7 FineWeb-Edu Llama 3 8B Mistral-7B-v0.3 MAP-Neo OLMo-1.7 Gemma-7B Llama 2 7B DeepSeek 7B

Builds on: OpenLM RefinedWeb DataComp LLM-Foundry Chinchilla scaling laws

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