大学习率启动训练,能帮模型找到更优解
Where Do Large Learning Rates Lead Us?
- 仅在略高于收敛阈值的极窄范围内初始学习率能达最优
- 该范围训练出的模型收敛到高质量极小值区域,特征稀疏且相关性强
- 适合关注训练初始化策略与泛化能力关系的研究者
普遍认为以大学习率启动神经网络训练可提升泛化性能。为理解此现象,我们在受控环境下开展实证研究,聚焦两个问题:1)达到最佳性能所需的初始学习率有多大;2)不同学习率下训练模型的关键差异。研究发现,仅有略高于收敛阈值的极窄学习率区间,在后续微调或权重平均后可获得最优结果。通过分析所达极小值的局部几何结构,我们发现该区间能使优化过程落入仅包含高质量极小值的盆地。此外,该学习率区间促使模型学习稀疏特征,集中于任务最相关的部分。而初始学习率过小会导致不稳定的极小值,并同时尝试学习所有特征,造成泛化性能差;过大则无法探测到优质解的盆地,难以从数据中提取有意义模式。
原文摘要 · Abstract (English)
It is generally accepted that starting neural networks training with large learning rates (LRs) improves generalization. Following a line of research devoted to understanding this effect, we conduct an empirical study in a controlled setting focusing on two questions: 1) how large an initial LR is required for obtaining optimal quality, and 2) what are the key differences between models trained with different LRs? We discover that only a narrow range of initial LRs slightly above the convergence threshold lead to optimal results after fine-tuning with a small LR or weight averaging. By studying the local geometry of reached minima, we observe that using LRs from this optimal range allows for the optimization to locate a basin that only contains high-quality minima. Additionally, we show that these initial LRs result in a sparse set of learned features, with a clear focus on those most relevant for the task. In contrast, starting training with too small LRs leads to unstable minima and attempts to learn all features simultaneously, resulting in poor generalization. Conversely, using initial LRs that are too large fails to detect a basin with good solutions and extract meaningful patterns from the data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。