arXiv:2604.01279cs.LGcs.AI2026-04

Sven用奇异值分解实现高效自然梯度,加速收敛且内存开销小。

Sven: Singular Value Descent as a Computationally Efficient Natural Gradient Method

  • 将损失按数据点分解,用伪逆求最小范数更新
  • 比标准SGD快,比LBFGS省时,收敛到更低损失
  • 适合科学计算中自定义损失的场景

我们提出Sven(奇异值下降),一种针对神经网络的新优化算法,利用损失函数可分解为单个数据点之和的特性,而非先将整体损失压缩为标量再更新参数。Sven将每个数据点的残差视为需同时满足的条件,使用损失梯度的莫尔-彭罗斯伪逆,求解能最优满足所有条件的最小范数参数更新。实际中通过截断奇异值分解近似伪逆,仅保留前k个主要方向,计算开销仅为SGD的k倍。相比传统自然梯度方法随参数量平方增长的开销,显著降低。Sven可被理解为过参数化情形下的自然梯度推广,在欠参数化极限下恢复自然梯度下降。在回归任务上,Sven显著优于Adam等一阶方法,收敛更快、最终损失更低,且与LBFGS性能相当,但耗时仅为后者的几分之一。我们讨论了其扩展的主要挑战——内存开销,并提出缓解策略。除标准机器学习基准外,预计Sven在科学计算中自定义损失函数分解为多个条件的场景具有天然适用性。

原文摘要 · Abstract (English)

We introduce Sven (Singular Value dEsceNt), a new optimization algorithm for neural networks that exploits the natural decomposition of loss functions into a sum over individual data points, rather than reducing the full loss to a single scalar before computing a parameter update. Sven treats each data point's residual as a separate condition to be satisfied simultaneously, using the Moore-Penrose pseudoinverse of the loss Jacobian to find the minimum-norm parameter update that best satisfies all conditions at once. In practice, this pseudoinverse is approximated via a truncated singular value decomposition, retaining only the $k$ most significant directions and incurring a computational overhead of only a factor of $k$ relative to stochastic gradient descent. This is in comparison to traditional natural gradient methods, which scale as the square of the number of parameters. We show that Sven can be understood as a natural gradient method generalized to the over-parametrized regime, recovering natural gradient descent in the under-parametrized limit. On regression tasks, Sven significantly outperforms standard first-order methods including Adam, converging faster and to a lower final loss, while remaining competitive with LBFGS at a fraction of the wall-time cost. We discuss the primary challenge to scaling, namely memory overhead, and propose mitigation strategies. Beyond standard machine learning benchmarks, we anticipate that Sven will find natural application in scientific computing settings where custom loss functions decompose into several conditions.

优化算法自然梯度奇异值分解高效训练

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