提出兼顾数据质量与多样性的高效筛选方法,提升大模型预训练效果。
QuaDMix: Quality-Diversity Balanced Data Selection for Efficient LLM Pretraining
- 统一框架联合优化数据质量与多样性,自动调节采样概率。
- 在多个基准上平均性能提升7.2%,优于单独优化质量或多样性的方法。
- 适合追求高效高质量预训练数据筛选的研究者与工程师。
数据质量和多样性是影响大语言模型性能的两个关键指标。现有方法通常分步处理:先过滤高质量数据,再调整比例,忽略了二者之间的内在权衡。本文提出统一的数据选择框架QuaDMix,能在固定训练预算下同时评估每条数据的质量及其对整体数据集多样性的贡献。通过领域分类区分数据点并衡量多样性,结合多维度质量标准,构建参数化采样函数,动态决定每条数据的采样概率。为加速参数搜索,在小模型上进行模拟实验,并借鉴RegMix方法使用LightGBM优化参数。在多种模型和数据集上的实验表明,QuaDMix在多个基准上实现平均7.2%的性能提升,显著优于独立优化质量或多样性的策略,证明了联合平衡的重要性与可行性。
原文摘要 · Abstract (English)
Quality and diversity are two critical metrics for the training data of large language models (LLMs), positively impacting performance. Existing studies often optimize these metrics separately, typically by first applying quality filtering and then adjusting data proportions. However, these approaches overlook the inherent trade-off between quality and diversity, necessitating their joint consideration. Given a fixed training quota, it is essential to evaluate both the quality of each data point and its complementary effect on the overall dataset. In this paper, we introduce a unified data selection framework called QuaDMix, which automatically optimizes the data distribution for LLM pretraining while balancing both quality and diversity. Specifically, we first propose multiple criteria to measure data quality and employ domain classification to distinguish data points, thereby measuring overall diversity. QuaDMix then employs a unified parameterized data sampling function that determines the sampling probability of each data point based on these quality and diversity related labels. To accelerate the search for the optimal parameters involved in the QuaDMix framework, we conduct simulated experiments on smaller models and use LightGBM for parameters searching, inspired by the RegMix method. Our experiments across diverse models and datasets demonstrate that QuaDMix achieves an average performance improvement of 7.2% across multiple benchmarks. These results outperform the independent strategies for quality and diversity, highlighting the necessity and ability to balance data quality and diversity.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。