Adaptive Pruning of Channel Spatial Dependability in Convolutional Neural Networks
Weiying Xie, Mei Yuan, Jitao Ma, Yunsong Li
摘要
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).
问问这篇 Paper
问问你的智能体。
Lune 读过与它相关的顶会 Paper,每个回答都会注明依据哪几篇。
相关 Paper
- Towards Compact CNNs via Collaborative CompressionYuchao Li, Shaohui Lin, Jianzhuang Liu, Qixiang Ye 等CVPR 2021
- Rethinking the Pruning Criteria for Convolutional Neural NetworkZhongzhan Huang, Wenqi Shao, Xinjiang Wang, Liang Lin 等NeurIPS 2021 · 被引用 75 次
- AKECP: Adaptive Knowledge Extraction from Feature Maps for Fast and Efficient Channel PruningHaonan Zhang, Longjun Liu, Hengyi Zhou, Wenxuan Hou 等ACM MM 2021 · 被引用 29 次
- 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 等IEEE VIS 2020 · 被引用 49 次
