Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

CHIP: CHannel Independence-based Pruning for Compact Neural Networks

About

Filter pruning has been widely used for neural network compression because of its enabled practical acceleration. To date, most of the existing filter pruning works explore the importance of filters via using intra-channel information. In this paper, starting from an inter-channel perspective, we propose to perform efficient filter pruning using Channel Independence, a metric that measures the correlations among different feature maps. The less independent feature map is interpreted as containing less useful information$/$knowledge, and hence its corresponding filter can be pruned without affecting model capacity. We systematically investigate the quantification metric, measuring scheme and sensitiveness$/$reliability of channel independence in the context of filter pruning. Our evaluation results for different models on various datasets show the superior performance of our approach. Notably, on CIFAR-10 dataset our solution can bring $0.90\%$ and $0.94\%$ accuracy increase over baseline ResNet-56 and ResNet-110 models, respectively, and meanwhile the model size and FLOPs are reduced by $42.8\%$ and $47.4\%$ (for ResNet-56) and $48.3\%$ and $52.1\%$ (for ResNet-110), respectively. On ImageNet dataset, our approach can achieve $40.8\%$ and $44.8\%$ storage and computation reductions, respectively, with $0.15\%$ accuracy increase over the baseline ResNet-50 model. The code is available at https://github.com/Eclipsess/CHIP_NeurIPS2021.

Yang Sui, Miao Yin, Yi Xie, Huy Phan, Saman Zonouz, Bo Yuan• 2021

Related benchmarks

TaskDatasetResultRank
Image ClassificationImageNet 1k (test)
Top-1 Accuracy76.3
880
Image ClassificationImageNet (val)
Top-1 Accuracy76.4
68
Image ClassificationCIFAR10 (test)
Test Accuracy92.88
49
Image ClassificationImageNet-1k (val)
Pruned Top-1 Acc76.3
46
PruningCIFAR-100 (test)
FLOPs-based Accuracy-Retention AUC0.284
27
Image ClassificationCIFAR-10
Primary Accuracy94.44
23
Image ClassificationCIFAR-10 (test)
Baseline Accuracy93.26
16
Image ClassificationCIFAR-10 (test)
Baseline Accuracy0.935
12
Image ClassificationCIFAR10 (test)
FLOPs (%)27.7
8
Showing 9 of 9 rows

Other info

Follow for update