通过正交分解提升数据选择的多样性,让大模型训练更高效
Learning from the Best, Differently: A Diversity-Driven Rethinking on Data Selection
- 将语言质量、知识质量等多维度评分正交化处理,避免指标互相干扰
- 选中的数据在各维度间重叠不足2%,显著提升多样性
- 适合追求模型泛化能力的开发者和研究者使用
高质量预训练数据对大语言模型至关重要,其中质量体现为事实可靠性与语义价值,多样性则确保覆盖广度与分布异质性。现有方法通常依赖单维或多维评分筛选数据,但直接选取高分样本常导致性能下降,需扩大采样范围才能恢复效果。这种数据评分与下游任务表现之间的非单调关系揭示了根本性偏差:基于评分的方法会压缩相关维度,使高分样本看似优质,却系统性忽略多样性。我们提出正交多样性感知选择(ODiS)算法,在数据选择中同时保障质量与多样性。首先,从语言质量、知识质量、理解难度等多个维度评估数据;接着通过主成分分析(PCA)对多维评分进行去相关,生成正交评价维度;针对每个维度训练基于Roberta的评分器,实现大规模语料的可扩展推理;最后在每个正交维度中选取最高分样本构建训练集。实验表明,ODiS选中的数据跨维度重叠低于2%,证实维度正交性;更重要的是,基于ODiS选数据训练的模型在下游基准上显著优于其他基线,凸显正交、多样性感知数据选择对大模型的必要性。
原文摘要 · Abstract (English)
High-quality pre-training data is crutial for large language models, where quality captures factual reliability and semantic value, and diversity ensures broad coverage and distributional heterogeneity. Existing approaches typically rely on single or multiple-dimensional score-based selection. However, directly selecting top-scored data often degrades performance, and sampling from a broader range is required to recover results. The above non-monotonicity between dataset scores and downstream benchmark results reveals a fundamental bias: score-based methods collapse correlated dimensions, causing top-scored data to appear high-quality while systematically overlooking diversity. We argue that ensuring diversity requires decomposing correlated metrics into orthogonal feature dimensions, from which the top-scored data can be directly selected. Therefore, we proposed the Orthogonal Diversity-Aware Selection (ODiS) algorithm, which preserves both quality and diversity during data selection. First, ODiS evaluates data from multiple dimensions, covering language quality, knowledge quality, and comprehension difficulty. The multi-dimensional scores are then decorrelated via Principal Component Analysis (PCA), yielding orthogonal evaluation dimensions. For each dimension, a Roberta-based scorer is trained to regress the data onto PCA-projected scores, enabling scalable inference on large corpora. Finally, ODiS constructs the training dataset by selecting top-scored data within each orthogonal dimension, thereby ensuring both quality and diversity. Empirical results show that ODiS-selected data exhibit less than 2\% inter-dimension overlap, confirming orthogonality between dimensions. More importantly, models trained with ODiS-selected data significantly outperform other baselines on downstream benchmarks, highlighting the necessity of orthogonal, diversity-aware data selection for LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。