We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Riemannian stochastic variance reduced gradient algorithm with retraction and vector transport

Abstract: In recent years, stochastic variance reduction algorithms have attracted considerable attention for minimizing the average of a large but finite number of loss functions. This paper proposes a novel Riemannian extension of the Euclidean stochastic variance reduced gradient (R-SVRG) algorithm to a manifold search space. The key challenges of averaging, adding, and subtracting multiple gradients are addressed with retraction and vector transport. For the proposed algorithm, we present a global convergence analysis with a decaying step size as well as a local convergence rate analysis with a fixed step size under some natural assumptions. In addition, the proposed algorithm is applied to the computation problem of the Riemannian centroid on the symmetric positive definite (SPD) manifold as well as the principal component analysis and low-rank matrix completion problems on the Grassmann manifold. The results show that the proposed algorithm outperforms the standard Riemannian stochastic gradient descent algorithm in each case.
Comments: Published in SIAM Journal on Optimization. Extended and revised version of arXiv:1605.07367
Subjects: Machine Learning (cs.LG); Optimization and Control (math.OC); Machine Learning (stat.ML)
Journal reference: SIAM Journal on Optimization 29 (2019) 1444-1472
DOI: 10.1137/17M1116787
Cite as: arXiv:1702.05594 [cs.LG]
  (or arXiv:1702.05594v3 [cs.LG] for this version)

Submission history

From: Hiroyuki Sato [view email]
[v1] Sat, 18 Feb 2017 10:39:23 GMT (630kb)
[v2] Mon, 10 Apr 2017 17:27:05 GMT (629kb)
[v3] Fri, 31 May 2019 10:03:25 GMT (63kb)

Link back to: arXiv, form interface, contact.