acceptodds
Under review as a conference paper at ICLR 2027

Inverse-Free Second-Order Momentum for Decentralized Riemannian Stochastic Optimization on Compact Submanifolds

Abstract

We study decentralized stochastic optimization on compact embedded submanifolds and propose decentralized projected Riemannian stochastic Hessian-corrected momentum (DPRSHCM). The method pairs a correction for the full ambient variation of the Riemannian gradient with a projected iterate difference, yielding a quadratic remainder as tangent spaces change. It combines this correction with gradient tracking and metric projection, using one stochastic gradient and one Hessian–vector product per node per iteration. With a diminishing stepsize and momentum parameters independent of the iteration budget, DPRSHCM achieves per-node oracle complexity for reducing both the expected squared Riemannian gradient norm and mean squared consensus error below . A constant stepsize chosen for a prescribed budget removes the logarithmic factor. With exact local gradients and stochastic Hessian–vector products, fixed parameters achieve the same joint criterion in iterations. Experiments on PCA and matrix completion demonstrate the method's effectiveness and the computational benefits of reusing gradient computations in matrix completion.

Then back it, or bet against it.

Related papers

Open the market on this paper to see 7 more related papers.