大规模指令微调数据选择中,简单方法反而更优且高效
Large-Scale Data Selection for Instruction Tuning
- 用预训练模型隐藏层加权均值池化进行数据筛选
- 在580万数据池中选250万样本,性能优于复杂方法
- 适合追求高效高质数据筛选的研究者与工程师
从更大数据池中选择高质量训练数据是指令微调语言模型的关键步骤。尽管现有自动化数据选择方法通常在小规模数据集(约1万样本)上测试,但实际部署的模型常基于数十万至数百万样本训练,且来自更大的数据池。本文系统研究了数据选择方法在大规模场景下的表现,从最大580万样本的数据池中选取最多250万样本,并在7个不同任务上评估。结果表明,许多近期提出的复杂方法在该设置下表现不如随机选择,甚至随数据池扩大而性能下降。而一种基于表示的筛选方法(RDS+),利用预训练语言模型隐藏状态的加权均值池化,始终优于其他方法,且计算效率更高。研究强调应更关注自动化选择方法的可扩展性。代码、数据和模型已开源。
原文摘要 · Abstract (English)
Selecting high-quality training data from a larger pool is a crucial step when instruction-tuning language models, as carefully curated datasets often produce models that outperform those trained on much larger, noisier datasets. Automated data selection approaches for instruction-tuning are typically tested by selecting small datasets (roughly 10k samples) from small pools (100-200k samples). However, popular deployed instruction-tuned models often train on hundreds of thousands to millions of samples, subsampled from even larger data pools. We present a systematic study of how well data selection methods scale to these settings, selecting up to 2.5M samples from pools of up to 5.8M samples and evaluating across 7 diverse tasks. We show that many recently proposed methods fall short of random selection in this setting (while using more compute), and even decline in performance when given access to larger pools of data to select over. However, we find that a variant of representation-based data selection (RDS+), which uses weighted mean pooling of pretrained LM hidden states, consistently outperforms more complex methods across all settings tested -- all whilst being more compute-efficient. Our findings highlight that the scaling properties of proposed automated selection methods should be more closely examined. We release our code, data, and models at https://github.com/hamishivi/automated-instruction-selection.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。