HyperAIHyperAI

Command Palette

Search for a command to run...

Guide du pré-entraîneur pour les données d'entraînement : mesure des effets de l'ancienneté, de la couverture des domaines, de la qualité et de la toxicité des données

Résumé

La conception des données de pré-entraînement est critiquement sous-documentée et souvent guidée par des intuitions non étayées empiriquement. Nous pré-entraînons des modèles sur des données triées (1) à différentes époques de collecte, (2) avec des filtres de toxicité et de qualité variés, et (3) avec des compositions de domaines différentes. Premièrement, nous constatons qu'un décalage temporel entre les données d'évaluation et les données de pré-entraînement entraîne une dégradation des performances, qui n'est pas compensée par l'ajustement fin. Deuxièmement, nous mesurons l'effet des filtres de qualité et de toxicité, révélant un compromis entre les performances sur les référentiels standards et le risque de génération de contenus toxiques. Nous constatons également que les effets des différents types de filtrage ne sont pas prévisibles à partir des caractéristiques des domaines textuels. Troisièmement, nous validons empiriquement que des sources de données hétérogènes, telles que les livres et le web, sont bénéfiques et méritent une plus grande priorisation. À ce jour, ces expérimentations constituent la plus grande étude empirique documentée et publiée sur les effets des données de pré-entraînement. Couvrant 28 modèles uniques de 1,5 milliard de paramètres pré-entraînés à partir de zéro, ces résultats valident, quantifient et exposent de nombreuses intuitions non documentées sur la pré-entraînement du texte, ce qui soutientultimativement des décisions plus éclairées centrées sur les données dans le développement de modèles.

One-sentence Summary

Researchers from MIT, Cornell, Google Research, and other institutions conduct the largest publicly documented empirical study of pretraining data design, training 28 unique 1.5-billion-parameter models from scratch to quantify the effects of data age, domain coverage, quality, and toxicity filters on performance, revealing trade-offs between benchmark scores and toxic generation risk and validating long-held data-centric intuitions.

Key Contributions

  • The study provides controlled empirical evidence on pretraining data curation by training 28 unique 1.5 billion parameter models from scratch with different collection times, toxicity and quality filters, and domain compositions. It quantifies how dataset age, filtering, and source composition shape downstream performance.

  • The analysis shows that temporal shift between pretraining and evaluation data causes performance degradation that finetuning does not fully overcome. This establishes pretraining corpus age as a consequential modeling factor rather than an incidental property.

  • The experiments reveal that quality and toxicity filtering trade off standard benchmark performance against toxic generation risk, while heterogeneous sources such as books and web data provide broad benefits. The results also indicate that the effects of filtering are not reliably predictable from text domain characteristics alone.

Introduction

The authors note that modern language models depend on self supervised pretraining over large text datasets, where developers make consequential choices about collection time, quality and toxicity filtering, and domain composition. These choices strongly shape model behavior, yet many prominent models rarely document why certain curation protocols were chosen or how they affect downstream performance. This documentation gap leaves practitioners to rely on unexamined intuitions and precedents, which can weaken responsible data use, model comparison, and development efficiency.

Prior work has mostly focused on the scale of pretraining data and the number of optimization steps, treating curation policies as implicit conventions rather than measurable model design choices. Existing filtering methods are also applied without a clear understanding of their downstream consequences, while temporal mismatch between pretraining and evaluation data can distort performance comparisons, especially for newer or larger models.

The authors contribute a systematic study of pretraining data curation by analyzing dataset age, content filtering, and domain composition. They combine observational measurements on major corpora, C4 and the Pile, with controlled pretraining experiments across 28 1.5B parameter decoder models. Their results show that temporal mismatch reduces performance, quality filtering can improve capability while increasing toxic generation, toxicity filtering can reduce harmfulness but harm generalization, and diverse high quality sources such as books and web data remain important for strong downstream performance.

