English

Manifold constrained steepest descent

Optimization and Control 2026-01-30 v1 Machine Learning

Abstract

Norm-constrained linear minimization oracle (LMO)-based optimizers such as spectral gradient descent and Muon are attractive in large-scale learning, but extending them to manifold-constrained problems is nontrivial and often leads to nested-loop schemes that solve tangent-space subproblems iteratively. We propose \emph{Manifold Constrained Steepest Descent} (MCSD), a single-loop framework for optimization over manifolds that selects a norm-induced steepest-descent direction via an LMO applied to the Riemannian gradient, and then returns to the manifold via projection. Under standard smoothness assumptions, we establish convergence guarantees for MCSD and a stochastic momentum variant. We further introduce \emph{SPEL}, the spectral-norm specialization of MCSD on the Stiefel manifold, which admits scalable implementations via fast matrix sign computations. Experiments on PCA, orthogonality-constrained CNNs, and manifold-constrained LLM adapter tuning demonstrate improved stability and competitive performance relative to standard Riemannian baselines and existing manifold-aware LMO methods.

Keywords

Cite

@article{arxiv.2601.21487,
  title  = {Manifold constrained steepest descent},
  author = {Kaiwei Yang and Lexiao Lai},
  journal= {arXiv preprint arXiv:2601.21487},
  year   = {2026}
}

Comments

23 pages, 7 figures, and 5 tables

R2 v1 2026-07-01T09:25:23.445Z