English

Stochastic gradient descent on Riemannian manifolds

Optimization and Control 2016-11-17 v4 Machine Learning Machine Learning

Abstract

Stochastic gradient descent is a simple approach to find the local minima of a cost function whose evaluations are corrupted by noise. In this paper, we develop a procedure extending stochastic gradient descent algorithms to the case where the function is defined on a Riemannian manifold. We prove that, as in the Euclidian case, the gradient descent algorithm converges to a critical point of the cost function. The algorithm has numerous potential applications, and is illustrated here by four examples. In particular a novel gossip algorithm on the set of covariance matrices is derived and tested numerically.

Keywords

Cite

@article{arxiv.1111.5280,
  title  = {Stochastic gradient descent on Riemannian manifolds},
  author = {Silvere Bonnabel},
  journal= {arXiv preprint arXiv:1111.5280},
  year   = {2016}
}

Comments

A slightly shorter version has been published in IEEE Transactions Automatic Control

R2 v1 2026-06-21T19:40:01.930Z