揭示大模型超参数快速迁移的机制,提升调参效率。
Understanding the Mechanisms of Fast Hyperparameter Transfer
- 提出快迁移理论框架,定义高效转移标准
- 证明快迁移比直接调参更省算力,具理论优势
- 发现宽度稳定项决定最优超参,适合大规模训练者
深度学习模型规模扩大使得传统超参数优化成本过高。一种有前景的解决方案是使用尺度感知超参数,可在小规模网格搜索后直接迁移至大模型,仅带来微小性能损失。为理解该策略的原理,本文构建了一个通用概念框架,将超参数迁移定义为‘快’,若其带来的次优性随规模增长消失速度超过有限尺度下的性能差距。我们形式化证明:快迁移等价于计算最优网格搜索中的有效迁移,意味着其长期算力效率高于直接调优。尽管已有实证发现最大更新参数化(μP)在扩展模型宽度时表现出快迁移,但其机制仍不清晰。本文通过构造合成场景表明,该性质高度依赖问题结构——某些情况下可实现理论计算优势,而另一些则无法超越直接调优。为解释实践中观察到的快迁移现象,我们提出假设:优化轨迹可分解为两个部分:(1) 宽度稳定的成分,决定最优超参数;(2) 宽度敏感的成分,随宽度提升改善,但对最优超参数扰动极小。我们在多种设置下(包括大语言模型预训练)提供了实证支持。
原文摘要 · Abstract (English)
The growing scale of deep learning models has rendered standard hyperparameter (HP) optimization prohibitively expensive. A promising solution is the use of scale-aware hyperparameters, which can enable direct transfer of optimal HPs from small-scale grid searches to large models with minimal performance loss. To understand the principles governing such transfer strategy, we develop a general conceptual framework for reasoning about HP transfer across scale, characterizing transfer as fast when the suboptimality it induces vanishes asymptotically faster than the finite-scale performance gap. We show formally that fast transfer is equivalent to useful transfer for compute-optimal grid search, meaning that transfer is asymptotically more compute-efficient than direct tuning. While empirical work has found that the Maximal Update Parameterization ($μ$P) exhibits fast transfer when scaling model width, the mechanisms remain poorly understood. We show that this property depends critically on problem structure by presenting synthetic settings where transfer either offers provable computational advantage or fails to outperform direct tuning even under $μ$P. To explain the fast transfer observed in practice, we conjecture that decomposing the optimization trajectory reveals two contributions to loss reduction: (1) a width-stable component that determines the optimal HPs, and (2) a width-sensitive component that improves with width but weakly perturbs the HP optimum. We present empirical evidence for this hypothesis across various settings, including large language model pretraining.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。