Skip to main navigation Skip to search Skip to main content

Accelerating Very Deep Convolutional Networks for Classification and Detection

  • Xi'an Jiaotong University
  • Microsoft USA

Research output: Contribution to journalArticlepeer-review

747 Scopus citations

Abstract

This paper aims to accelerate the test-time computation of convolutional neural networks (CNNs), especially very deep CNNs [1] that have substantially impacted the computer vision community. Unlike previous methods that are designed for approximating linear filters or linear responses, our method takes the nonlinear units into account. We develop an effective solution to the resulting nonlinear optimization problem without the need of stochastic gradient descent (SGD). More importantly, while previous methods mainly focus on optimizing one or two layers, our nonlinear method enables an asymmetric reconstruction that reduces the rapidly accumulated error when multiple (e.g., ≥ 10) layers are approximated. For the widely used very deep VGG-16 model [1] , our method achieves a whole-model speedup of 4× with merely a 0.3 percent increase of top-5 error in ImageNet classification. Our 4× accelerated VGG-16 model also shows a graceful accuracy degradation for object detection when plugged into the Fast R-CNN detector [2].

Original languageEnglish
Article number7332968
Pages (from-to)1943-1955
Number of pages13
JournalIEEE Transactions on Pattern Analysis and Machine Intelligence
Volume38
Issue number10
DOIs
StatePublished - 1 Oct 2016

Keywords

  • Convolutional neural networks
  • acceleration
  • image classification
  • object detection

Fingerprint

Dive into the research topics of 'Accelerating Very Deep Convolutional Networks for Classification and Detection'. Together they form a unique fingerprint.

Cite this