Greedy Optimization Provably Wins the Lottery: Logarithmic Number of Winning Tickets is Enough
Mao Ye, Lemeng Wu, Qiang Liu
Abstract
Despite the great success of deep learning, recent works show that large deep neural networks are often highly redundant and can be significantly reduced in size. However, the theoretical question of how much we can prune a neural network given a specified tolerance of accuracy drop is still open. This paper provides one answer to this question by proposing a greedy optimization based pruning method. The proposed method has the guarantee that the discrepancy between the pruned network and the original network decays with exponentially fast rate w.r.t. the size of the pruned network, under weak assumptions that apply for most practical settings. Empirically, our method improves prior arts on pruning various network architectures including ResNet, MobilenetV2/V3 on ImageNet.
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 6b8033d3-247f-4a27-b9ca-e6344243d2a3Cited by top-tier papers6
- Validating the Lottery Ticket Hypothesis with Inertial Manifold TheoryZeru Zhang, Jiayin Jin, Zijie Zhang, Yang Zhou et al.NeurIPS 2021 · 45 citations
- Data-Efficient Structured Pruning via Submodular OptimizationMarwa El Halabi, Suraj Srinivas, Simon Lacoste-JulienNeurIPS 2022 · 31 citations
- Why Lottery Ticket Wins? A Theoretical Perspective of Sample Complexity on Sparse Neural NetworksShuai Zhang, Meng Wang, Sijia Liu, Pin-Yu Chen et al.NeurIPS 2021 · 28 citations
- A Probabilistic Approach to Neural Network PruningXin Qian, Diego KlabjanICML 2021 · 24 citations
- Efficient Transformer-based 3D Object Detection with Dynamic Token HaltingMao Ye, Gregory P. Meyer, Yuning Chai, Qiang LiuICCV 2023 · 10 citations
Builds on9
- Searching for MobileNetV3Andrew Howard, Ruoming Pang, Hartwig Adam, Quoc V. Le et al.ICCV 2019 · 9,163 citations
- MetaPruning: Meta Learning for Automatic Neural Network Channel PruningZechun Liu, Haoyuan Mu, Xiangyu Zhang, Zichao Guo et al.ICCV 2019 · 633 citations
- Proving the Lottery Ticket Hypothesis: Pruning is All You NeedEran Malach, Gilad Yehudai, Shai Shalev-Shwartz, Ohad ShamirICML 2020 · 327 citations
- Provable Filter Pruning for Efficient Neural NetworksLucas Liebenwein, Cenk Baykal, Harry Lang, Dan Feldman et al.ICLR 2020 · 161 citations
- Good Subnetworks Provably Exist: Pruning via Greedy Forward SelectionMao Ye, Chengyue Gong, Lizhen Nie, Denny Zhou et al.ICML 2020 · 123 citations
Related papers
- Accelerate CNN via Recursive Bayesian PruningYuefu Zhou, Ya Zhang, Yanfeng Wang, Qi TianICCV 2019 · 64 citations
- Data-Independent Neural Pruning via CoresetsBen Mussay, Margarita Osadchy, Vladimir Braverman, Samson Zhou et al.ICLR 2020 · 65 citations
- NeuralScale: Efficient Scaling of Neurons for Resource-Constrained Deep Neural NetworksEugene Lee, Chen-Yi LeeCVPR 2020
- Automatic Channel Pruning with Hyper-parameter Search and Dynamic MaskingBaopu Li, Yanwen Fan, Zhihong Pan, Yuchen Bian et al.ACM MM 2021 · 3 citations
- Convolutional Neural Network Pruning With Structural Redundancy ReductionZi Wang, Chengcheng Li, Xiangyang WangCVPR 2021
