Elastic Filter Prune in Deep Neural Networks Using Modified Weighted Hybrid Criterion

被引:0
|
作者
Hu, Wei [1 ,2 ]
Han, Yi [1 ,2 ]
Liu, Fang [3 ,4 ]
Hu, Mingce [1 ,2 ]
Li, Xingyuan [1 ,2 ]
机构
[1] Wuhan Univ Sci & Technol, Coll Comp Sci, Wuhan, Peoples R China
[2] Hubei Prov Key Lab Intelligent Informat Proc & Re, Wuhan, Peoples R China
[3] Wuhan Univ, Sch Comp Sci, Wuhan, Peoples R China
[4] Wuhan Inst City, Dept Informat Engn, Wuhan, Peoples R China
来源
KNOWLEDGE SCIENCE, ENGINEERING AND MANAGEMENT, PT I, KSEM 2024 | 2024年 / 14884卷
关键词
Convolutional Neural Network; Filter Pruning; Elastic Net; Accelerator;
D O I
10.1007/978-981-97-5492-2_2
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The deployment of Convolutional Neural Networks (CNNs) on edge devices has gradually become a hot topic in research and application. However, simply pursuing high-performance networks is no longer suitable for scenarios that require comprehensive optimization of limited resources, low power consumption, and real-time performance, due to the huge storage and computational overhead required by deep convolutional networks. Based on the Weighted Hybrid Criterion (WHC) method and inspired by Elastic Net, we propose an improved filter pruning method called Elastic Weighted Hybrid Criterion (E-WHC). This approach introduces a regularization parameter to balance the sparsity and importance of filters. Experimental results show that on the CIFAR-10 dataset, E-WHC on the ResNet-32 network achieves a 0.25% accuracy improvement while reducing more than 41.5% of floating-point operations compared to WHC. Similarly, on the ResNet-56 network, E-WHC achieves a 0.1% accuracy improvement while reducing over 28.4% of floating-point operations. We further analyze the reasons for this improvement and discuss the role of the regularization parameter in filter selection. The proposed research provides new insights for the improvement of filter selection methods in convolutional neural networks.
引用
收藏
页码:16 / 27
页数:12
相关论文
empty
未找到相关数据