Recovering from Random Pruning: On the Plasticity of Deep Convolutional Neural Networks

By Deepak Mittal, Shweta Bhardwaj, Mitesh M. Khapra, Balaraman Ravindran. 2018

68 citations recorded by the source corpus.

Recently there has been a lot of work on pruning filters from deep convolutional neural networks (CNNs) with the intention of reducing computations. The key idea is to rank the filters based on a certain criterion (say, l1-norm, average percentage of zeros, etc) and retain only the top ranked filters. Once the low scoring filters are pruned away the remainder of the network is fine tuned and is shown to give performance comparable to the original unpruned network. In this work, we report experiments which suggest that the comparable performance of the pruned network is not due to the specific criterion chosen but due to the inherent plasticity of deep neural networks which allows them to recover from the loss of pruned filters once the rest of the filters are fine-tuned. Specifically, we show counter-intuitive results wherein by randomly pruning 25-50% filters from deep CNNs we are able to obtain the same performance as obtained by using state of the art pruning methods. We empirically validate our claims by doing an exhaustive evaluation with VGG-16 and ResNet-50. Further, we also evaluate a real world scenario where a CNN trained on all 1000 ImageNet classes needs to be tested on only a small set of classes at test time (say, only animals). We create a new benchmark dataset from ImageNet to evaluate such class specific pruning and show that even here a random pruning strategy gives close to state of the art performance. Lastly, unlike existing approaches which mainly focus on the task of image classification, in this work we also report results on object detection. We show that using a simple random pruning strategy we can achieve significant speed up in object detection (74% improvement in fps) while retaining the same accuracy as that of the original Faster RCNN model.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Model Architectures, Efficiency & Compression

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Quoc V. Le
  2. Armand Joulin
  3. Herv'e J'egou
  4. Luke S. Zettlemoyer
  5. Edouard Grave
  6. Mohit Bansal
  7. Jianfeng Gao
  8. Yoshua Bengio

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Ming-Hsuan Yang
  3. Trevor Darrell
  4. Dacheng Tao
  5. Xiaogang Wang
  6. Jitendra Malik
  7. Andrew Zisserman
  8. Yi Yang

In Model Architectures

  1. Herv'e J'egou
  2. Armand Joulin
  3. Edouard Grave
  4. Pierre Stock
  5. Benjamin Graham
  6. Hesham Mostafa
  7. Gao Huang
  8. Quoc V. Le

In Efficiency & Compression

  1. Edouard Grave
  2. Armand Joulin
  3. Angela Fan
  4. Herv'e J'egou
  5. Zhuang Liu
  6. Hesham Mostafa
  7. Xin Wang
  8. Pierre Stock

Citations in the corpus