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 条
  • [31] A Comparison of Mutual and Fuzzy-Mutual Information-Based Feature Selection Strategies
    Tsai, Yu-Shuen
    Yang, Ueng-Cheng
    Chung, I-Fang
    Huang, Chuen-Der
    2013 IEEE INTERNATIONAL CONFERENCE ON FUZZY SYSTEMS (FUZZ - IEEE 2013), 2013,
  • [32] Filter Pruning Based on Information Capacity and Independence
    Tang, Xiaolong
    Ye, Shuo
    Shi, Yufeng
    Hu, Tianheng
    Peng, Qinmu
    You, Xinge
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2024,
  • [33] OPTIMAL FILTER BASED ON MUTUAL INFORMATION
    OMATSU, S
    KIKUCHI, A
    MIYASHITA, T
    SOEDA, T
    ELECTRONICS & COMMUNICATIONS IN JAPAN, 1978, 61 (09): : 11 - 19
  • [34] MICAL: Mutual Information-Based CNN-Aided Learned Factor Graphs for Seizure Detection From EEG Signals
    Salafian, Bahareh
    Ben-Knaan, Eyal Fishel
    Shlezinger, Nir
    De Ribaupierre, Sandrine
    Farsad, Nariman
    IEEE ACCESS, 2023, 11 : 23085 - 23096
  • [35] Mutual information-based rigid and nonrigid registration of ultrasound volumes
    Shekhar, R
    Zagrodsky, V
    IEEE TRANSACTIONS ON MEDICAL IMAGING, 2002, 21 (01) : 9 - 22
  • [36] Two-step mutual information-based stereo matching
    Heo, Y. S.
    ELECTRONICS LETTERS, 2016, 52 (14) : 1225 - 1226
  • [37] MIRA: mutual information-based reporter algorithm for metabolic networks
    Cicek, A. Ercument
    Roeder, Kathryn
    Ozsoyoglu, Gultekin
    BIOINFORMATICS, 2014, 30 (12) : 175 - 184
  • [38] DEFAULT: Mutual Information-based Crash Triage for Massive Crashes
    Zhang, Xing
    Chen, Jiongyi
    Feng, Chao
    Li, Ruilin
    Diao, Wenrui
    Zhang, Kehuan
    Lei, Jing
    Tang, Chaojing
    2022 ACM/IEEE 44TH INTERNATIONAL CONFERENCE ON SOFTWARE ENGINEERING (ICSE 2022), 2022, : 635 - 646
  • [39] Study on mutual information-based feature selection for text categorization
    Xu, Yan
    Jones, Gareth
    Li, Jintao
    Wang, Bin
    Sun, Chunming
    Journal of Computational Information Systems, 2007, 3 (03): : 1007 - 1012
  • [40] A Mutual Information-based Approach to Quantifying Logography in Japanese and Sumerian
    Hermalin, Noah
    Proceedings of the Annual Meeting of the Association for Computational Linguistics, 2023, : 105 - 110