Lossless-Reconstruction-of-Convolutional-Neural-Network-for-Channel-Based-Network-Pruning

This is the official implementation of "Lossless Reconstruction of Convolutional Neural Network for Channel-Based Network Pruning".

Contents

1. Abstract
2. Methods
3. Experiments
4. Reference

Abstract

Network pruning reduces the number of parameters and computational costs of convolutional neural networks while maintaining high performance. Although existing pruning methods have achieved excellent results, they do not consider reconstruction after pruning in order to apply the network to actual devices. This study proposes a reconstruction process for channel-based network pruning. For lossless reconstruction, we focus on three components of the network: the residual block, skip connection, and convolution layer. Union operation and index alignment are applied to the residual block and skip connection, respectively. Furthermore, we reconstruct a compressed convolution layer by considering batch normalization. We apply our method to existing channel-based pruning methods for downstream tasks such as image classification, object detection, and semantic segmentation. Experimental results show that compressing a large model has a 1.93% higher accuracy in image classification, 2.2 higher mean Intersection over Union (mIoU) in semantic segmentation, and 0.054 higher mean Average Precision (mAP) in object detection than well-designed small models. Moreover, we demonstrate that our method can reduce the actual latency by 8.15× and 5.29× on Raspberry Pi and Jetson Nano, respectively.

Methods

Pruning process with reconstruction

Residual Block

Skip Connection

Batch Normalization

Experiments

Reference

@article{lee2023lossless,
  title={Lossless Reconstruction of Convolutional Neural Network for Channel-Based Network Pruning},
  author={Lee, Donghyeon and Lee, Eunho and Hwang, Youngbae},
  journal={Sensors},
  volume={23},
  number={4},
  pages={2102},
  year={2023},
  publisher={MDPI}
}