Coordinating Filters for Faster Deep Neural Networks

By Wujie Wen, Cong Xu, Chunpeng Wu, Yandan Wang, Yiran Chen, Hai Helen Li. 2017

143 citations recorded by the source corpus.

Very large-scale Deep Neural Networks (DNNs) have achieved remarkable successes in a large variety of computer vision tasks. However, the high computation intensity of DNNs makes it challenging to deploy these models on resource-limited systems. Some studies used low-rank approaches that approximate the filters by low-rank basis to accelerate the testing. Those works directly decomposed the pre-trained DNNs by Low-Rank Approximations (LRA). How to train DNNs toward lower-rank space for more efficient DNNs, however, remains as an open area. To solve the issue, in this work, we propose Force Regularization, which uses attractive forces to enforce filters so as to coordinate more weight information into lower-rank space1. We mathematically and empirically verify that after applying our technique, standard LRA methods can reconstruct filters using much lower basis and thus result in faster DNNs. The effectiveness of our approach is comprehensively evaluated in ResNets, AlexNet, and GoogLeNet. In AlexNet, for example, Force Regularization gains 2× speedup on modern GPU without accuracy loss and 4:05× speedup on CPU by paying small accuracy degradation. Moreover, Force Regularization better initializes the low-rank DNNs such that the fine-tuning can converge faster toward higher accuracy. The obtained lower-rank DNNs can be further sparsified, proving that Force Regularization can be integrated with state-of-the-art sparsity-based acceleration methods.

Subfields: Artificial Intelligence, Computational Mechanics, Computer Vision and Pattern Recognition

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. Wujie Wen
  2. Mathieu Salzmann
  3. José Manuel Álvarez
  4. Jiashi Feng
  5. Yerlan Idelbayev
  6. Hai Helen Li
  7. M. A. Carreira-Perpiñán
  8. Huanrui Yang

In Computational Mechanics

  1. Ulrich Neumann
  2. Song Han
  3. Yerlan Idelbayev
  4. M. A. Carreira-Perpiñán
  5. Yuan Xie
  6. Wujie Wen
  7. Suya You
  8. Jingdong Wang

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Dacheng Tao
  3. Yi Yang
  4. Jiashi Feng
  5. Jingdong Wang
  6. Xiaogang Wang
  7. Ming-Hsuan Yang
  8. Trevor Darrell

Citations in the corpus