
arXiv: 2402.03467
We give quantitative estimates for the rate of convergence of Riemannian stochastic gradient descent (RSGD) to Riemannian gradient flow and to a diffusion process, the so-called Riemannian stochastic modified flow (RSMF). Using tools from stochastic differential geometry we show that, in the small learning rate regime, RSGD can be approximated by the solution to the RSMF driven by an infinite-dimensional Wiener process. The RSMF accounts for the random fluctuations of RSGD and, thereby, increases the order of approximation compared to the deterministic Riemannian gradient flow. The RSGD is build using the concept of a retraction map, that is, a cost efficient approximation of the exponential map, and we prove quantitative bounds for the weak error of the diffusion approximation under assumptions on the retraction map, the geometry of the manifold, and the random estimators of the gradient.
FOS: Computer and information sciences, Riemannian stochastic gradient descent, Computer Science - Machine Learning, weak error, Diffusion processes and stochastic analysis on manifolds, Stochastic approximation, Probability (math.PR), diffusion approximation, Primary 62L20, Secondary 58J65, 60J20, 65K05, Machine Learning (stat.ML), supervised learning, Machine Learning (cs.LG), Applications of Markov chains and discrete-time Markov processes on general state spaces (social mobility, learning theory, industrial processes, etc.), Numerical mathematical programming methods, Statistics - Machine Learning, Optimization and Control (math.OC), Riemannian gradient flow, FOS: Mathematics, Riemannian stochastic gradient descent diffusion approximation supervised learning weak error Riemannian gradient flow, Mathematics - Optimization and Control, Mathematics - Probability
FOS: Computer and information sciences, Riemannian stochastic gradient descent, Computer Science - Machine Learning, weak error, Diffusion processes and stochastic analysis on manifolds, Stochastic approximation, Probability (math.PR), diffusion approximation, Primary 62L20, Secondary 58J65, 60J20, 65K05, Machine Learning (stat.ML), supervised learning, Machine Learning (cs.LG), Applications of Markov chains and discrete-time Markov processes on general state spaces (social mobility, learning theory, industrial processes, etc.), Numerical mathematical programming methods, Statistics - Machine Learning, Optimization and Control (math.OC), Riemannian gradient flow, FOS: Mathematics, Riemannian stochastic gradient descent diffusion approximation supervised learning weak error Riemannian gradient flow, Mathematics - Optimization and Control, Mathematics - Probability
| selected citations These citations are derived from selected sources. This is an alternative to the "Influence" indicator, which also reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically). | 0 | |
| popularity This indicator reflects the "current" impact/attention (the "hype") of an article in the research community at large, based on the underlying citation network. | Average | |
| influence This indicator reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically). | Average | |
| impulse This indicator reflects the initial momentum of an article directly after its publication, based on the underlying citation network. | Average |
