arXiv:2601.21750cs.LG2026-01被引 8

FISMO优化器融合曲率信息与自适应更新,提升大模型训练效率。

FISMO: Fisher-Structured Momentum-Orthogonalized Optimizer

  • 基于费雪信息几何重构更新,实现结构化预处理。
  • 在图像分类和语言建模上超越现有基线,收敛更快。
  • 适合追求高效训练的大规模神经网络研究者使用。

大规模神经网络训练面临非凸优化问题,优化器选择直接影响收敛性和计算效率。尽管自适应方法如Adam长期主导实践,近期提出的Muon优化器通过正交化动量更新实现了各向同性几何与均匀奇异值,但其严格各向同性会丢失梯度谱中潜在的曲率信息。为此,我们提出FISMO(Fisher-Structured Momentum-Orthogonalized)优化器,将各向同性更新推广为融入各向异性曲率信息的策略,利用克罗内克分解的费雪信息度量将更新形式化为信任域问题,实现对局部损失景观几何的自适应预处理,同时保持计算可扩展性。我们在随机非凸设置下建立了收敛性保证,证明了期望平方梯度范数以$\mathcal{O}(1/\sqrt{T})$速率下降,并通过小批量采样显式刻画了方差降低效果。在图像分类与语言建模基准上的实验表明,FISMO在训练效率和最终性能上均优于现有基线。

原文摘要 · Abstract (English)

Training large-scale neural networks requires solving nonconvex optimization where the choice of optimizer fundamentally determines both convergence behavior and computational efficiency. While adaptive methods like Adam have long dominated practice, the recently proposed Muon optimizer achieves superior performance through orthogonalized momentum updates that enforce isotropic geometry with uniform singular values. However, this strict isotropy discards potentially valuable curvature information encoded in gradient spectra, motivating optimization methods that balance geometric structure with adaptivity. We introduce FISMO (Fisher-Structured Momentum-Orthogonalized) optimizer, which generalizes isotropic updates to incorporate anisotropic curvature information through Fisher information geometry. By reformulating the optimizer update as a trust-region problem constrained by a Kronecker-factored Fisher metric, FISMO achieves structured preconditioning that adapts to local loss landscape geometry while maintaining computational tractability. We establish convergence guarantees for FISMO in stochastic nonconvex settings, proving an $\mathcal{O}(1/\sqrt{T})$ rate for the expected squared gradient norm with explicit characterization of variance reduction through mini-batching. Empirical evaluation on image classification and language modeling benchmarks demonstrates that FISMO achieves superior training efficiency and final performance compared to established baselines.

优化器神经网络曲率信息训练效率

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