Exploring the Regularity of Sparse Structure in Convolutional Neural Networks

By Huizi Mao, Song Han, Jeff Pool, Wenshuo Li, Xingyu Liu, Yu Wang, William (Bill) J. Dally. 2017

258 citations recorded by the source corpus.

Sparsity helps reduce the computational complexity of deep neural networks by skipping zeros. Taking advantage of sparsity is listed as a high priority in next generation DNN accelerators such as TPU. The structure of sparsity, i.e., the granularity of pruning, affects the efficiency of hardware accelerator design as well as the prediction accuracy. Coarse-grained pruning creates regular sparsity patterns, making it more amenable for hardware acceleration but more challenging to maintain the same accuracy. In this paper we quantitatively measure the trade-off between sparsity regularity and prediction accuracy, providing insights in how to maintain accuracy while having more a more structured sparsity pattern. Our experimental results show that coarse-grained pruning can achieve a sparsity ratio similar to unstructured pruning without loss of accuracy. Moreover, due to the index saving effect, coarse-grained pruning is able to obtain a better compression ratio than fine-grained sparsity at the same accuracy threshold. Based on the recent sparse convolutional neural network accelerator (SCNN), our experiments further demonstrate that coarse-grained sparsity saves about 2x the memory references compared to fine-grained sparsity. Since memory reference is more than two orders of magnitude more expensive than arithmetic operations, the regularity of sparse structure leads to more efficient hardware design.

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

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. Kurt Keutzer
  2. Michael W. Mahoney
  3. A. Gholami
  4. Zi-Mei Yao
  5. Quoc V. Le
  6. Zhen Dong
  7. Luke S. Zettlemoyer
  8. Gao Huang

In Computer Vision and Pattern Recognition

  1. Kurt Keutzer
  2. Michael W. Mahoney
  3. Luc Van Gool
  4. A. Gholami
  5. Alan L. Yuille
  6. Dacheng Tao
  7. Jitendra Malik
  8. Trevor Darrell

In Model Architectures

  1. L. V. D. van der Maaten
  2. Kilian Q. Weinberger
  3. Gao Huang
  4. Wayne W. C. Luk
  5. Ruizhe Zhao
  6. Michael W. Mahoney
  7. Kurt Keutzer
  8. A. Gholami

Citations in the corpus