Abstract:Random label noise (or observational noise) widely exists in practical machine learning settings. While previous studies primarily focused on the effects of label noise to the performance of learning, our work intends to investigate the implicit regularization effects of label noise, under mini-batch sampling settings of stochastic gradient descent (SGD), with the assumption that label noise is unbiased. Specifically, we analyze the learning dynamics of SGD over the quadratic loss with unbiased label noise (ULN), where we model the dynamics of SGD as a stochastic differentiable equation with two diffusion terms (namely a doubly stochastic model). While the first diffusion term is caused by mini-batch sampling over the (label-noiseless) loss gradients, as in many other works on SGD (Zhu et al 2019 ICML 7654–63; Wu et al 2020 Int. Conf. on Machine Learning (PMLR) pp 10367–76), our model investigates the second noise term of SGD dynamics, which is caused by mini-batch sampling over the label noise, as an implicit regularizer. Our theoretical analysis finds such an implicit regularizer would favor some convergence points that could stabilize model outputs against perturbations of parameters (namely inference stability ). Though similar phenomenon have been investigated by Blanc et al (2020 Conf. on Learning Theory (PMLR) pp 483–513), our work does not assume SGD as an Ornstein–Uhlenbeck-like process and achieves a more generalizable result with convergence of the approximation proved. To validate our analysis, we design two sets of empirical studies to analyze the implicit regularizer of SGD with unbiased random label noise for deep neural network training and linear regression. Our first experiment studies the noisy self-distillation tricks for deep learning, where student networks are trained using the outputs from well-trained teachers with additive unbiased random label noise. Our experiment shows that the implicit regularizer caused by the label noise tends to select models with improved inference stability. We also carry out experiments on SGD-based linear regression with ULN, where we plot the trajectories of parameters learned in every step and visualize the effects of implicit regularization. The results back up our theoretical findings.

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

The Regularization Effects of Anisotropic Noise in Stochastic Gradient Descent.

A Theoretical Analysis of Noise Geometry in Stochastic Gradient Descent

Dynamic of Stochastic Gradient Descent with State-Dependent Noise

The Noise Geometry of Stochastic Gradient Descent: A Quantitative and Analytical Characterization

Revisiting the Characteristics of Stochastic Gradient Noise and Dynamics

The alignment property of SGD noise and how it helps select flat minima: A stability analysis

Revisiting the Noise Model of Stochastic Gradient Descent

Stochastic Gradient Descent Introduces an Effective Landscape-Dependent Regularization Favoring Flat Solutions

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

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

Stochastic Gradient Descent and Anomaly of Variance-flatness Relation in Artificial Neural Networks

The Limiting Dynamics of SGD: Modified Loss, Phase Space Oscillations, and Anomalous Diffusion

Stochastic Collapse: How Gradient Noise Attracts SGD Dynamics Towards Simpler Subnetworks

Stochastic Gradient Descent with Random Label Noises: Doubly Stochastic Models and Inference Stabilizer

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

An Alternative View: When Does SGD Escape Local Minima?

Noisy Truncated SGD: Optimization and Generalization

On the Unstable Convergence Regime of Gradient Descent

Global Convergence of Noisy Gradient Descent.

Towards Theoretically Understanding Why Sgd Generalizes Better Than Adam in Deep Learning