arXiv:2504.20644cs.LG2025-04被引 6

通过多样化文件选择,解决大模型预训练中的维度坍缩问题。

Combatting Dimensional Collapse in LLM Pre-Training Data via Diversified File Selection

  • 基于特征空间去相关性,用贪心算法选择更均匀的文本文件。
  • 在500亿训练预算下,仅用1.5%数据实现优于全量数据的性能。
  • 适合追求高效训练与高泛化能力的大模型研发团队。

为提升大语言模型在有限算力下的训练与样本效率,高质量预训练数据的选择至关重要。现有文件选择方法依赖代理模型评估样本与目标领域(如BookCorpus、Wikipedia)的相似性,但导致特征空间维度坍缩:虽提升特定任务性能,却严重损害通用能力。为此,本文提出去相关文件选择算法DiSF,通过贪心策略使所选文本特征协方差矩阵的特征值分布更均匀,理论分析其在γ-弱子模优化下的近似最优性。我们在TinyLlama架构上构建基准,对120M至1.1B参数模型进行实验。在Harness框架九项任务上,DiSF显著提升整体性能:在SlimPajama中仅需保留590M文件的1.5%(约98.5%节省),于50B训练预算内超越全量数据预训练,实现约1.5倍训练效率与5倍数据效率提升。

原文摘要 · Abstract (English)

Selecting high-quality pre-training data for large language models (LLMs) is crucial for enhancing their overall performance under limited computation budget, improving both training and sample efficiency. Recent advancements in file selection primarily rely on using an existing or trained proxy model to assess the similarity of samples to a target domain, such as high quality sources BookCorpus and Wikipedia. However, upon revisiting these methods, the domain-similarity selection criteria demonstrates a diversity dilemma, i.e.dimensional collapse in the feature space, improving performance on the domain-related tasks but causing severe degradation on generic performance. To prevent collapse and enhance diversity, we propose a DiverSified File selection algorithm (DiSF), which selects the most decorrelated text files in the feature space. We approach this with a classical greedy algorithm to achieve more uniform eigenvalues in the feature covariance matrix of the selected texts, analyzing its approximation to the optimal solution under a formulation of $γ$-weakly submodular optimization problem. Empirically, we establish a benchmark and conduct extensive experiments on the TinyLlama architecture with models from 120M to 1.1B parameters. Evaluating across nine tasks from the Harness framework, DiSF demonstrates a significant improvement on overall performance. Specifically, DiSF saves 98.5% of 590M training files in SlimPajama, outperforming the full-data pre-training within a 50B training budget, and achieving about 1.5x training efficiency and 5x data efficiency.

大模型训练数据选择去相关

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