提出新指标NovelSum,精准衡量指令微调数据多样性。
Measuring Data Diversity for Instruction Tuning: A Systematic Analysis and A Reliable Metric
- 基于样本新颖性定义新度量NovelSum,兼顾差异与信息密度。
- 在真实和模拟数据上相关性达0.97,显著优于现有方法。
- 可指导数据筛选,适合做数据工程或模型优化的研究者。
数据多样性对大语言模型的指令微调至关重要。现有研究虽探索了多种多样性感知的数据选择方法以构建高质量数据集并提升模型性能,但数据多样性的准确定义与测量仍缺乏系统研究,限制了数据工程的清晰指引。为此,我们通过大规模微调实验,系统评估了11种现有多样性度量方法与模型性能的相关性。结果表明,可靠的多样性度量应同时考虑样本间差异与样本空间的信息密度。基于此,我们提出NovelSum,一种基于样本级“新颖性”的新度量。在模拟与真实数据上的实验显示,NovelSum能准确捕捉多样性变化,在指令微调模型性能上达到0.97的相关性,凸显其在数据工程中的价值。进一步地,以NovelSum为优化目标,我们设计了一种贪心的多样性导向数据选择策略,效果超越现有方法,验证了该度量的有效性与实用性。代码已开源:https://github.com/UmeanNever/NovelSum。
原文摘要 · Abstract (English)
Data diversity is crucial for the instruction tuning of large language models. Existing studies have explored various diversity-aware data selection methods to construct high-quality datasets and enhance model performance. However, the fundamental problem of precisely defining and measuring data diversity remains underexplored, limiting clear guidance for data engineering. To address this, we systematically analyze 11 existing diversity measurement methods by evaluating their correlation with model performance through extensive fine-tuning experiments. Our results indicate that a reliable diversity measure should properly account for both inter-sample differences and the information density in the sample space. Building on this, we propose NovelSum, a new diversity metric based on sample-level "novelty." Experiments on both simulated and real-world data show that NovelSum accurately captures diversity variations and achieves a 0.97 correlation with instruction-tuned model performance, highlighting its value in guiding data engineering practices. With NovelSum as an optimization objective, we further develop a greedy, diversity-oriented data selection strategy that outperforms existing approaches, validating both the effectiveness and practical significance of our metric. The code is available at https://github.com/UmeanNever/NovelSum.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。