arXiv:2512.04299cs.LGmath.OC2025-12被引 27

提出判断谱梯度更新是否优于普通梯度的简单条件。

When do spectral gradient updates help in deep learning?

  • 基于参数块的梯度核范数与弗罗贝尼乌斯范数比值和激活稳定秩比较
  • 低稳定秩激活下,谱更新优势随数据维度增大而提升
  • 适用于深层网络与Transformer训练,尤其在大模型场景中

谱梯度方法(如近期流行的Muon优化器)是深度神经网络和Transformer训练的有前景替代方案,但其适用范围尚不明确。本文提出一个简单的逐层判据:当参数块的梯度核范数平方与弗罗贝尼乌斯范数比值大于前向激活的稳定秩时,谱更新比欧氏梯度步长能带来更大损失下降。我们证明,在高斯初始化下,随机特征回归、前馈网络及Transformer模块中的后激活矩阵具有低稳定秩。在尖峰随机特征模型中,经过短暂预热后,欧氏梯度的核-弗罗贝尼乌斯比随数据维度增长,而激活稳定秩保持有界,因此谱更新的优势随维度上升。合成回归实验与NanoGPT规模语言模型训练验证了该预测:中间激活始终低稳定秩,梯度保持高核-弗罗贝尼乌斯比。结果明确了谱梯度方法在深层网络与Transformer训练中的有效条件。

原文摘要 · Abstract (English)

Spectral gradient methods, such as the recently popularized Muon optimizer, are a promising alternative to standard Euclidean gradient descent for training deep neural networks and transformers, but it is still unclear in which regimes they are expected to perform better. We propose a simple layerwise condition that predicts when a spectral update yields a larger decrease in the loss than a Euclidean gradient step. This condition compares, for each parameter block, the squared nuclear-to-Frobenius ratio of the gradient to the stable rank of the incoming activations. To understand when this condition may be satisfied, we first prove that post-activation matrices have low stable rank at Gaussian initialization in random feature regression, feedforward networks, and transformer blocks. In spiked random feature models we then show that, after a short burn-in, the Euclidean gradient's nuclear-to-Frobenius ratio grows with the data dimension while the stable rank of the activations remains bounded, so the predicted advantage of spectral updates scales with dimension. We validate these predictions in synthetic regression experiments and in NanoGPT-scale language model training, where we find that intermediate activations have low-stable-rank throughout training and the corresponding gradients maintain large nuclear-to-Frobenius ratios. Together, these results identify conditions for spectral gradient methods, such as Muon, to be effective in training deep networks and transformers.

谱梯度优化器Transformer深度学习

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