Exploiting Linear Structure Within Convolutional Networks for Efficient Evaluation
Emily L. DentonWojciech ZarembaJoan BrunaYann LeCunR. Fergus
Develops low-rank tensor approximation methods for convolutional filters that cut deep neural network inference latency in half across CPUs and GPUs while preserving original model accuracy within one percent.
Large convolutional neural networks deliver state-of-the-art accuracy in image recognition, but their evaluation requires millions of floating-point operations and vast memory storage. This high computational burden makes deployment difficult on resource-constrained mobile devices and expensive on large-scale cloud servers that process hundreds of millions of images daily. The article evaluates linear compression techniques designed to speed up test-time model execution and reduce memory overhead by exploiting structural redundancies within the network parameters, all while preserving prediction accuracy.
To achieve this, the article introduces specialized low-rank tensor approximation methods combined with brief fine-tuning of network layers. For the initial convolutional layer, it applies a monochromatic approximation that projects redundant color channels onto a lower-dimensional subspace. For subsequent convolutional layers, it groups filters using balanced biclustering and applies low-rank decompositions, using data-driven distance metrics to preserve critical features. In fully connected layers, standard low-rank matrix decomposition is applied. The approach was validated on a deep 15-layer architecture trained on the ImageNet dataset across standard central processing unit (CPU) and graphics processing unit (GPU) hardware.
Key findings demonstrate that the first two convolutional layers account for the majority of execution time, making them prime targets for optimization. Applying the proposed linear approximations to these layers achieved practical 2x to 2.5x speedups on both CPU and GPU hardware, with less than a 1% drop in classification accuracy. When cascading approximations across multiple layers alongside a single pass of fine-tuning, accuracy remained within 1% of the original uncompressed model. Furthermore, the techniques reduced parameter counts in the convolutional layers by a factor of 3x to 5x and compressed fully connected layers by a factor of 5x to 13x with negligible impact on error rates.
These results demonstrate that organizations can significantly decrease hardware latency, lower cloud power and cooling costs, and deploy sophisticated computer vision models to edge devices without meaningful sacrifices in accuracy. The compression methods are complementary to other acceleration techniques, such as integer quantization and Fourier-domain processing, allowing them to be combined for further performance gains. Decision-makers evaluating computer vision deployments should adopt these linear compression pipelines for performance-critical systems and consider brief retraining passes to restore baseline accuracy. However, stakeholders should note that empirical speedups depend heavily on specific hardware and software implementations, and practical deployment should be tested on target devices before full-scale operational rollout.
- Paper: ImageNet Classification with Deep Convolutional Neural Networks, Alex Krizhevsky et al. (2012). It introduces the foundational deep convolutional neural network architecture whose computationally expensive convolutional layers motivate the need for linear filter approximations and acceleration.
- Paper: Network In Network, Min Lin et al. (2014). It explores structure within convolutional feature spaces and 1x1 convolutions, establishing concepts of inter-channel linear combinations relevant to low-rank filter decomposition.
- Paper: Learning Fast Approximations of Sparse Coding, Karol Gregor et al. (2010). It provides foundational principles for learning fast feed-forward approximations to computationally intensive linear and sparse visual representations.
- Paper: Deep Compression: Compressing Deep Neural Network with Pruning, Trained Quantization and Huffman Coding, Song Han et al. (2015). It extends the paradigm of accelerating convolutional networks by combining structured pruning with trained quantization and entropy coding.
- Paper: Channel Pruning for Accelerating Very Deep Neural Networks, Yihui He et al. (2017). It builds directly on layer-wise linear reconstruction techniques to speed up convolutional inference via channel selection and least-squares feature reconstruction.
- Paper: Pruning Filters for Efficient ConvNets, Hao Li et al. (2016). It advances convolutional acceleration by pruning whole filters to reduce FLOPs on dense hardware without requiring low-rank tensor decomposition.
- Paper: MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications, Andrew G. Howard et al. (2017). It translates the principle of factorizing standard convolutions into linear and spatial stages into an efficient, lightweight architecture using depthwise separable convolutions.
- Paper: Learning Structured Sparsity in Deep Neural Networks, Wei Wen et al. (2016). It provides an alternative regularization approach to discover structured, hardware-friendly linear sparsity directly during network training.
- Paper: GhostNet: More Features From Cheap Operations, Kai Han et al. (2019). It exploits redundancy across convolutional feature maps by generating intrinsic features with standard convolutions and deriving remaining features through cheap linear operations.
- Paper: SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size, Forrest N. Iandola et al. (2016). It redesigns convolutional blocks using 1x1 squeeze layers to reduce parameter and computational redundancy while maintaining AlexNet-level accuracy.
- Paper: Learning Efficient Convolutional Networks through Network Slimming, Zhuang Liu et al. (2017). It proposes an end-to-end channel slimming method via batch normalization scaling factors to automatically prune redundant convolutional channels.
