Most Activation Functions Can Win the Lottery Without Excessive Depth
Rebekka Burkholz
Abstract
The strong lottery ticket hypothesis has highlighted the potential for training deep neural networks by pruning, which has inspired interesting practical and theoretical insights into how neural networks can represent functions. For networks with ReLU activation functions, it has been proven that a target network with depth can be approximated by the subnetwork of a randomly initialized neural network that has double the target's depth and is wider by a logarithmic factor. We show that a depth network is sufficient. This result indicates that we can expect to find lottery tickets at realistic, commonly used depths while only requiring logarithmic overparametrization. Our novel construction approach applies to a large class of activation functions and is not limited to ReLUs.
Ask about this paper
Ask your agent about it.
Lune has read the top-tier papers around this one, so every answer names the papers it rests on.
Cited by top-tier papers15
- Spectral Graph Pruning Against Over-Squashing and Over-SmoothingAdarsh Jamadandi, Celia Rubio-Madrigal, Rebekka BurkholzNeurIPS 2024 · 33 citations
- Why Random Pruning Is All We Need to Start SparseAdvait Harshal Gadhikar, Sohom Mukherjee, Rebekka BurkholzICML 2023 · 33 citations
- On the Surprising Effectiveness of Attention Transfer for Vision TransformersAlexander C. Li, Yuandong Tian, Beidi Chen, Deepak Pathak et al.NeurIPS 2024 · 21 citations
- Convolutional and Residual Networks Provably Contain Lottery TicketsRebekka BurkholzICML 2022 · 18 citations
- Masks, Signs, And Learning Rate RewindingAdvait Harshal Gadhikar, Rebekka BurkholzICLR 2024 · 15 citations
Related papers
- Optimal Lottery Tickets via Subset Sum: Logarithmic Over-Parameterization is SufficientAnkit Pensia, Shashank Rajput, Alliot Nagle, Harit Vishwakarma et al.NeurIPS 2020 · 115 citations
- On the Existence of Universal Lottery TicketsRebekka Burkholz, Nilanjana Laha, Rajarshi Mukherjee, Alkis GotovosICLR 2022 · 38 citations
- Logarithmic Pruning is All You NeedLaurent Orseau, Marcus Hutter, Omar RivasplataNeurIPS 2020 · 102 citations
- Proving the Lottery Ticket Hypothesis for Convolutional Neural NetworksArthur da Cunha, Emanuele Natale, Laurent ViennotICLR 2022 · 31 citations
- Polynomially Over-Parameterized Convolutional Neural Networks Contain Structured Strong Winning Lottery TicketsArthur da Cunha, Francesco d'Amore, Emanuele NataleNeurIPS 2023 · 5 citations
