Zeroth-Order Riemannian Averaging Stochastic Approximation Algorithms

Jiaxiang Li,Krishnakumar Balasubramanian,Shiqian Ma
DOI: https://doi.org/10.1137/23m1605405
IF: 2.763
2024-10-02
SIAM Journal on Optimization
Abstract:SIAM Journal on Optimization, Volume 34, Issue 4, Page 3314-3341, December 2024. We present Zeroth-order Riemannian Averaging Stochastic Approximation (Zo-RASA) algorithms for stochastic optimization on Riemannian manifolds. We show that Zo-RASA achieves optimal sample complexities for generating [math]-approximation first-order stationary solutions using only one-sample or constant-order batches in each iteration. Our approach employs Riemannian moving-average stochastic gradient estimators and a novel Riemannian–Lyapunov analysis technique for convergence analysis. We provably improve the algorithm's practicality by using retractions and vector transport, instead of exponential mappings and parallel transports, thereby reducing per-iteration complexity. To do so, we introduce a novel geometric condition, satisfied by manifolds with bounded second fundamental form, which enables new error bounds for approximating parallel transport with vector transport.
mathematics, applied
What problem does this paper attempt to address?