arXiv:2506.15954cs.LG2025-06被引 4

发现深度模型训练关键期,提前终止耗时操作,大幅降本增效。

One Period to Rule Them All: Identifying Critical Learning Periods in Deep Networks

  • 基于泛化预测定位训练中的关键学习期
  • 主流模型训练时间缩短最高59.67%,碳排放减少59.47%
  • 适合资源受限场景,推动可持续深度学习

关键学习期是深度学习中的重要现象,早期训练阶段对数据增强等训练策略的成功至关重要。现有研究虽证实该现象存在,但缺乏对具体发生时段的精确识别。本文提出一种系统性方法,通过泛化预测机制精准定位深度网络训练中的关键学习期,避免依赖计算密集型正则化技术,采用数据剪枝等手段降低计算开销。在关键期后停止高成本训练策略,显著加速训练过程,实现训练时间、能耗与二氧化碳排放的大幅下降。实验在标准架构与基准上验证了有效性:主流模型训练时间最多减少59.67%,二氧化碳排放下降59.47%,财务成本降低60%,且性能不受影响。本工作深化了对训练动态的理解,为资源受限环境下的高效可持续深度学习提供新范式。代码已开源。

原文摘要 · Abstract (English)

Critical Learning Periods comprehend an important phenomenon involving deep learning, where early epochs play a decisive role in the success of many training recipes, such as data augmentation. Existing works confirm the existence of this phenomenon and provide useful insights. However, the literature lacks efforts to precisely identify when critical periods occur. In this work, we fill this gap by introducing a systematic approach for identifying critical periods during the training of deep neural networks, focusing on eliminating computationally intensive regularization techniques and effectively applying mechanisms for reducing computational costs, such as data pruning. Our method leverages generalization prediction mechanisms to pinpoint critical phases where training recipes yield maximum benefits to the predictive ability of models. By halting resource-intensive recipes beyond these periods, we significantly accelerate the learning phase and achieve reductions in training time, energy consumption, and CO$_2$ emissions. Experiments on standard architectures and benchmarks confirm the effectiveness of our method. Specifically, we achieve significant milestones by reducing the training time of popular architectures by up to 59.67%, leading to a 59.47% decrease in CO$_2$ emissions and a 60% reduction in financial costs, without compromising performance. Our work enhances understanding of training dynamics and paves the way for more sustainable and efficient deep learning practices, particularly in resource-constrained environments. In the era of the race for foundation models, we believe our method emerges as a valuable framework. The repository is available at https://github.com/baunilhamarga/critical-periods

训练优化节能降耗模型效率

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