English

Optimal Regularization Can Mitigate Double Descent

Machine Learning 2021-04-30 v2 Neural and Evolutionary Computing Statistics Theory Machine Learning Statistics Theory

Abstract

Recent empirical and theoretical studies have shown that many learning algorithms -- from linear regression to neural networks -- can have test performance that is non-monotonic in quantities such the sample size and model size. This striking phenomenon, often referred to as "double descent", has raised questions of if we need to re-think our current understanding of generalization. In this work, we study whether the double-descent phenomenon can be avoided by using optimal regularization. Theoretically, we prove that for certain linear regression models with isotropic data distribution, optimally-tuned 2\ell_2 regularization achieves monotonic test performance as we grow either the sample size or the model size. We also demonstrate empirically that optimally-tuned 2\ell_2 regularization can mitigate double descent for more general models, including neural networks. Our results suggest that it may also be informative to study the test risk scalings of various algorithms in the context of appropriately tuned regularization.

Keywords

Cite

@article{arxiv.2003.01897,
  title  = {Optimal Regularization Can Mitigate Double Descent},
  author = {Preetum Nakkiran and Prayaag Venkat and Sham Kakade and Tengyu Ma},
  journal= {arXiv preprint arXiv:2003.01897},
  year   = {2021}
}

Comments

v2: Accepted to ICLR 2021. Minor edits to Intro and Appendix