arXiv:2510.16882cs.LGcs.AI2025-10中稿 · paper被引 13

提出UDS框架,动态筛选高效多样数据,加速LLM微调。

Utility-Diversity Aware Online Batch Selection for LLM Supervised Fine-tuning

  • 用核范数捕捉数据价值与样本内多样性
  • 仅需少量内存存储历史样本,实现快速多样性评估
  • 无需参考模型或验证集,训练速度比全量数据快40%以上

监督微调(SFT)是将大语言模型适配下游任务的常用方法。实际中,使用全量数据进行SFT计算成本高,且易过拟合或放大偏差。这促使数据筛选技术的发展,旨在优先选择最有价值的数据。本文研究在线批次选择方法,即在训练过程中动态打分并过滤样本。然而现有主流方法往往仅依赖数据效用而忽视多样性,依赖外部资源如参考模型或验证集,且额外增加训练时间。为此,本文提出UDS(Utility-Diversity Sampling)框架,通过逻辑值矩阵的核范数同时捕捉数据效用与样本内多样性,并利用轻量级记忆缓冲区对历史样本进行低维嵌入比较,高效估计样本间多样性。该设计无需外部资源和额外反向传播,保障计算效率。在多个基准测试上,UDS在不同数据预算下均优于当前最优在线选择方法,且显著缩短训练时间,相较全量数据微调提速超40%。代码已开源。

原文摘要 · Abstract (English)

Supervised fine-tuning (SFT) is a commonly used technique to adapt large language models (LLMs) to downstream tasks. In practice, SFT on a full dataset is computationally expensive and sometimes suffers from overfitting or bias amplification. This facilitates the rise of data curation in SFT, which prioritizes the most valuable data to optimze. This work studies the online batch selection family that dynamically scores and filters samples during the training process. However, existing popular methods often (i) rely merely on the utility of data to select a subset while neglecting other crucial factors like diversity, (ii) rely on external resources such as reference models or validation sets, and (iii) incur extra training time over full-dataset training. To address these limitations, this work develops UDS (Utility-Diversity Sampling), a framework for efficient online batch selection in SFT. UDS leverages the nuclear norm of the logits matrix to capture both data utility and intra-sample diversity, while estimating inter-sample diversity through efficient low-dimensional embedding comparisons with a lightweight memory buffer of historical samples. Such a design eliminates the need for external resources and unnecessary backpropagation, securing computational efficiency. Experiments on multiple benchmarks demonstrate that UDS consistently outperforms state-of-the-art online batch selection methods under varying data budgets, and significantly reduces training time compared to full-dataset fine-tuning. Code is available at https://github.com/gfyddha/UDS.

LLM微调数据筛选在线学习效率优化

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