学习率衰减会浪费高质量数据,调整策略可提升大模型预训练效果
How Learning Rate Decay Wastes Your Best Data in Curriculum-Based LLM Pretraining
- 用质量排序数据+恒定学习率能更好利用优质数据
- 标准学习率衰减让高质量数据优势消失,改进后基准测试提升1.64%
- 适合关注数据质量与优化方法协同设计的研究者
由于高质量数据稀缺,大语言模型常在混合不同质量的数据上训练,即使经过精细筛选。一种自然的提升方式是课程式预训练,即按质量指标从低到高排序数据进行训练。但以往研究发现此类方法提升有限。本文发现关键制约因素在于:质量递增顺序与学习率衰减策略不兼容。实验表明,在恒定学习率下,课程训练显著优于随机打乱;但在标准衰减策略下优势减弱。通过采用更温和的学习率衰减(最终学习率仅略低于峰值)或以模型平均替代学习率衰减(对最后几个检查点加权平均),可有效缓解该问题。结合两种策略,在300亿词、15亿参数模型上,使用多种质量度量,标准基准平均得分提升1.64%,无需额外数据清洗。结果呼吁重新评估课程式预训练,并强调数据课程与优化方法协同设计的重要性。
原文摘要 · Abstract (English)
Due to the scarcity of high-quality data, large language models (LLMs) are often trained on mixtures of data with varying quality levels, even after sophisticated data curation. A natural approach to better leverage high-quality data is curriculum-based pretraining, where the model is trained on data sorted in ascending order of quality as determined by a quality metric. However, prior studies have reported limited improvements from such curriculum-based pretraining strategies. This work identifies a critical factor constraining these methods: the incompatibility between the ascending data quality order and the decaying learning rate (LR) schedule. We find that while curriculum-based training substantially outperforms random shuffling when using a constant LR, its advantage diminishes under standard LR decay schedules. Our experiments show this incompatibility can be mitigated by two simple strategies: (1) employing a more moderate LR decay schedule, where the final LR is only moderately smaller than the peak LR, and (2) replacing LR decay with model averaging, i.e., computing a weighted average of the final few checkpoints. By combining these strategies, we improve the average score on a suite of standard benchmarks by 1.64% over random shuffling, without additional data refinement. Validated on 1.5B-parameter models trained over 30B tokens with various data-quality metrics, our findings call for a re-evaluation of curriculum-based LLM pretraining and underscore the potential of co-designing data curricula with optimization methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。