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.

Revisiting the Noise Model of Stochastic Gradient Descent

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

Dynamic of Stochastic Gradient Descent with State-Dependent Noise

Revisiting the Characteristics of Stochastic Gradient Noise and Dynamics

Understanding Short-Range Memory Effects in Deep Neural Networks

A Theoretical Analysis of Noise Geometry in Stochastic Gradient Descent

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

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

On the Validity of Modeling SGD with Stochastic Differential Equations (SDEs)

Correlated Noise in Epoch-Based Stochastic Gradient Descent: Implications for Weight Variances

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

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

The Regularization Effects of Anisotropic Noise in Stochastic Gradient Descent.

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

Stochastic Gradient and Langevin Processes

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

Nonlinear Stochastic Gradient Descent and Heavy-tailed Noise: A Unified Framework and High-probability Guarantees

Stochastic Differential Equations models for Least-Squares Stochastic Gradient Descent

Noisy Truncated SGD: Optimization and Generalization

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

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