arXiv:2411.14318cs.CL2024-11ACL被引 11

动态调整数据比例,让学得慢的领域多学一点。

Velocitune: A Velocity-based Dynamic Domain Reweighting Method for Continual Pre-training

  • 根据学习速度动态重加权数据,慢学领域优先
  • 在数学、代码推理和命令生成任务上提升性能
  • 适合持续预训练中需平衡多个领域的场景

大规模语言模型的训练依赖多样化的语料库,通常由多种领域混合构成。以往方法多采用静态比例采样或固定调整数据比例,但缺乏对领域自适应持续预训练的系统性处理。为此,我们提出 Velocitune,一种基于学习速度动态评估并重加权数据的新框架:优先分配给学习较慢的领域,避开学习过快的领域,并依据缩放定律设定各领域期望的学习目标,以较低成本实现优化。我们在 CodeLlama 上针对推理任务、以及 Llama3 与 Mistral 在系统命令生成语料上的实验表明,Velocitune 在数学与代码推理及命令行生成任务上均取得性能提升。进一步分析揭示,目标损失预测与数据排序是其有效性的关键因素。

原文摘要 · Abstract (English)

It is well-known that a diverse corpus is critical for training large language models, which are typically constructed from a mixture of various domains. In general, previous efforts resort to sampling training data from different domains with static proportions, as well as adjusting data proportions during training. However, few methods have addressed the complexities of domain-adaptive continual pre-training. To fill this gap, we propose Velocitune, a novel framework dynamically assesses learning velocity and adjusts data proportions accordingly, favoring slower-learning domains while shunning faster-learning ones, which is guided by a scaling law to indicate the desired learning goal for each domain with less associated cost. To evaluate the effectiveness of Velocitune, we conduct experiments in a reasoning-focused dataset with CodeLlama, as well as in a corpus specialised for system command generation with Llama3 and Mistral. Velocitune achieves performance gains in both math and code reasoning tasks and command-line generation benchmarks. Further analysis reveals that key factors driving Velocitune's effectiveness include target loss prediction and data ordering.

持续预训练动态重加权学习速度语言模型

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