Dataset

  • The authors use two public pretraining corpora, C4 and the Pile. Both are subjected to heuristic filtering for English language and content quality, followed by deduplication.

  • They create four C4 variants from Common Crawl snapshots taken in 2013, 2016, 2019, and 2022 to study how the age of pretraining data affects results.

  • They group the Pile's 22 original sources into 9 categories: Common Crawl, OpenWebText, books, Wikipedia, legal, social, biomedical, academic, and code. Each category is ablated one at a time to measure its impact on diverse question answering tasks.

  • They apply a document level quality filter used by PaLM and GLaM, which scores each document from 0 for high quality to 1 for low quality. Documents are filtered at thresholds of 0.975, 0.95, 0.9, and 0.7, and an inverse filter is tested to remove the highest quality documents.

  • They apply toxicity filtering using the Jigsaw Perspective API at thresholds of 0.95, 0.9, 0.7, 0.5, and 0.3, plus an inverse filter that removes the least toxic documents. They also use the original C4 bad words filter, which removes documents containing listed profanity.

  • The paper analyzes dataset characteristics and finds that quality and toxicity are not well aligned. The books subset has higher predicted quality but also more profane, toxic, and sexually explicit content. Technical domains such as PubMed, code, and academic text receive lower predicted quality scores.

  • These datasets are used as base pretraining corpora and as controlled variations for evaluating the effects of data age, source composition, and filtering choices.

Method

The authors design their experiments around a controlled data curation and evaluation pipeline. They start from a large pretraining corpus, apply a document level filter to produce a curated subset, train a language model on that subset, and then finetune and evaluate the model on downstream tasks. This setup allows them to connect specific pretraining data choices to measurable changes in final model performance.

As shown in the figure below:

The pipeline is organized as a sequence of three main operations. First, the full pretraining dataset is represented across several axes, including data source domain, data age, and document level quality or toxicity. Second, the authors subselect data from this full pool by varying one dimension at a time. Third, the selected data variants are used to pretrain models, after which downstream performance is measured. This makes the method a causal style comparison between otherwise similar model training runs, with differences isolated to the curation of the pretraining data.

The data filtering module is central to the method. The authors consider three types of interventions: temporal variation, source composition variation, and filter based variation. For temporal variation, they create multiple snapshots of a web based corpus from different years. For source composition variation, they partition a mixed source corpus into categories such as web, books, encyclopedia, legal, social, biomedical, academic, and code, then remove one source group at a time to observe the downstream impact. This part of the pipeline treats the pretraining corpus as a structured mixture rather than a monolithic text pool.

The quality filtering module assigns each document a scalar quality score and removes documents that exceed a chosen threshold. The authors sweep multiple thresholds to create progressively stronger filtered datasets. They also experiment with the inverse of this filter, removing the highest quality documents instead, which allows them to test whether quality filtering improves performance because it removes bad documents, preserves good documents, or both.

The toxicity filtering module uses a trained classifier that predicts the likelihood that a document contains harmful or offensive content. The authors apply several confidence thresholds and also evaluate an inverse filtering variant that removes documents with the least predicted toxicity. In addition, they test a simpler n-gram based filter that removes any document containing a word from a profanity list. This gives a comparison between classifier-based filtering and rule-based filtering, both of which operate at the document level.

The model training module uses decoder-only Transformer language models. The main experiments rely on a 1.5B parameter model trained with an autoregressive next-token prediction objective. For scaling studies, they also use a much smaller 20M parameter model with the same general architecture family. Training is performed within a T5X style implementation, and all variants share the same basic model design so that differences in evaluation results can be more cleanly attributed to differences in the pretraining data rather than to model architecture or training procedure.

After pretraining, the models are evaluated through downstream finetuning and benchmarking. This evaluation stage is essential to the method because the paper’s central question is not whether filtered data improves validation loss on the pretraining corpus, but whether different curation strategies change task performance after adaptation. In this way, the full pipeline links data selection decisions, filtering thresholds, and data mixture changes directly to downstream model utility.

