Proving the Lottery Ticket Hypothesis for Convolutional Neural Networks
Arthur da Cunha, Emanuele Natale, Laurent Viennot
Abstract
The lottery ticket hypothesis states that a randomly-initialized neural network contains a small subnetwork which, when trained in isolation, can compete with the performance of the original network. Recent theoretical works proved an even stronger version: every sufficiently overparameterized (dense) neural network contains a subnetwork that, even without training, achieves accuracy comparable to that of the trained large network. These works left as an open problem to extend the result to convolutional neural networks (CNNs).In this work we provide such generalization by showing that, with high probability, it is possible to approximate any CNN by pruning a random CNN whose size is larger by a logarithmic factor.
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.
Your agent calls
Lunesearch_papers
Free to start. No credit card required.
Terminal
Install the CLIlune papers get 0fbb2280-040d-4f10-9963-f3bb459dc404Cited by top-tier papers10
- Why Random Pruning Is All We Need to Start SparseAdvait Harshal Gadhikar, Sohom Mukherjee, Rebekka BurkholzICML 2023 · 33 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
- Batch normalization is sufficient for universal function approximation in CNNsRebekka BurkholzICLR 2024 · 8 citations
- Optimal Eye Surgeon: Finding image priors through sparse generators at initializationAvrajit Ghosh, Xitong Zhang, Kenneth K. Sun, Qing Qu et al.ICML 2024 · 6 citations
Related papers
- Logarithmic Pruning is All You NeedLaurent Orseau, Marcus Hutter, Omar RivasplataNeurIPS 2020 · 102 citations
- Proving the Lottery Ticket Hypothesis: Pruning is All You NeedEran Malach, Gilad Yehudai, Shai Shalev-Shwartz, Ohad ShamirICML 2020 · 327 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 Existence of Universal Lottery TicketsRebekka Burkholz, Nilanjana Laha, Rajarshi Mukherjee, Alkis GotovosICLR 2022 · 38 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
