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 条
  • [21] CCL: CLASS-WISE CURRICULUM LEARNING FOR CLASS IMBALANCE PROBLEMS.
    Escudero-Vinolo, Marcos
    Lopez-Cifuentes, Alejandro
    2022 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING, ICIP, 2022, : 1476 - 1480
  • [22] A Game Theoretic Approach to Class-wise Selective Rationalization
    Chang, Shiyu
    Zhang, Yang
    Yu, Mo
    Jaakkola, Tommi S.
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 32 (NIPS 2019), 2019, 32
  • [23] SAR Image Quality Assessment: From Sample-Wise to Class-Wise
    Yu, Ziyi
    Dong, Ganggang
    Liu, Hongwei
    REMOTE SENSING, 2023, 15 (08)
  • [24] Multi-label Aerial Image Classification using A Bidirectional Class-wise Attention Network
    Hua, Yuansheng
    Mou, Lichao
    Zhu, Xiao Xiang
    2019 JOINT URBAN REMOTE SENSING EVENT (JURSE), 2019,
  • [25] Class-Wise Denoising for Robust Learning Under Label Noise
    Gong, Chen
    Ding, Yongliang
    Han, Bo
    Niu, Gang
    Yang, Jian
    You, Jane
    Tao, Dacheng
    Sugiyama, Masashi
    IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2023, 45 (03) : 2835 - 2848
  • [26] Class-wise Thresholding for Robust Out-of-Distribution Detection
    Guarrera, Matteo
    Jin, Baihong
    Lin, Tung-Wei
    Zuluaga, Maria A.
    Chen, Yuxin
    Sangiovanni-Vincentelli, Alberto
    2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS, CVPRW 2022, 2022, : 2836 - 2845
  • [27] Class-Wise Adaptive Strategy for Semi Supervised Semantic Segmentation
    Ju, Jingi
    Noh, Hyeoncheol
    Wang, Yooseung
    Choi, Dong-Geol
    IEEE ACCESS, 2024, 12 : 21662 - 21672
  • [28] Class-wise boundary regression by uncertainty in temporal action detection
    Chen, Yunze
    Chen, Mengjuan
    Gu, Qingyi
    IET IMAGE PROCESSING, 2022, 16 (14) : 3854 - 3862
  • [29] PCA: Progressive class-wise attention for skin lesions diagnosis
    Naveed, Asim
    Naqvi, Syed S.
    Khan, Tariq M.
    Razzak, Imran
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2024, 127
  • [30] Discovering Class-wise Trends of Max-pooling in Subspace
    Zheng, Yuchen
    Iwana, Brian Kenji
    Uchida, Seiichi
    PROCEEDINGS 2018 16TH INTERNATIONAL CONFERENCE ON FRONTIERS IN HANDWRITING RECOGNITION (ICFHR), 2018, : 98 - 103