Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Conferences, Workshops, Symposiums, and Seminars
  4. Implicit Bias of SGD for Diagonal Linear Networks: a Provable Benefit of Stochasticity
 
conference paper not in proceedings

Implicit Bias of SGD for Diagonal Linear Networks: a Provable Benefit of Stochasticity

Pesme, Scott  
•
Pillaud-Vivien, Loucas  
•
Flammarion, Nicolas  
June 16, 2021
35th Conference on Neural Information Processing Systems (NeurIPS 2021)

Understanding the implicit bias of training algorithms is of crucial importance in order to explain the success of overparametrised neural networks. In this paper, we study the dynamics of stochastic gradient descent over diagonal linear networks through its continuous time version, namely stochastic gradient flow. We explicitly characterise the solution chosen by the stochastic flow and prove that it always enjoys better generalisation properties than that of gradient flow. Quite surprisingly, we show that the convergence speed of the training loss controls the magnitude of the biasing effect: the slower the convergence, the better the bias. To fully complete our analysis, we provide convergence guarantees for the dynamics. We also give experimental results which support our theoretical claims. Our findings highlight the fact that structured noise can induce better generalisation and they help explain the greater performances observed in practice of stochastic gradient descent over gradient descent.

  • Files
  • Details
  • Metrics
Type
conference paper not in proceedings
ArXiv ID

2106.09524

Author(s)
Pesme, Scott  
Pillaud-Vivien, Loucas  
Flammarion, Nicolas  
Date Issued

2021-06-16

Subjects

Implicit bias

•

SGD

Editorial or Peer reviewed

REVIEWED

Written at

EPFL

EPFL units
TML  
Event nameEvent placeEvent date
35th Conference on Neural Information Processing Systems (NeurIPS 2021)

Virtual Conference

December 6-14, 2021

Available on Infoscience
October 7, 2021
Use this identifier to reference this record
https://infoscience.epfl.ch/handle/20.500.14299/181904
Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés