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 条
  • [21] Mutual information-based feature selection for multilabel classification
    Doquire, Gauthier
    Verleysen, Michel
    NEUROCOMPUTING, 2013, 122 : 148 - 155
  • [22] Mutual Information-based Exploration on Continuous Occupancy Maps
    Jadidi, Maani Ghaffari
    Miro, Jaime Valls
    Dissanayake, Gamini
    2015 IEEE/RSJ INTERNATIONAL CONFERENCE ON INTELLIGENT ROBOTS AND SYSTEMS (IROS), 2015, : 6086 - 6092
  • [23] Mutual information-based recommender system using autoencoder
    Noshad, Zahra
    Bouyer, Asgarali
    Noshad, Mohammad
    APPLIED SOFT COMPUTING, 2021, 109
  • [24] Mutual Information-Based Texture Spectral Similarity Criterion
    Haindl, Michal
    Havlicek, Michal
    ADVANCES IN VISUAL COMPUTING, ISVC 2019, PT I, 2020, 11844 : 302 - 314
  • [25] A Study on Mutual Information-Based Feature Selection in Classifiers
    Arundhathi, B.
    Athira, A.
    Rajan, Ranjidha
    ARTIFICIAL INTELLIGENCE AND EVOLUTIONARY COMPUTATIONS IN ENGINEERING SYSTEMS, ICAIECES 2016, 2017, 517 : 479 - 486
  • [26] Entropy Selective Mutual Information-Based Image Registration
    Karvir, Hrishikesh V.
    Skipper, Julie A.
    Repperger, Daniel W.
    NAECON 2008 - IEEE NATIONAL AEROSPACE AND ELECTRONICS CONFERENCE, 2008, : 173 - +
  • [27] CONDITIONAL DYNAMIC MUTUAL INFORMATION-BASED FEATURE SELECTION
    Liu, Huawen
    Mo, Yuchang
    Zhao, Jianmin
    COMPUTING AND INFORMATICS, 2012, 31 (06) : 1193 - 1216
  • [28] Normalized Mutual Information-based equilibrium optimizer with chaotic maps for wrapper-filter feature selection
    Agrawal, Utkarsh
    Rohatgi, Vasudha
    Katarya, Rahul
    EXPERT SYSTEMS WITH APPLICATIONS, 2022, 207
  • [29] Mutual information-based filter hybrid feature selection method for medical datasets using feature clustering
    Sadegh Asghari
    Hossein Nematzadeh
    Ebrahim Akbari
    Homayun Motameni
    Multimedia Tools and Applications, 2023, 82 : 42617 - 42639
  • [30] Mutual information-based filter hybrid feature selection method for medical datasets using feature clustering
    Asghari, Sadegh
    Nematzadeh, Hossein
    Akbari, Ebrahim
    Motameni, Homayun
    MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 82 (27) : 42617 - 42639