Learning Structured Sparsity in Deep Neural Networks
Wei WenChunpeng WuYandan WangYiran ChenHai Li
Introduces Structured Sparsity Learning, a regularization framework that prunes filters, channels, and entire layers to generate hardware-friendly deep neural networks that accelerate CPU and GPU inference while maintaining or improving accuracy.
The article addresses the challenge of deploying large deep neural networks on resource-limited devices, where high computation and memory demands create barriers to practical use. Non-structured sparsity methods often fail to deliver real speed gains because of irregular memory access patterns, while low-rank approximations require repeated decomposition steps and fix layer structures in ways that limit flexibility.
The article set out to develop and test a Structured Sparsity Learning method that applies group Lasso regularization to enforce compact structures across filters, channels, filter shapes, and network depth during training.
The approach involved training and evaluating the method on standard models including LeNet, ConvNet, ResNet, and AlexNet using the MNIST, CIFAR-10, and ImageNet datasets. Structured sparsity was measured through changes in floating-point operations, layer dimensions, and runtime on both CPU and GPU platforms with common libraries, while accuracy was tracked before and after regularization and fine-tuning.
The experiments showed that SSL produced average speedups of 5.1 times on CPU and 3.1 times on GPU for AlexNet convolutional layers, roughly double the gains from non-structured sparsity at similar accuracy levels. On CIFAR-10, depth regularization reduced a 20-layer ResNet to 18 layers while raising accuracy from 91.25 percent to 92.60 percent, exceeding the original 32-layer ResNet. For AlexNet on ImageNet, the method lowered top-1 error by about one percent or maintained accuracy with substantially fewer parameters. Structured sparsity also enabled direct reductions in weight-matrix dimensions for efficient matrix-multiplication routines.
These results indicate that enforcing structured compactness during training yields models that run faster on existing hardware without custom optimizations, while the regularization effect can sometimes improve classification performance. The approach therefore supports faster inference on constrained devices and may reduce both energy use and latency in deployed systems.
The findings support combining SSL with complementary techniques such as low-rank approximation for further gains. Additional testing on newer architectures and broader hardware platforms would strengthen confidence before large-scale adoption.
The work relies on specific published models and datasets from 2016, and depth regularization assumes the presence of shortcut connections; results may vary under different training conditions or without fine-tuning after structure removal.
- Paper: Learning both Weights and Connections for Efficient Neural Networks, Song Han et al. (2015). Introduces foundational non-structured connection pruning and fine-tuning, which directly motivates the source paper's shift toward regularized structured sparsity.
- Paper: Deep Compression: Compressing Deep Neural Network with Pruning, Trained Quantization and Huffman Coding, Song Han et al. (2015). Establishes the three-stage deep compression pipeline that the source contrasts against to highlight why unstructured sparsity struggles to achieve direct hardware speedups without specialized accelerators.
- Paper: Deep Residual Learning for Image Recognition, Kaiming He et al. (2016). Presents deep residual learning and skip connections, which the source explicitly builds upon and adapts for depth-regularization structured sparsity.
- Paper: Optimal Brain Damage, Yann LeCun et al. (1989). Provides the seminal framework of weight saliency and network pruning to simplify architectures and improve generalization.
- Paper: Learning Efficient Convolutional Networks through Network Slimming, Zhuang Liu et al. (2017). Extends structured channel-level sparsity by applying regularized scaling factors in batch normalization layers rather than directly regularizing weight tensors.
- Paper: Channel Pruning for Accelerating Very Deep Neural Networks, Yihui He et al. (2017). Applies LASSO-based channel selection to reconstruct layer outputs iteratively for deep network acceleration, building on structured filter reduction principles.
- Paper: Rethinking the Value of Network Pruning, Zhuang Liu et al. (2019). Critically re-evaluates structured pruning algorithms like the source paper's by investigating whether inherited weights or discovered compact architectures drive performance.
- Paper: DepGraph: Towards Any Structural Pruning, Gongfan Fang et al. (2023). Generalizes structured pruning across arbitrary neural network architectures by building explicit dependency graphs to resolve complex inter-layer coupling.
- Paper: Efficient Processing of Deep Neural Networks: A Tutorial and Survey, Vivienne Sze et al. (2017). Synthesizes structured sparsity, low-rank approximations, and hardware-software co-design into a comprehensive survey on efficient deep learning execution.
