Abstract:In this paper, we study the convergence properties of the Stochastic Gradient Descent (SGD) method for finding a stationary point of a given objective function $J(\cdot)$. The objective function is not required to be convex. Rather, our results apply to a class of ``invex'' functions, which have the property that every stationary point is also a global minimizer. First, it is assumed that $J(\cdot)$ satisfies a property that is slightly weaker than the Kurdyka-Lojasiewicz (KL) condition, denoted here as (KL'). It is shown that the iterations $J(\boldsymbol{\theta}_t)$ converge almost surely to the global minimum of $J(\cdot)$. Next, the hypothesis on $J(\cdot)$ is strengthened from (KL') to the Polyak-Lojasiewicz (PL) condition. With this stronger hypothesis, we derive estimates on the rate of convergence of $J(\boldsymbol{\theta}_t)$ to its limit. Using these results, we show that for functions satisfying the PL property, the convergence rate of both the objective function and the norm of the gradient with SGD is the same as the best-possible rate for convex functions. While some results along these lines have been published in the past, our contributions contain two distinct improvements. First, the assumptions on the stochastic gradient are more general than elsewhere, and second, our convergence is almost sure, and not in expectation. We also study SGD when only function evaluations are permitted. In this setting, we determine the ``optimal'' increments or the size of the perturbations. Using the same set of ideas, we establish the global convergence of the Stochastic Approximation (SA) algorithm under more general assumptions on the measurement error, compared to the existing literature. We also derive bounds on the rate of convergence of the SA algorithm under appropriate assumptions.

On the Discrepancy Principle for Stochastic Gradient Descent

Convergence Rates for Stochastic Approximation: Biased Noise with Unbounded Variance, and Applications

On the Diffusion Approximation of Nonconvex Stochastic Gradient Descent

Stochastic Gradient Descent Revisited

High Probability Convergence Bounds for Non-convex Stochastic Gradient Descent with Sub-Weibull Noise

What Happens after SGD Reaches Zero Loss? --A Mathematical Framework

Stochastic Gradient Descent Method with Convex Penalty for Ill-Posed Problems in Banach Spaces

Stochastic Gradient and Langevin Processes

Towards Noise-adaptive, Problem-adaptive (Accelerated) Stochastic Gradient Descent

Stochastic Gradient Descent for Linear Inverse Problems in Hilbert Spaces.

High Probability Guarantees for Nonconvex Stochastic Gradient Descent with Heavy Tails.

Revisiting the Last-Iterate Convergence of Stochastic Gradient Methods

Stopping Criteria for, and Strong Convergence of, Stochastic Gradient Descent on Bottou-Curtis-Nocedal Functions

Derivatives of Stochastic Gradient Descent in parametric optimization

Stochastic gradient descent with noise of machine learning type. Part I: Discrete time analysis

The Anisotropic Noise in Stochastic Gradient Descent: Its Behavior of Escaping from Sharp Minima and Regularization Effects

Implicit Regularization or Implicit Conditioning? Exact Risk Trajectories of SGD in High Dimensions

Stochastic Gradient Descent in Continuous Time: A Central Limit Theorem

The Regularization Effects of Anisotropic Noise in Stochastic Gradient Descent.

What is the long-run distribution of stochastic gradient descent? A large deviations analysis

Beyond the Edge of Stability via Two-step Gradient Updates