Abstract:We study high-probability convergence guarantees of learning on streaming data in the presence of heavy-tailed noise. In the proposed scenario, the model is updated in an online fashion, as new information is observed, without storing any additional data. To combat the heavy-tailed noise, we consider a general framework of nonlinear stochastic gradient descent (SGD), providing several strong results. First, for non-convex costs and component-wise nonlinearities, we establish a convergence rate arbitrarily close to $\mathcal{O}\left(t^{-\frac{1}{4}}\right)$, whose exponent is independent of noise and problem parameters. Second, for strongly convex costs and component-wise nonlinearities, we establish a rate arbitrarily close to $\mathcal{O}\left(t^{-\frac{1}{2}}\right)$ for the weighted average of iterates, with exponent again independent of noise and problem parameters. Finally, for strongly convex costs and a broader class of nonlinearities, we establish convergence of the last iterate, with a rate $\mathcal{O}\left(t^{-\zeta} \right)$, where $\zeta \in (0,1)$ depends on problem parameters, noise and nonlinearity. As we show analytically and numerically, $\zeta$ can be used to inform the preferred choice of nonlinearity for given problem settings. Compared to state-of-the-art, who only consider clipping, require bounded noise moments of order $\eta \in (1,2]$, and establish convergence rates whose exponents go to zero as $\eta \rightarrow 1$, we provide high-probability guarantees for a much broader class of nonlinearities and symmetric density noise, with convergence rates whose exponents are bounded away from zero, even when the noise has finite first moment only. Moreover, in the case of strongly convex functions, we demonstrate analytically and numerically that clipping is not always the optimal nonlinearity, further underlining the value of our general framework.

Convergence of continuous-time stochastic gradient descent with applications to linear deep neural networks

Convergence of stochastic gradient descent under a local Lojasiewicz condition for deep neural networks

Convergence of Stochastic Gradient Descent in Deep Neural Network

A Convergence Analysis of Gradient Descent for Deep Linear Neural Networks

Novel Convergence Results of Adaptive Stochastic Gradient Descents

Stochastic Gradient Descent in Continuous Time: A Central Limit Theorem

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

Analysis of Stochastic Gradient Descent in Continuous Time

Convergence of SGD for Training Neural Networks with Sliced Wasserstein Losses

Aiming towards the minimizers: fast convergence of SGD for overparametrized problems

Convergence in High Probability of Distributed Stochastic Gradient Descent Algorithms

On the convergence of gradient descent for two layer neural networks

Convergence of Constant Step Stochastic Gradient Descent for Non-Smooth Non-Convex Functions

High-probability Convergence Bounds for Nonlinear Stochastic Gradient Descent Under Heavy-tailed Noise

Continuous-time stochastic gradient descent for optimizing over the stationary distribution of stochastic differential equations

Convergence proof for stochastic gradient descent in the training of deep neural networks with ReLU activation for constant target functions

A Mean-Field Analysis of Neural Stochastic Gradient Descent-Ascent for Functional Minimax Optimization

Non-asymptotic convergence analysis of the stochastic gradient Hamiltonian Monte Carlo algorithm with discontinuous stochastic gradient with applications to training of ReLU neural networks

Tight Nonparametric Convergence Rates for Stochastic Gradient Descent under the Noiseless Linear Model

A Geometric Approach of Gradient Descent Algorithms in Linear Neural Networks

On the Convergence of Gradient Descent for Large Learning Rates