arXiv:2501.12147cs.CLcs.AI2025-01EMNLP被引 20

用新方法选训练数据,让大模型各项能力更均衡。

Improving Influence-based Instruction Tuning Data Selection for Balanced Learning of Diverse Capabilities

  • 先归一化数据影响度,再优先选短板任务的样本。
  • 用15%数据选出来效果比全量数据还强且更均衡。
  • 适合希望模型全面发展的研究者和开发者。

选择合适的训练数据对大语言模型的指令微调至关重要,目标是(1)激发强大能力,(2)在不同任务间实现平衡表现。基于影响度的方法在实现(1)方面有潜力,但常难以达成(2)。我们的系统性研究发现,这种表现不佳源于固有偏差:某些任务本身具有更高影响力,导致数据选择偏向这些任务,不仅损害其他任务表现,反而也削弱了高影响力任务自身性能。为此,我们提出BIDS——一种平衡且具影响力的训练数据选择算法。BIDS首先对训练数据的影响度进行归一化,然后迭代选择对最薄弱任务影响最大的样本。在包含七项基准测试、覆盖五类多样化能力的Llama-3与Mistral-v0.3实验中,BIDS始终优于当前最先进的基于影响度的方法及其他非影响度框架。令人惊讶的是,仅使用15%的BIDS选出的数据,其性能就超过全量数据训练,且表现更加均衡。分析表明,实例级归一化与迭代优化对实现多样化能力的均衡学习至关重要。

原文摘要 · Abstract (English)

Selecting appropriate training data is crucial for instruction fine-tuning of large language models (LLMs), which aims to (1) elicit strong capabilities, and (2) achieve balanced performance across different tasks. Influence-based methods show promise in achieving (1), by estimating the contribution of each training example to the model's predictions, but often struggle with (2). Our systematic investigation reveals that this underperformance can be attributed to an inherent bias, where some tasks intrinsically have greater influence than others. As a result, data selection is often biased towards these tasks, not only hurting the model's performance on others but also, counterintuitively, harming performance on these high-influence tasks themselves. To address this, we propose BIDS, a Balanced and Influential Data Selection algorithm. BIDS first normalizes influence scores of the training data, and then iteratively chooses the training example with the highest influence on the most underrepresented task. Experiments with both Llama-3 and Mistral-v0.3 on seven benchmarks spanning five diverse capabilities show that BIDS consistently outperforms both state-of-the-art influence-based algorithms and other non-influence-based frameworks. Surprisingly, training on a 15% subset selected by BIDS can even outperform full-dataset training with a much more balanced performance. Our analysis highlights the importance of both instance-level normalization and iterative optimization of selected data for balanced learning of diverse capabilities.

指令微调数据选择模型平衡影响度分析

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