Beyond Implicit Bias: The Insignificance of SGD Noise in Online Learning

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Vyas, Nikhil, Morwani, Depen, Zhao, Rosie, Kaplun, Gal, Kakade, Sham, Barak, Boaz
Format: Preprint
Publié: 2023
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909218761605120
author Vyas, Nikhil
Morwani, Depen
Zhao, Rosie
Kaplun, Gal
Kakade, Sham
Barak, Boaz
author_facet Vyas, Nikhil
Morwani, Depen
Zhao, Rosie
Kaplun, Gal
Kakade, Sham
Barak, Boaz
contents The success of SGD in deep learning has been ascribed by prior works to the implicit bias induced by finite batch sizes ("SGD noise"). While prior works focused on offline learning (i.e., multiple-epoch training), we study the impact of SGD noise on online (i.e., single epoch) learning. Through an extensive empirical analysis of image and language data, we demonstrate that small batch sizes do not confer any implicit bias advantages in online learning. In contrast to offline learning, the benefits of SGD noise in online learning are strictly computational, facilitating more cost-effective gradient steps. This suggests that SGD in the online regime can be construed as taking noisy steps along the "golden path" of the noiseless gradient descent algorithm. We study this hypothesis and provide supporting evidence in loss and function space. Our findings challenge the prevailing understanding of SGD and offer novel insights into its role in online learning.
format Preprint
id arxiv_https___arxiv_org_abs_2306_08590
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Beyond Implicit Bias: The Insignificance of SGD Noise in Online Learning
Vyas, Nikhil
Morwani, Depen
Zhao, Rosie
Kaplun, Gal
Kakade, Sham
Barak, Boaz
Machine Learning
The success of SGD in deep learning has been ascribed by prior works to the implicit bias induced by finite batch sizes ("SGD noise"). While prior works focused on offline learning (i.e., multiple-epoch training), we study the impact of SGD noise on online (i.e., single epoch) learning. Through an extensive empirical analysis of image and language data, we demonstrate that small batch sizes do not confer any implicit bias advantages in online learning. In contrast to offline learning, the benefits of SGD noise in online learning are strictly computational, facilitating more cost-effective gradient steps. This suggests that SGD in the online regime can be construed as taking noisy steps along the "golden path" of the noiseless gradient descent algorithm. We study this hypothesis and provide supporting evidence in loss and function space. Our findings challenge the prevailing understanding of SGD and offer novel insights into its role in online learning.
title Beyond Implicit Bias: The Insignificance of SGD Noise in Online Learning
topic Machine Learning
url https://arxiv.org/abs/2306.08590