让训练中动态调整数据权重,提升大模型泛化能力。
Rethinking Data Curation in LLM Training: Online Reweighting Offers Better Generalization than Offline Methods

- 训练时动态重加权样本,基于相似性信号自适应调整学习率
- 在同等算力下,跨基准测试表现优于传统离线方法
- 无需删减数据量,自动引导模型从粗到细学习语义
数据筛选是大语言模型训练中的关键环节,但尚未被充分探索。现有方法如数据选择与混合多采用离线处理,与训练过程分离,导致工程开销大且脆弱:模型或任务变化时需重新运行整个流程。此外,离线方法通过硬过滤或重采样改变数据规模,常牺牲多样性并损害泛化性能。本文提出将数据筛选视为在线重加权问题,即在训练过程中通过损失加权动态调整样本重要性,而非静态预处理。具体提出ADAPT(预训练与微调的自适应数据重加权)框架,利用基于相似性的质量信号,以自适应样本级学习率动态重加权,不改变样本数量。相比离线方法强制固定数据分布,ADAPT充当隐式课程学习器,随模型演化逐步聚焦于更精细的语义差异。在指令微调与大规模预训练实验中,ADAPT持续优于离线选择/混合及已有在线方法,在相同FLOPs下实现更强的跨基准泛化能力。
原文摘要 · Abstract (English)
Data curation is a critical yet under-explored area in large language model (LLM) training. Existing methods, such as data selection and mixing, operate in an offline paradigm, detaching themselves from training. This separation introduces engineering overhead and makes the curation brittle: the entire pipeline must be re-run under model/task shifts. Moreover, offline methods alter data size through hard filtering or resampling, often sacrificing data diversity and harming generalization. We propose to rethink data curation as an online reweighting problem, where sample importance is dynamically adjusted during training via loss weighting rather than static pre-processing. Specifically, we introduce ADAPT (Adaptive Data reweighting for Pretraining and FineTuning), a dynamic online framework that reweights training samples with adaptive per-sample learning rates guided by similarity-based quality signals, without changing the number of training samples. Unlike offline methods that enforce a static data distribution, ADAPT acts as an implicit curriculum learner, progressively shifting focus from coarse-grained patterns to fine-grained semantic distinctions as the model evolves. Experiments on both instruction tuning and large-scale pretraining show that ADAPT consistently outperforms offline selection/mixing and prior online methods, achieving stronger cross-benchmark generalization under equal FLOPs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。