通过零/一层渐进扩展,大幅降低训练算力消耗同时保持高精度。
Scaling depth capacity via zero/one-layer model expansion
- 新层初始化与超参数迁移策略提升训练效率
- GPT2实验节省约80%算力,实现5倍加速
- 适合大规模模型训练与资源受限场景
模型深度在深度学习中是一把双刃剑:更深的模型精度更高,但计算成本也更大。为高效训练大规模模型,渐进式训练(又称模型扩展)在训练过程中逐步增加模型容量,显著降低计算开销而性能损失极小。本文从优化理论和特征学习视角研究大模型深度扩展,提出零/一层渐进训练策略,系统分析新层初始化、超参数传递、学习率调度及扩展时机。例如,在GPT2上应用该方法可节省约80%算力,相当于实现约5倍加速,且损失水平接近全量训练的60层70亿参数模型,表现出损失上的混合特性。在LLAMA3和DeepSeekV3模型上的缩放规律表明,算力效率提升达3~5倍,且规模越大优势越明显。
原文摘要 · Abstract (English)
Model depth is a double-edged sword in deep learning: deeper models achieve higher accuracy but require higher computational cost. To efficiently train models at scale, progressive training (also known as model expansion) scales up model capacity during training and significantly reduces computation with little performance degradation. In this work, we study the depth expansion of large-scale models through the lens of optimization theory and feature learning, offering insights on the initialization of new layers, hyperparameter transfer, learning rate schedule, and timing of model expansion. Specifically, we propose zero/one-layer progressive training to achieve an optimal tradeoff between computation and loss, with a comprehensive ablations on our expansion strategy. For example, zero/one-layer progressive training on GPT2 can save $\approx 80\%$ compute, or equivalently achieve an $\approx 5\times$ acceleration, while attaining a loss comparable to that of a fully trained 60-layer model with 7B parameters, thus demonstrating a mixing behavior in terms of loss. Furthermore, scaling laws on LLAMA3 and DeepSeekV3 models show a $3\sim 5\times$ improvement in compute efficiency, with an increasing advantage at larger scales.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。