English

Zeroth-order Riemannian Averaging Stochastic Approximation Algorithms

Optimization and Control 2023-09-28 v1 Machine Learning Machine Learning

Abstract

We present Zeroth-order Riemannian Averaging Stochastic Approximation (\texttt{Zo-RASA}) algorithms for stochastic optimization on Riemannian manifolds. We show that \texttt{Zo-RASA} achieves optimal sample complexities for generating ϵ\epsilon-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 improve the algorithm's practicality by using retractions and vector transport, instead of exponential mappings and parallel transports, thereby reducing per-iteration complexity. Additionally, 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.

Keywords

Cite

@article{arxiv.2309.14506,
  title  = {Zeroth-order Riemannian Averaging Stochastic Approximation Algorithms},
  author = {Jiaxiang Li and Krishnakumar Balasubramanian and Shiqian Ma},
  journal= {arXiv preprint arXiv:2309.14506},
  year   = {2023}
}
R2 v1 2026-06-28T12:32:09.763Z