Activation-Based Pruning of Neural Networks

被引:2
作者
Ganguli, Tushar [1 ]
Chong, Edwin K. P. [1 ]
Werner, Frank [1 ]
机构
[1] Colorado State Univ, Dept Elect & Comp Engn, Ft Collins, CO 80523 USA
关键词
machine learning; network pruning; dimensionality reduction; computer vision;
D O I
10.3390/a17010048
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
We present a novel technique for pruning called activation-based pruning to effectively prune fully connected feedforward neural networks for multi-object classification. Our technique is based on the number of times each neuron is activated during model training. We compare the performance of activation-based pruning with a popular pruning method: magnitude-based pruning. Further analysis demonstrated that activation-based pruning can be considered a dimensionality reduction technique, as it leads to a sparse low-rank matrix approximation for each hidden layer of the neural network. We also demonstrate that the rank-reduced neural network generated using activation-based pruning has better accuracy than a rank-reduced network using principal component analysis. We provide empirical results to show that, after each successive pruning, the amount of reduction in the magnitude of singular values of each matrix representing the hidden layers of the network is equivalent to introducing the sum of singular values of the hidden layers as a regularization parameter to the objective function.
引用
收藏
页数:23
相关论文
共 27 条
[1]  
Frankle J., 2019, arXiv, P1
[2]  
Frankle J, 2020, Arxiv, DOI [arXiv:1903.01611, 10.48550/arXiv.1903.01611]
[3]  
Frantar E, 2023, Arxiv, DOI [arXiv:2301.00774, DOI 10.48550/ARXIV.2301.00774]
[4]   Weighted Nuclear Norm Minimization with Application to Image Denoising [J].
Gu, Shuhang ;
Zhang, Lei ;
Zuo, Wangmeng ;
Feng, Xiangchu .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :2862-2869
[5]  
Guo YW, 2016, ADV NEUR IN, V29
[6]  
Han S, 2016, Arxiv, DOI arXiv:1510.00149
[7]  
Han S, 2015, ADV NEUR IN, V28
[8]   Channel Pruning for Accelerating Very Deep Neural Networks [J].
He, Yihui ;
Zhang, Xiangyu ;
Sun, Jian .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :1398-1406
[9]  
Hu HY, 2016, Arxiv, DOI arXiv:1607.03250
[10]  
Huang G, 2018, Arxiv, DOI arXiv:1711.09224