arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.06757cs.CV

Channel Pruning Guided by Classification Loss and Feature Importance

The University of Sydney(悉尼大学);SenseTime Computer Vision Research Group(商汤科技计算机视觉研究组)

Jinyang Guo, Wanli Ouyang, Dong Xu

英文摘要:

In this work, we propose a new layer-by-layer channel pruning method called Channel Pruning guided by classification Loss and feature Importance (CPLI). In contrast to the existing layer-by-layer channel pruning approaches that only consider how to reconstruct the features from the next layer, our approach additionally take the classification loss into account in the channel pruning process. We also observe that some reconstructed features will be removed at the next pruning stage. So it is unnecessary to reconstruct these features. To this end, we propose a new strategy to suppress the influence of unimportant features (i.e., the features will be removed at the next pruning stage). Our comprehensive experiments on three benchmark datasets, i.e., CIFAR-10, ImageNet, and UCF-101, demonstrate the effectiveness of our CPLI method.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