SeNPIS: Sequential Network Pruning by class-wise Importance Score

被引:7
|
作者
Pachon, Cesar G. [1 ]
Ballesteros, Dora M. [1 ]
Renza, Diego [1 ]
机构
[1] Univ Mil Nueva Granada, Carrera 11 101-80, Bogota 110111, Colombia
关键词
Deep learning; Model compression; Pruning algorithm; Importance score; Convolutional neural network;
D O I
10.1016/j.asoc.2022.109558
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In the last decade, pattern recognition and decision making from images has mainly focused on the development of deep learning architectures, with different types of networks such as sequential, residual and parallel. Although the depth and size varies between models, they all have in common that they can contain multiple filters or neurons that are not important for the purpose of prediction, and that do negatively impact the size of the model and their inference times. Therefore, it is advantageous to use pruning methods that, while largely maintaining the initial performance of the classifier, significantly reduce its size and FLOPs. In parameter reduction, the decision rule is generally based on mathematical criteria, e.g. the amplitude of the weights, but not on the actual impact of the filter or neuron on the classifier performance for each of the classes. Therefore, we propose SeNPIS as a method that involves both filter and neuron selection based on a class-wise importance score, and network resizing to increase parameter reduction and FLOPs in sequential CNN networks. Several tests were performed to compare SeNPIS with other representative state-of-the-art methods, for the CIFAR-10 and Scene-15 datasets. It was found that for similar values of accuracy, and even in some cases with a slight increase in accuracy, SeNPIS significantly reduces the number of parameters by up to an additional 23.5% (i.e., a 51.05% reduction with SeNPIS versus a 27.53% reduction with Gradient) and FLOPs by up to an additional 26.6% (i.e., a 74.82% reduction with SeNPIS versus a 48.16% reduction with Weight) compared to the Weight, Taylor, Gradient and LRP methods.(c) 2022 Elsevier B.V. All rights reserved.
引用
收藏
页数:13
相关论文
共 50 条
  • [41] Class-wise Centroid Distance Metric Learning for Acoustic Event Detection
    Lu, Xugang
    Shen, Peng
    Li, Sheng
    Tsao, Yu
    Kawai, Hisashi
    INTERSPEECH 2019, 2019, : 3614 - 3618
  • [42] Predicting protein structural class by SVM with class-wise optimized features and decision probabilities
    Anand, Ashish
    Pugalenthi, Ganesan
    Suganthan, P. N.
    JOURNAL OF THEORETICAL BIOLOGY, 2008, 253 (02) : 375 - 380
  • [43] Class-wise Metric Scaling for Improved Few-Shot Classification
    Liu, Ge
    Zhao, Linglan
    Li, Wei
    Guo, Dashan
    Fang, Xiangzhong
    2021 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV 2021), 2021, : 586 - 595
  • [44] Class-wise two-dimensional PCA method for face recognition
    Turhan, Ceren Guzel
    Bilge, Hasan Sakir
    IET COMPUTER VISION, 2017, 11 (04) : 286 - 300
  • [45] Bi-directional class-wise adversaries for unsupervised domain adaptation
    Guanglei Yang
    Mingli Ding
    Yongqiang Zhang
    Applied Intelligence, 2022, 52 : 3623 - 3639
  • [46] Class-Wise Feature Alignment Based Transfer Network for Multi-Temporal Remote Sensing Image Classification
    Guo Y.
    Song J.
    Ma L.
    Yang M.
    Diqiu Kexue - Zhongguo Dizhi Daxue Xuebao/Earth Science - Journal of China University of Geosciences, 2021, 46 (10): : 3730 - 3739
  • [47] Device-Wise Federated Network Pruning
    Gao, Shangqian
    Li, Junyi
    Zhang, Zeyu
    Zhang, Yanfu
    Cai, Weidong
    Huang, Heng
    2024 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2024, : 12342 - 12352
  • [48] Class-wise confidence-aware active learning for laparoscopic images segmentation
    Jie Qiu
    Yuichiro Hayashi
    Masahiro Oda
    Takayuki Kitasaka
    Kensaku Mori
    International Journal of Computer Assisted Radiology and Surgery, 2023, 18 : 473 - 482
  • [49] Genetic programming and class-wise orthogonal transformation for dimension reduction in classification problems
    Neshatian, Kourosh
    Zhang, Mengjie
    GENETIC PROGRAMMING, PROCEEDINGS, 2008, 4971 : 242 - 253
  • [50] CLASS-WISE OPTIMAL-ALGORITHMS FOR OPTIMIZATION IN CORRELATED-NOISE CONDITIONS
    POZNYAK, AS
    TSYPKIN, YZ
    USSR COMPUTATIONAL MATHEMATICS AND MATHEMATICAL PHYSICS, 1984, 24 (03): : 112 - 122