Logarithmic Pruning is All You Need
Laurent Orseau, Marcus Hutter, Omar Rivasplata
Abstract
The Lottery Ticket Hypothesis is a conjecture that every large neural network contains a subnetwork that, when trained in isolation, achieves comparable performance to the large network. An even stronger conjecture has been proven recently: Every sufficiently overparameterized network contains a subnetwork that, at random initialization, but without training, achieves comparable accuracy to the trained large network. This latter result, however, relies on a number of strong assumptions and guarantees a polynomial factor on the size of the large network compared to the target function. In this work, we remove the most limiting assumptions of this previous work while providing significantly tighter bounds: the overparameterized network only needs a logarithmic factor (in all variables but depth) number of neurons per weight of the target subnetwork.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 0e2d5192-0cd1-46ac-a081-403eeab2ff2aCited by top-tier papers34
- Monarch: Expressive Structured Matrices for Efficient and Accurate TrainingTri Dao, Beidi Chen, Nimit Sharad Sohoni, Arjun D. Desai et al.ICML 2022 · 125 citations
- Optimal Lottery Tickets via Subset Sum: Logarithmic Over-Parameterization is SufficientAnkit Pensia, Shashank Rajput, Alliot Nagle, Harit Vishwakarma et al.NeurIPS 2020 · 115 citations
- Pixelated Butterfly: Simple and Efficient Sparse training for Neural Network ModelsBeidi Chen, Tri Dao, Kaizhao Liang, Jiaming Yang et al.ICLR 2022 · 94 citations
- A Winning Hand: Compressing Deep Networks Can Improve Out-of-Distribution RobustnessJames Diffenderfer, Brian R. Bartoldson, Shreya Chaganti, Jize Zhang et al.NeurIPS 2021 · 88 citations
- The Combinatorial Brain Surgeon: Pruning Weights That Cancel One Another in Neural NetworksXin Yu, Thiago Serra, Srikumar Ramalingam, Shandian ZheICML 2022 · 60 citations
Builds on2
- Proving the Lottery Ticket Hypothesis: Pruning is All You NeedEran Malach, Gilad Yehudai, Shai Shalev-Shwartz, Ohad ShamirICML 2020 · 327 citations
- Optimal Lottery Tickets via Subset Sum: Logarithmic Over-Parameterization is SufficientAnkit Pensia, Shashank Rajput, Alliot Nagle, Harit Vishwakarma et al.NeurIPS 2020 · 115 citations
Related papers
- Proving the Lottery Ticket Hypothesis for Convolutional Neural NetworksArthur da Cunha, Emanuele Natale, Laurent ViennotICLR 2022 · 31 citations
- Most Activation Functions Can Win the Lottery Without Excessive DepthRebekka BurkholzNeurIPS 2022 · 27 citations
- On the Existence of Universal Lottery TicketsRebekka Burkholz, Nilanjana Laha, Rajarshi Mukherjee, Alkis GotovosICLR 2022 · 38 citations
- Polynomially Over-Parameterized Convolutional Neural Networks Contain Structured Strong Winning Lottery TicketsArthur da Cunha, Francesco d'Amore, Emanuele NataleNeurIPS 2023 · 5 citations
- On the Sparsity of the Strong Lottery Ticket HypothesisEmanuele Natale, Davide Ferré, Giordano Giambartolomei, Frédéric Giroire et al.NeurIPS 2024 · 5 citations
