arXiv:2510.22980cs.LGstat.ML2025-10被引 14

谱优化器让模型更均衡学习数据特征,提升在不平衡数据上的泛化能力。

How Muon's Spectral Design Benefits Generalization: A Study on Imbalanced Data

  • 用谱梯度下降模拟优化器,让所有主成分同步学习
  • 在不平衡数据上,谱方法早期即展现更强的分类平衡性
  • 适合关注泛化性能与数据偏差问题的研究者

谱感知矩阵值优化器(如 Muon、Shampoo)在深度学习中的应用日益广泛,推动了对其泛化能力的系统研究。本文以不平衡数据为测试基准,引入规范形式——谱梯度下降(SpecGD),其更新步长为梯度截断SVD分解的$UV^T$。在高斯混合数据模型及线性和双线性模型下,我们证明:与优先学习主导主成分的欧氏梯度下降(GD)不同,SpecGD以相等速率学习所有主成分。这导致在训练初期,谱方法在类别平衡损失上显著领先,且即使在使用自适应步长时仍保持优势。扩展至深层线性模型后,深度进一步放大此效应。我们在多种不平衡数据集上验证了理论结果,对比了实际谱方法(Muon、Shampoo)与欧氏方法及Adam,结果一致表明:谱优化器通过促进数据成分的均衡学习,实现更优泛化性能。

原文摘要 · Abstract (English)

The growing adoption of spectrum-aware matrix-valued optimizers such as Muon and Shampoo in deep learning motivates a systematic study of their generalization properties and, in particular, when they might outperform competitive algorithms. We approach this question by introducing appropriate simplifying abstractions as follows: First, we use imbalanced data as a testbed. Second, we study the canonical form of such optimizers, which is Spectral Gradient Descent (SpecGD) -- each update step is $UV^T$ where $UΣV^T$ is the truncated SVD of the gradient. Third, within this framework we identify a canonical setting for which we precisely quantify when SpecGD outperforms vanilla Euclidean GD. For a Gaussian mixture data model and both linear and bilinear models, we show that unlike GD, which prioritizes learning dominant principal components of the data first, SpecGD learns all principal components of the data at equal rates. We demonstrate how this translates to a growing gap in class balanced loss favoring SpecGD early in training and further show that the gap remains consistent even when the GD counterpart uses adaptive step-sizes via normalization. By extending the analysis to deep linear models, we show that depth amplifies these effects. We empirically verify our theoretical findings on a variety of imbalanced datasets. Our experiments compare practical variants of spectral methods, like Muon and Shampoo, against their Euclidean counterparts and Adam. The results validate our findings that these spectral optimizers achieve superior generalization by promoting a more balanced learning of the data's underlying components.

谱优化泛化能力不平衡数据

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