arrow
返回

CSHE: network pruning by using cluster similarity and matrix eigenvalues

delete2021-09-13
delete14
PRE
AI
邵明文 封面图
邵明文 (Mingwen Shao) *
J
Junhui Dai
王
王冉 (Ran Wang)
J
Jiandong Kuang
左
左旺孟 (Wangmeng Zuo)
DOI:10.1007/s13042-021-01411-8delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Although deep convolutional neural networks (CNNs) have achieved significant success in computer vision applications, the real-world deployment of CNNs is often limited by computing resources and memory constraints. As a mainstream deep model compression technology, neural network pruning offers a promising prospect to reduce models' parameters and calculation. In this paper, we proposed a novel filter pruning method that combines convolution filters and feature maps information for convolutional neural network compression, namely network pruning by using cluster similarity and large eigenvalues (CSHE). First, based on the convolution operation, we explore the similarity relationship of feature maps generated by the corresponding filters. Concretely, the clustering algorithm is used to classify the similarity of filter to guide the classification of feature map. Secondly, the proposed method utilizes the information of the large eigenvalues of the feature maps to rank the importance of filters. Finally, we prune the low-ranking filters and remain the high-ranking ones. The proposed method eliminates redundancy in convolution filters by applying large eigenvalues of feature maps based on filters similarity. In this way, most of the representative information in the network can be retained and the pruned results can be easily reproduced. Experiments show that the accuracy of the pruned sparse deep network obtained by the CSHE method in the classification tasks of CIFAR-10 and ImageNet ILSVRC-12 is almost the same as that of the reference network without any additional constraints.
Keyword:
Model compression
Feature map
Filter pruning
Filter clustering
Matrix eigenvalues

期刊

International Journal of Machine Learning and Cybernetics 封面图
International Journal of Machine Learning and Cybernetics
IF:
2.7
论文数:
3.2K
被引数:
5.6K

机构

H
harbin institute of technology
学者数:
8.0W
论文数: 6.6W
被引数: 66
C
china university of petroleum
学者数:
4.1W
论文数: 2.7W
被引数: 30
引用论文

引用论文

Calcium-modulating cyclophilin ligand regulates membrane trafficking of postsynaptic GABAA receptors
err2008-06-01
err0
errOAAI
errXu Yuan; Jun Yao; David Norris; David D. Tran; Richard J. Bram; Gong Chen; Bernhard Luscher
err分享
err收藏
Major Adverse Limb Events and Mortality in Patients With Peripheral Artery Disease
err2018-05-01
err0
errOAAI
errSonia S. Anand; Francois Caron; John W. Eikelboom; Jackie Bosch; Leanne Dyal; Victor Aboyans; Maria Teresa Abola; Kelley R.H. Branch; Katalin Keltai; Deepak L. Bhatt; Peter Verhamme; Keith A.A. Fox; Nancy Cook-Bruns; Vivian Lanius; Stuart J. Connolly; Salim Yusuf
err分享
err收藏
Modelling and dynamic analysis of a MEMS ring resonator supported by circular curved shaped inner beams
err2023-08-16
err0
errOAAI
errAhmad Rahbar Ranji; Jingshuai Guo; Shahpour Alirezaee; Mohammed Jalal Ahamed
err分享
err收藏
ImageNet Large Scale Visual Recognition ChallengeImageNet大规模视觉识别挑战
err2015-04-11
err2.7W
PREAI
errRussakovsky, Olga; Deng, Jia; Su, Hao; Krause, Jonathan; Satheesh, Sanjeev; Ma, Sean; Huang, Zhiheng; Karpathy, Andrej; Khosla, Aditya; Bernstein, Michael; Berg, Alexander C.; Fei-Fei, Li
err分享
err收藏
Predictors of walking capacity in peripheral arterial disease patients
err2013-04-01
err0
errOAAI
errBreno Quintella Farah; João Paulo dos Anjos Souza Barbosa; Gabriel Grizzo Cucato; Marcel da Rocha Chehuen; Luis Alberto Gobbo; Nelson Wolosker; Cláudia Lúcia de Moraes Forjaz; Raphael Mendes Ritti-Dias
err分享
err收藏
学者 查看更多内容