A Survey on Data Selection for Language Models

Alon Albalak,Yanai Elazar,Sang Michael Xie,Shayne Longpre,Nathan Lambert,Xinyi Wang,Niklas Muennighoff,Bairu Hou,Liangming Pan,Haewon Jeong,Colin Raffel,Shiyu Chang,Tatsunori Hashimoto,William Yang Wang

2024-08-03

Abstract:A major factor in the recent success of large language models is the use of enormous and ever-growing text datasets for unsupervised pre-training. However, naively training a model on all available data may not be optimal (or feasible), as the quality of available text data can vary. Filtering out data can also decrease the carbon footprint and financial costs of training models by reducing the amount of training required. Data selection methods aim to determine which candidate data points to include in the training dataset and how to appropriately sample from the selected data points. The promise of improved data selection methods has caused the volume of research in the area to rapidly expand. However, because deep learning is mostly driven by empirical evidence and experimentation on large-scale data is expensive, few organizations have the resources for extensive data selection research. Consequently, knowledge of effective data selection practices has become concentrated within a few organizations, many of which do not openly share their findings and methodologies. To narrow this gap in knowledge, we present a comprehensive review of existing literature on data selection methods and related research areas, providing a taxonomy of existing approaches. By describing the current landscape of research, this work aims to accelerate progress in data selection by establishing an entry point for new and established researchers. Additionally, throughout this review we draw attention to noticeable holes in the literature and conclude the paper by proposing promising avenues for future research.

Computation and Language,Machine Learning

What problem does this paper attempt to address?

The paper primarily explores the importance and application of data selection methods in language model training. Its core objectives are to address the following issues:

1. **Optimizing Datasets**: How to filter out the optimal dataset from a large amount of raw data to improve model performance, data efficiency, and reduce training costs.

2. **Defining a Unified Framework**: Proposing a unified conceptual framework to classify and compare existing data selection methods, and creating a taxonomy of data selection methods.

3. **Addressing Needs at Different Training Stages**: Studying data selection strategies suitable for various training stages such as language model pre-training, instruction-tuning, alignment, in-context learning, and task-specific fine-tuning.

4. **Considering Multiple Objectives**: Besides improving model performance, data selection also needs to focus on data efficiency, ensuring the integrity of evaluation, reducing bias and toxicity, and the efficiency of selection.

5. **Promoting Knowledge Sharing**: Currently, effective data selection practices are concentrated in a few organizations, and these organizations often do not openly share their research findings and methodologies. Therefore, the paper aims to bridge this knowledge gap and provide an entry point into the field for both new and experienced researchers.

By achieving the above objectives, the paper hopes to accelerate research progress in the field of data selection and provide guidance for future research directions.