Enhancing CNN efficiency through mutual information-based filter pruning

被引:0
|
作者
Lu, Jingqi [1 ]
Wang, Ruiqing [1 ]
Zuo, Guanpeng [1 ]
Zhang, Wu [1 ,2 ]
Jin, Xiu [1 ,2 ]
Rao, Yuan [1 ,2 ]
机构
[1] Anhui Agr Univ, Dept Sch Informat & Artificial Intelligence, 130, Changjiang West Rd, Hefei 230036, Anhui, Peoples R China
[2] Anhui Agr Univ, Dept Anhui Prov Key Lab Smart Agr Technol & Equipm, 130,Changjiang West Rd, Hefei 230036, Anhui, Peoples R China
关键词
Convolutional neural networks; Filter pruning; Mutual information; Filter relevance; Filter redundancy; NEURAL-NETWORKS;
D O I
10.1016/j.dsp.2024.104547
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
This study presents RRWFP, a novel filter pruning technique for convolutional neural networks (CNNs) designed to improve their deployment on resource -constrained devices. Relevance-Redundancy Filter -Level Weights Pruning (RRWFP) utilises mutual information theory to determine filter relevance. It does this by analysing the mutual information inside filter output activation mappings. This metric helps to find and remove filters based on their redundancy and relevance, achieving a balance that minimises the effect on model accuracy. The empirical evaluations we conducted on CIFAR-10, CIFAR-100, and ImageNet datasets demonstrate the effectiveness of RRWFP. Notably, it achieves minimal accuracy reductions (0.24 % for CIFAR-100 on VGG-16 and 1.01 % for ImageNet on ResNet-50), while significantly reducing model complexity (up to 94.35 % parameter reduction in VGG-16). The results highlight the benefit of incorporating both relevance and redundancy in filter pruning, demonstrating greater performance compared to conventional techniques that address these factors separately.
引用
收藏
页数:15
相关论文
共 50 条
  • [41] Mutual information-based method for selecting informative feature sets
    Herman, Gunawan
    Zhang, Bang
    Wang, Yang
    Ye, Getian
    Chen, Fang
    PATTERN RECOGNITION, 2013, 46 (12) : 3315 - 3327
  • [42] Traffic Prediction With Transfer Learning: A Mutual Information-Based Approach
    Huang, Yunjie
    Song, Xiaozhuang
    Zhu, Yuanshao
    Zhang, Shiyao
    Yu, James J. Q.
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2023, 24 (08) : 8236 - 8252
  • [43] Conditional Mutual Information-Based Generalization Bound for Meta Learning
    Rezazadeh, Arezou
    Jose, Sharu Theresa
    Durisi, Giuseppe
    Simeone, Osvaldo
    2021 IEEE INTERNATIONAL SYMPOSIUM ON INFORMATION THEORY (ISIT), 2021, : 1176 - 1181
  • [44] A Fuzzy Mutual Information-based Feature Selection Method for Classification
    Hogue, N.
    Ahmed, H. A.
    Bhattacharyya, D. K.
    Kalita, J. K.
    FUZZY INFORMATION AND ENGINEERING, 2016, 8 (03) : 355 - 384
  • [45] Gradient intensity: A new mutual information-based registration method
    Shams, Ramtin
    Sadeghi, Parastoo
    Kennedy, Rodney A.
    2007 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, VOLS 1-8, 2007, : 3249 - +
  • [46] Mutual information-based fitness functions for evolutionary circuit synthesis
    Aguirre, AH
    Coello, CAC
    CEC2004: PROCEEDINGS OF THE 2004 CONGRESS ON EVOLUTIONARY COMPUTATION, VOLS 1 AND 2, 2004, : 1309 - 1316
  • [47] Mutual information-based feature selection for intrusion detection systems
    Amiri, Fatemeh
    Yousefi, MohammadMahdi Rezaei
    Lucas, Caro
    Shakery, Azadeh
    Yazdani, Nasser
    JOURNAL OF NETWORK AND COMPUTER APPLICATIONS, 2011, 34 (04) : 1184 - 1199
  • [48] Mutual Information-Based 3D Object Tracking
    Giorgio Panin
    Alois Knoll
    International Journal of Computer Vision, 2008, 78 : 107 - 118
  • [49] Mutual Information-Based Integrated Sensing and Communications: A WMMSE Framework
    Peng, Yizhou
    Yang, Songjie
    Lyu, Wanting
    Li, Ya
    He, Hongjun
    Zhang, Zhongpei
    Assi, Chadi
    IEEE WIRELESS COMMUNICATIONS LETTERS, 2024, 13 (10) : 2642 - 2646
  • [50] Camera Localization using Mutual Information-based Multiplane Tracking
    Delabarre, Bertrand
    Marchand, Eric
    2013 IEEE/RSJ INTERNATIONAL CONFERENCE ON INTELLIGENT ROBOTS AND SYSTEMS (IROS), 2013, : 1620 - 1625