MetaPruning: Meta Learning for Automatic Neural Network Channel Pruning
Hong Kong University of Science and Technology · Tsinghua University · +3 more institutions
Abstract
In this paper, we propose a novel meta learning approach for automatic channel pruning of very deep neural networks. We first train a PruningNet, a kind of meta network, which is able to generate weight parameters for any pruned structure given the target network. We use a simple stochastic structure sampling method for training the PruningNet. Then, we apply an evolutionary procedure to search for good-performing pruned networks. The search is highly efficient because the weights are directly generated by the trained PruningNet and we do not need any finetuning at search time. With a single PruningNet trained for the target network, we can search for various Pruned Networks under different constraints with…
Citation impact
- FWCI
- 46.02
- Percentile
- 100%
- References
- 108
Authors
7Topics & keywords
- Pruning
- Computer science
- Artificial intelligence
- Artificial neural network
- Machine learning
- Simple (philosophy)
- Meta learning (computer science)
- Deep learning
- Peace, Justice and strong institutions