arXiv:2503.11441cs.LG2025-03IJCAI被引 23

用多样性、难度和可靠性筛选数据,小样本也能高效训练大模型指令跟随能力。

D3: Diversity, Difficulty, and Dependability-Aware Data Selection for Sample-Efficient LLM Instruction Tuning

  • 基于三维度评估数据价值:多样性、难度、可靠性,自动识别优质样本。
  • 仅用10%数据即达顶尖性能,实测在淘宝直播场景表现优异。
  • 适合追求高效训练、资源有限的AI研发团队使用。

大型语言模型(LLM)的指令微调近期表明,少量高质量数据可显著提升模型指令遵循能力,优于常因质量差和冗余问题而低效的大规模数据集。但如何从海量数据中自动筛选出有价值子集仍是挑战。本文提出D3方法,基于多样性、难度和依赖性三个维度定义数据价值,并设计评分与选择两阶段流程。评分阶段通过区分度函数衡量样本独特性,采用基于不确定性的预测难度评估样本难易程度,同时引入外部LLM进行可靠性判断。选择阶段构建加权核集合目标,联合优化三类价值,自适应迭代优化选择结果。在公开数据集及真实淘宝直播应用上的实验表明,仅使用全量数据的10%以下,D3即可使模型具备竞争力甚至更优的指令遵循能力。

原文摘要 · Abstract (English)

Recent advancements in instruction tuning for large language models (LLMs) suggest that a small, high-quality dataset can significantly equip LLMs with instruction-following capabilities, outperforming large datasets often burdened by quality and redundancy issues. However, the challenge lies in automatically identifying valuable subsets from large datasets to boost both the effectiveness and efficiency of instruction tuning. In this paper, we first establish data selection criteria based on three distinct aspects of data value: diversity, difficulty, and dependability, and then propose the D3 method comprising two key steps of scoring and selection. Specifically, in the scoring step, we define the diversity function to measure sample distinctiveness and introduce the uncertainty-based prediction difficulty to evaluate sample difficulty by mitigating the interference of context-oriented generation diversity. Additionally, we integrate an external LLM for dependability assessment. In the selection step, we formulate the D3 weighted coreset objective, which jointly optimizes three aspects of data value to solve for the most valuable subset. The two steps of D3 can iterate multiple rounds, incorporating feedback to refine the selection focus adaptively. Experiments on both public datasets and the real-world Taobao Live application demonstrate the effectiveness of D3 in endowing LLMs with competitive or even superior instruction-following capabilities using less than 10\% of the entire dataset.

数据筛选指令微调高效训练LLM

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