Experiment

The experiments pretrain models on curated variants of C4 and the Pile, then finetune and evaluate them on English toxicity, question answering, and time-sensitive tasks to study dataset age, quality filtering, toxicity filtering, and domain composition. Results show that temporal misalignment between pretraining and evaluation continues to degrade performance despite relevant finetuning, especially for larger models and when evaluation data is newer than pretraining data. Quality filters generally improve most downstream tasks, though their effects vary by domain, while toxicity filtering creates a trade-off between reducing toxic generation and preserving toxicity identification. Domain ablations indicate that diverse web and book data are particularly important for broad downstream performance, supporting the value of combining heterogeneity, quality, and scale in pretraining corpora.

The evaluation setup examines how corpus age, toxicity and quality filtering, and domain composition affect finetuned model behavior. It links each curation choice to targeted tasks, including temporal degradation, toxic generation, toxicity identification, and question answering across varied domains. These benchmarks are chosen to capture broad dataset ablation effects while acknowledging that toxicity and quality labels remain imperfect proxies. Corpus age is evaluated through temporal datasets that measure degradation as time increases. Toxicity and quality filters are assessed using both toxic generation and toxicity identification tasks. Domain composition is evaluated through question answering benchmarks spanning multiple domains.

The results show that temporal misalignment degrades performance for both finetuning and pretraining, with one year of finetuning drift producing larger losses than one year of pretraining drift. Pretraining-related degradation persists after finetuning and is most visible in news and model-size-dependent settings, while correlations with time gaps are generally strong enough to support a systematic effect. Finetuning-year mismatches cause the largest degradation, while pretraining-year mismatches still lead to notable performance loss. News tasks on the larger model show strong temporal sensitivity, and most time-performance correlations are significant, with a few exceptions.

Quality filtering on C4 generally improves average QA performance, with the strongest gains on Academic and Contrast Set tasks. These gains can occur even when a substantial share of pretraining data is removed, though the effect varies by domain and source quality. Toxicity filtering creates a tradeoff: it can reduce toxic generation but may weaken toxicity identification and harm unrelated QA performance. Stronger quality filtering raises average QA performance and especially benefits lower quality source domains such as Academic. Toxicity filtering does not uniformly improve model behavior; it can hurt unrelated QA while improving only selected toxicity related outcomes.

Removing source domains from the pretraining mixture produces uneven effects across QA evaluations. Removing books yields the clearest average decline, while removing academic text leaves the average near baseline and can improve some QA scores. Web and common-sense QA are broadly sensitive to domain removal, whereas biomedical scores remain relatively stable. Books removal produces the largest overall QA drop Academic removal is the least harmful on average and raises some QA scores Web QA tends to decline after most domain removals Common-sense QA is sensitive to removing several domains Biomedical QA changes little across most removals

The experiments evaluate how corpus age, quality and toxicity filtering, and domain composition influence finetuned model behavior across temporal, toxicity, and question answering benchmarks. Results show that temporal misalignment systematically harms performance, especially when finetuning and pretraining years do not match, with news and larger models most sensitive. Quality filtering generally improves question answering, while toxicity filtering offers only selective benefits and can harm unrelated QA, and domain removal produces uneven effects, with books removal the most damaging and biomedical QA relatively stable.


Créer de l'IA avec l'IA

De l'idée au lancement — accélérez votre développement IA avec le co-codage IA gratuit, un environnement prêt à l'emploi et le meilleur prix pour les GPU.

Codage assisté par IA
GPU prêts à l’emploi
Tarifs les plus avantageux

HyperAI Newsletters

Abonnez-vous à nos dernières mises à jour
Nous vous enverrons les dernières mises à jour de la semaine dans votre boîte de réception à neuf heures chaque lundi matin
Propulsé par MailChimp