arXiv:2509.03594cs.LGcs.AI2025-09被引 1

用损失曲面几何构造新优化器,自动调节学习率。

The Optimiser Hidden in Plain Sight: Training with the Loss Landscape's Induced Metric

  • 基于损失曲面嵌入高维空间的黎曼度量设计优化器
  • 在低维任务中表现优异,比SOTA方法略优
  • 无需手动调参,适合希望自动调节学习率的研究者

我们提出一类新型神经网络优化器,利用损失曲面嵌入高维空间时自然产生的黎曼度量。这一度量正是常见损失曲面可视化所依赖的基础。通过将这种几何视角落到实处并使用诱导度量,我们开发了一种新优化器,并在多种任务和架构上与SGD、Adam、AdamW及Muon进行对比。实验表明,这类新优化器在低维示例中极为有效,对训练神经网络有轻微提升。该优化器具有理论上的理想性质:在高曲率区域自动降低有效学习率,相当于平滑的梯度裁剪;其中一种变体可视为实现自适应学习率调度,而解耦权重衰减从几何角度看是自然选择。基础方法可应用于任何现有预处理优化方法,其计算复杂度与Adam相当。

原文摘要 · Abstract (English)

We present a class of novel optimisers for training neural networks that makes use of the Riemannian metric naturally induced when the loss landscape is embedded in higher-dimensional space. This is the same metric that underlies common visualisations of loss landscapes. By taking this geometric perspective literally and using the induced metric, we develop a new optimiser and compare it to existing methods, namely: SGD, Adam, AdamW, and Muon, across a range of tasks and architectures. Empirically, we conclude that this new class of optimisers is highly effective in low dimensional examples, and provides slight improvement over state-of-the-art methods for training neural networks. These new optimisers have theoretically desirable properties. In particular, the effective learning rate is automatically decreased in regions of high curvature acting as a smoothed out form of gradient clipping. Similarly, one variant of these optimisers can also be viewed as inducing an effective scheduled learning rate and decoupled weight decay is the natural choice from our geometric perspective. The basic method can be used to modify any existing preconditioning method. The new optimiser has a computational complexity comparable to that of Adam.

优化器几何深度学习损失曲面自适应学习率

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