提出学习率路径切换训练法,高效更新大模型版本。
A Learning Rate Path Switching Training Paradigm for Version Updates of Large Language Models
- 用最大学习率预训练主路径,分叉路径更新新数据。
- 四版本更新总成本降至原方法58%,性能相当。
- 适合需频繁迭代的LLM部署场景,降低算力开销。
由于新数据持续涌现,大语言模型(LLMs)的版本更新已成为必要需求。现有更新范式包括从头预训练(PTFS)和持续预训练(CPT)。初步实验表明,PTFS性能更优,但训练成本高;而CPT成本较低,但性能与成本差距随版本迭代加剧。为探究原因,我们分析了CPT中两个阶段的学习率调整影响:第一阶段初始化检查点时采用大学习率,第二阶段基于该检查点的持续预训练需完整学习率衰减。据此,我们提出学习率路径切换训练范式:一条主路径以最大学习率预训练模型,多条分支路径分别对应不同版本的增量数据更新。大量实验验证该范式的有效性与泛化性。特别地,当训练四个版本模型时,总训练成本仅为PTFS的58%,且保持相当的预训练性能。
原文摘要 · Abstract (English)
Due to the continuous emergence of new data, version updates have become an indispensable requirement for Large Language Models (LLMs). The training paradigms for version updates of LLMs include pre-training from scratch (PTFS) and continual pre-training (CPT). Preliminary experiments demonstrate that PTFS achieves better pre-training performance, while CPT has lower training cost. Moreover, their performance and training cost gaps widen progressively with version updates. To investigate the underlying reasons for this phenomenon, we analyze the effect of learning rate adjustments during the two stages of CPT: preparing an initialization checkpoint and continual pre-training based on this checkpoint. We find that a large learning rate in the first stage and a complete learning rate decay process in the second stage are crucial for version updates of LLMs. Hence, we propose a learning rate path switching training paradigm. Our paradigm comprises one main path, where we pre-train a LLM with the maximal learning rate, and multiple branching paths, each of which corresponds to an update of the LLM with newly-added training data. Extensive experiments demonstrate the effectiveness and generalization of our paradigm. Particularly, when training four versions of LLMs, our paradigm reduces the total training cost to 58% compared to PTFS, while maintaining comparable pre-training performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。