Title
TENSOR REORDERING FOR CNN COMPRESSION
Abstract
We show how parameter redundancy in Convolutional Neural Network (CNN) filters can be effectively reduced by pruning in spectral domain. Specifically, the representation extracted via Discrete Cosine Transform (DCT) is more conducive for pruning than the original space. By relying on a combination of weight tensor reshaping and reordering we achieve high levels of layer compression with just minor accuracy loss. Our approach is applied to compress pre-trained CNNs and we show that minor additional fine-tuning allows our method to recover the original model performance after a significant parameter reduction. We validate our approach on ResNet-50 and MobileNet-V2 architectures for ImageNet classification task.
Year
DOI
Venue
2021
10.1109/ICASSP39728.2021.9413944
2021 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP 2021)
Keywords
DocType
Citations 
Compression, Pruning, Reordering, DCT, CNN
Conference
0
PageRank 
References 
Authors
0.34
0
3
Name
Order
Citations
PageRank
Matej Ulicny111.03
Vladimir A. Krylov213314.81
Rozenn Dahyot334032.62