提出新方法预防持续学习中训练失效问题。
A Unified Noise-Curvature View of Loss of Trainability
- 从优化角度分析训练失效成因,引入噪声与曲率双重指标。
- 设计自适应步长调度器,使各层更新量保持在安全阈值内。
- 无需调参即可生成类人工设计的步长衰减轨迹,适合持续学习场景。
训练失效指持续学习中参数更新无法推动优化目标进步,导致准确率停滞或下降。本文通过优化视角分析该现象,发现现有指标(如海森矩阵秩、尖锐度、权重/梯度范数、梯度-参数比、单位符号熵)均无法可靠预测。基于此,我们提出两个互补指标:与批量大小相关的梯度噪声上界和曲率波动控制上界,并将其整合为逐层自适应的噪声阈值,用于约束有效步长。据此设计步长调度器,确保每层参数更新低于该阈值,从而避免训练失效。实验表明,该调度器可提升已有方法(如CReLU、Wasserstein正则化、L2权重衰减)维持的准确率。令人意外的是,其自动生成的步长轨迹无需调参即与人工设计的衰减策略高度相似。
原文摘要 · Abstract (English)
Loss of trainability refers to a phenomenon in continual learning where parameter updates no longer make progress on the optimization objective, so accuracy stalls or degrades as the learning problem changes over time. In this paper, we analyze loss of trainability through an optimization lens and find that the phenomenon is not reliably predicted by existing individual indicators such as Hessian rank, sharpness level, weight or gradient norms, gradient-to-parameter ratios, and unit-sign entropy. Motivated by our analysis, we introduce two complementary indicators: a batch-size-aware gradient-noise bound and a curvature volatility-controlled bound. We then combine these two indicators into a per-layer adaptive noise threshold on the effective step-size that anticipates trainability behavior. Using this insight, we propose a step-size scheduler that keeps each layer's effective parameter update below this bound, thereby avoiding loss of trainability. We demonstrate that our scheduler can improve the accuracy maintained by previously proposed approaches, such as concatenated ReLU (CReLU), Wasserstein regularizer, and L2 weight decay. Surprisingly, our scheduler produces adaptive step-size trajectories that, without tuning, mirror the manually engineered step-size decay schedules.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。