arXiv:2410.09335cs.CLcs.AI2024-10EMNLP被引 22

大规模数据微调中,随机选择几乎就足够了。

Rethinking Data Selection at Scale: Random Selection is Almost All You Need

  • 在百万级数据集上,随机选取效果接近甚至超过复杂筛选方法。
  • 数据多样性比单一追求高质量更能提升微调效果。
  • 按词元长度过滤可稳定提升弱模型性能,适合长文本训练。

监督微调(SFT)对大语言模型(LLM)与人类指令对齐至关重要。其核心目标是从大规模数据池中挑选出少量代表性数据子集,使微调结果达到甚至超越使用全量数据的效果。然而,现有数据选择方法多针对小规模数据设计,难以满足真实世界中的大规模SFT需求。本文在两百万规模的数据集上复现了多种无需外部模型辅助的自评分方法,发现几乎所有方法在大规模场景下均无法显著优于随机选择。此外,我们的分析表明,数据多样性比单纯追求高质量数据更为关键。我们还揭示了当前方法在大规模数据上的局限性及其不适用原因。最后,发现按词元长度过滤能提供稳定高效的改进,尤其对较弱的基础模型(如Llama3)在长文本训练中效果显著。

原文摘要 · Abstract (English)

Supervised fine-tuning (SFT) is crucial for aligning Large Language Models (LLMs) with human instructions. The primary goal during SFT is to select a small yet representative subset of training data from the larger pool, such that fine-tuning with this subset achieves results comparable to or even exceeding those obtained using the entire dataset. However, most existing data selection techniques are designed for small-scale data pools, which fail to meet the demands of real-world SFT scenarios. In this paper, we replicated several self-scoring methods those that do not rely on external model assistance on two million scale datasets, and found that nearly all methods struggled to significantly outperform random selection when dealing with such large-scale data pools. Moreover, our comparisons suggest that, during SFT, diversity in data selection is more critical than simply focusing on high quality data. We also analyzed the limitations of several current approaches, explaining why they perform poorly on large-scale datasets and why they are unsuitable for such contexts. Finally, we found that filtering data by token length offers a stable and efficient method for improving results. This approach, particularly when training on long text data, proves highly beneficial for relatively weaker base models, such as Llama3.

数据筛选大模型训练随机选择微调

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。