用更少数据更快训练大模型,效果还更好
LEAD: Iterative Data Selection for Efficient LLM Instruction Tuning
- 在训练循环内估算样本价值,无需额外推理
- 仅用2.5%数据,性能提升6.1%-10.8%
- 适合资源有限但追求高效训练的研究者
指令微调已成为提升大语言模型能力与对齐性的关键范式。然而,现有迭代式模型感知数据选择方法计算开销巨大,需反复进行全数据集模型推理以估计样本效用,造成显著效率瓶颈。本文提出LEAD,一种高效的迭代数据选择框架,可完全在标准训练循环内估算样本效用,避免昂贵的额外推理。核心是实例级动态不确定性(IDU),一个理论驱动的效用函数,结合即时训练损失、基于梯度的损失变化近似及历史损失信号的指数平滑。为高效扩展至大规模数据集,LEAD采用两阶段粗到精选择策略:先通过多臂赌博机机制自适应优先处理高信息量聚类,再使用IDU进行细粒度高价值样本筛选。在四个不同基准上的大量实验表明,LEAD显著优于现有最佳方法,在仅使用2.5%训练数据的情况下,平均性能提升6.1%-10.8%,整体训练时间减少5-10倍。
原文摘要 · Abstract (English)
Instruction tuning has emerged as a critical paradigm for improving the capabilities and alignment of large language models (LLMs). However, existing iterative model-aware data selection methods incur significant computational overhead, as they rely on repeatedly performing full-dataset model inference to estimate sample utility for subsequent training iterations, creating a fundamental efficiency bottleneck. In this paper, we propose LEAD, an efficient iterative data selection framework that accurately estimates sample utility entirely within the standard training loop, eliminating the need for costly additional model inference. At its core, LEAD introduces Instance-Level Dynamic Uncertainty (IDU), a theoretically grounded utility function combining instantaneous training loss, gradient-based approximation of loss changes, and exponential smoothing of historical loss signals. To further scale efficiently to large datasets, LEAD employs a two-stage, coarse-to-fine selection strategy, adaptively prioritizing informative clusters through a multi-armed bandit mechanism, followed by precise fine-grained selection of high-utility samples using IDU. Extensive experiments across four diverse benchmarks show that LEAD significantly outperforms state-of-the-art methods, improving average model performance by 6.1%-10.8% while using only 2.5% of the training data and reducing overall training time by 5-10x.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。