Adaptive Pruning of Channel Spatial Dependability in Convolutional Neural Networks
Weiying Xie, Mei Yuan, Jitao Ma, Yunsong Li
Abstract
Deep Convolutional Neural Networks (CNNs) have demonstrated excellent performance in various multimedia application scenarios. However, complex models often require significant computational resources and energy costs. Therefore, CNN compression is crucial for addressing deployment challenges of multimedia application on resource constrained edge devices. However, existing CNN channel pruning strategies primarily focus on the "weights" or "activations" of the model, overlooking its "interpretability" information. In this paper, we explore CNN pruning strategies from the perspective of model interpretability. We model the correspondence between channel feature maps and interpretable visual perception based on class saliency maps, aiming to assess the contribution of each channel to the desired output. Additionally, we utilize Discrete Wavelet Transform (DWT) to capture the global features and structure of class saliency maps. Based on this, we propose a Channel Spatial Dependability (CSD) metric, evaluating the importance and contribution of channels in a bidirectional manner to guide model pruning. And we dynamically adjust the pruning rate of each layer based on performance changes, in order to achieve more accurate and efficient adaptive pruning. Our method achieves significant results across a range of different networks and datasets. For instance, we achieved a 51.3% pruning on the ResNet-56 model while maintaining an accuracy of 94.16%, outperforming feature-map or other State-of-the-Art (SOTA).
Ask about this paper
Ask your agent about it.
Lune has read the top-tier papers around this one, so every answer names the papers it rests on.
Related papers
- Towards Compact CNNs via Collaborative CompressionYuchao Li, Shaohui Lin, Jianzhuang Liu, Qixiang Ye et al.CVPR 2021
- Rethinking the Pruning Criteria for Convolutional Neural NetworkZhongzhan Huang, Wenqi Shao, Xinjiang Wang, Liang Lin et al.NeurIPS 2021 · 75 citations
- AKECP: Adaptive Knowledge Extraction from Feature Maps for Fast and Efficient Channel PruningHaonan Zhang, Longjun Liu, Hengyi Zhou, Wenxuan Hou et al.ACM MM 2021 · 29 citations
- Multi-Dimensional Pruning: A Unified Framework for Model CompressionJinyang Guo, Wanli Ouyang, Dong XuCVPR 2020
- CNNPruner: Pruning Convolutional Neural Networks with Visual AnalyticsGuan Li, Junpeng Wang, Han-Wei Shen, Kaixin Chen et al.IEEE VIS 2020 · 49 citations
