arXiv:2412.00631cs.LGcs.AI2024-12EMNLP被引 11

用奖励信号选数据,5%样本抵全量训练效果

ROSE: A Reward-Oriented Data Selection Framework for LLM Task-Specific Instruction Tuning

  • 以成对偏好损失为奖励,优化任务相关数据选择
  • 仅用5%数据即达全量训练性能,超越现有方法
  • 适配多数据集与模型架构,通用性强

指令微调凸显了大语言模型在多个领域生成更可控、高效输出的潜力。本文聚焦于特定任务指令微调中的数据选择问题。现有方法主要依赖手工设计的相似性度量,选取与测试数据分布一致的训练数据,目标是降低测试数据上的微调损失,从而提升目标任务性能。然而广泛观察到,大语言模型中的指令微调损失(即下一个词预测的交叉熵损失)与实际任务性能之间往往缺乏单调关系,这种错位削弱了当前数据选择方法的有效性。为此,我们提出ROSE——一种基于奖励的指令数据选择方法,利用成对偏好损失作为奖励信号,优化特定任务指令微调的数据选择。具体而言,ROSE采用影响公式,近似训练数据点相对于少量提示偏好验证集的影响,以筛选最相关的训练数据点。实验结果表明,仅使用5%的训练数据,该方法即可达到与全量数据微调相当的性能,并优于其他先进数据选择方法。定性分析进一步验证了该方法在多个基准数据集和多样模型架构下的鲁棒泛化能力。

原文摘要 · Abstract (English)

Instruction tuning has underscored the significant potential of large language models (LLMs) in producing more human controllable and effective outputs in various domains. In this work, we focus on the data selection problem for task-specific instruction tuning of LLMs. Prevailing methods primarily rely on the crafted similarity metrics to select training data that aligns with the test data distribution. The goal is to minimize instruction tuning loss on the test data, ultimately improving performance on the target task. However, it has been widely observed that instruction tuning loss (i.e., cross-entropy loss for next token prediction) in LLMs often fails to exhibit a monotonic relationship with actual task performance. This misalignment undermines the effectiveness of current data selection methods for task-specific instruction tuning. To address this issue, we introduce ROSE, a novel Reward-Oriented inStruction data sElection method which leverages pairwise preference loss as a reward signal to optimize data selection for task-specific instruction tuning. Specifically, ROSE adapts an influence formulation to approximate the influence of training data points relative to a few-shot preference validation set to select the most task-related training data points. Experimental results show that by selecting just 5\% of the training data using ROSE, our approach can achieve competitive results compared to fine-tuning with the full training dataset, and it surpasses other state-of-the-art data selection methods for task-specific instruction tuning. Our qualitative analysis further confirms the robust generalizability of our method across multiple benchmark datasets and diverse model architectures.

数据选择指令微调奖励机制

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