arXiv:2411.07340cs.LGcs.AI2024-11被引 3

用小模型启动大模型训练,显著降低调参成本。

Warmstarting for Scaling Language Models

  • 从小型模型迁移权重,通过μTransfer实现零样本超参数转移。
  • 缩放权重+零填充+μP初始化,使大模型训练收敛更快且稳定。
  • 适合追求高效预训练的团队,尤其在资源受限时效果突出。

将模型规模扩大以提升性能是当前大语言模型范式的核心策略。大量规模研究带来了新的规律和指导原则。然而,当前大规模数据与模型的训练成本高昂,导致对如何调优并构建这些训练配置的理解不足。一种降低预训练成本的方法是利用更易调优的小模型来热启动大规模训练。本文探讨在热启动条件下,最优超参数的行为是否可保留。我们采用μTransfer框架,探索简单操作以应用理论驱动的零样本超参数迁移方法。研究了影响收敛速度和训练动态稳定性保持的关键因素。结果表明,对小模型权重进行缩放、零填充,并使用μP的缩放初始化对大模型进行扰动,可有效实现μTransfer的热启动。

原文摘要 · Abstract (English)

Scaling model sizes to scale performance has worked remarkably well for the current large language models paradigm. The research and empirical findings of various scaling studies led to novel scaling results and laws that guides subsequent research. High training costs for contemporary scales of data and models result in a lack of thorough understanding of how to tune and arrive at such training setups. One direction to ameliorate the cost of pretraining large models is to warmstart the large-scale training from smaller models that are cheaper to tune. In this work, we attempt to understand if the behavior of optimal hyperparameters can be retained under warmstarting for scaling. We explore simple operations that allow the application of theoretically motivated methods of zero-shot transfer of optimal hyperparameters using μTransfer. We investigate the aspects that contribute to the speedup in convergence and the preservation of stable training dynamics under warmstarting with μTransfer. We find that shrinking smaller model weights, zero-padding, and perturbing the resulting larger model with scaled initialization from μP enables effective warmstarting of $\mut{}$.

模型缩放热启动超参数迁移高效训练

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