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 条
  • [1] A Mutual Information-Based Likelihood Function for Particle Filter Flood Extent Assimilation
    Dasgupta, Antara
    Hostache, Renaud
    Ramsankaran, Raaj
    Schumann, Guy J-P
    Grimaldi, Stefania
    Pauwels, Valentijn R. N.
    Walker, Jeffrey P.
    WATER RESOURCES RESEARCH, 2021, 57 (02)
  • [2] Mutual information-based context quantization
    Cagnazzo, Marco
    Antonini, Marc
    Barlaud, Michel
    SIGNAL PROCESSING-IMAGE COMMUNICATION, 2010, 25 (01) : 64 - 74
  • [3] Mutual Information-Based Visual Servoing
    Dame, Amaury
    Marchand, Eric
    IEEE TRANSACTIONS ON ROBOTICS, 2011, 27 (05) : 958 - 969
  • [4] Enhancing Soft Biometric Face Template Privacy with Mutual Information-Based Image Attacks
    Rezgui, Zohra
    Strisciuglio, Nicola
    Veldhuis, Raymond
    2024 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION WORKSHOPS, WACVW 2024, 2024, : 1141 - 1149
  • [5] Mutual Information-Based Modeling for Services Dependency
    ElGhondakly, Roaa A.
    Moussa, Sherin M.
    Badr, Nagwa L.
    IEEE TRANSACTIONS ON SERVICES COMPUTING, 2023, 16 (03) : 2013 - 2029
  • [6] On Mutual Information-Based Optimal Quantizer Design
    Dulek, Berkan
    IEEE COMMUNICATIONS LETTERS, 2022, 26 (05) : 1008 - 1011
  • [7] Improving mutual information-based visual servoing
    Dame, Amaury
    Marchand, Eric
    2010 IEEE INTERNATIONAL CONFERENCE ON ROBOTICS AND AUTOMATION (ICRA), 2010, : 5531 - 5536
  • [8] Mutual information-based feature selection for radiomics
    Oubel, Estanislao
    Beaumont, Hubert
    Iannessi, Antoine
    MEDICAL IMAGING 2016: PACS AND IMAGING INFORMATICS: NEXT GENERATION AND INNOVATIONS, 2016, 9789
  • [9] Mutual Information-based RBM Neural Networks
    Peng, Kang-Hao
    Zhang, Heng
    2016 23RD INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2016, : 2458 - 2463
  • [10] Mutual Information-based Facial Expression Recognition'
    Hazar, Mliki
    Mohamed, Hammami
    Hanene, Ben-Abdallah
    SIXTH INTERNATIONAL CONFERENCE ON MACHINE VISION (ICMV 2013), 2013, 9067