Organize the Web: Constructing Domains Enhances Pre-Training Data Curation

TL;DR

This paper introduces WebOrganizer, a framework for organizing web data into domains by topic and format, improving pre-training data curation and model performance through domain-aware data mixing.

Contribution

The paper presents a novel method to automatically annotate and organize web data into domains, enhancing data curation and downstream model performance.

Findings

Domain mixing improves model performance on downstream tasks.

Organizing data by topic and format enhances data curation.

Combining domain insights with quality-based methods boosts effectiveness.

Abstract

Modern language models are trained on large, unstructured datasets consisting of trillions of tokens and obtained by crawling the web. The unstructured nature makes it difficult to reason about their contents and develop systematic approaches to data curation. In this paper, we unpack monolithic web corpora by developing taxonomies of their contents and organizing them into domains. We introduce WebOrganizer, a framework for organizing web pages in terms of both their topic and format. Using these two complementary notions of domains, we automatically annotate pre-training data by distilling annotations from a large language model into efficient classifiers. This allows us to study how data from different domains should be mixed to improve models on downstream tasks, and we show that we can combine insights about effective topics and formats to further boost performance. We demonstrate that our domain mixing also improves existing methods that select data based on quality. Furthermore, we study and compare how quality-based methods will implicitly change the domain mixture. Overall, our work demonstrates that constructing and mixing domains provides a valuable complement to quality-based data curation methods, opening new avenues for effective and insightful pre-training data curation.