Proving the Lottery Ticket Hypothesis: Pruning is All You Need

  • Eran Malach*
  • , Gilad Yehudai*
  • , Shai Shalev-Shwartz
  • , Ohad Shamir
  • *Corresponding author for this work

Research output: Contribution to journalConference articlepeer-review

54 Scopus citations

Abstract

The lottery ticket hypothesis (Frankle and Carbin, 2018), states that a randomly-initialized network contains a small subnetwork such that, when trained in isolation, can compete with the per- formance of the original network. We prove an even stronger hypothesis (as was also con- jectured in Ramanujan et al., 2019), showing that for every bounded distribution and every tar- get network with bounded weights, a sufficiently over-parameterized neural network with random weights contains a subnetwork with roughly the same accuracy as the target network, without any further training.

Original languageEnglish
JournalProceedings of Machine Learning Research
Volume119
StatePublished - 2020
Event37th International Conference on Machine Learning, ICML 2020 - Virtual, Online
Duration: 13 Jul 202018 Jul 2020

Bibliographical note

Publisher Copyright:
© 2020 by the author(s).

Fingerprint

Dive into the research topics of 'Proving the Lottery Ticket Hypothesis: Pruning is All You Need'. Together they form a unique fingerprint.

Cite this