Enhancing CNN efficiency through mutual information-based filter pruning

Jingqi Lu, Ruiqing Wang, Guanpeng Zuo, Wu Zhang, Xiu Jin, Yuan Rao

Research output: Contribution to journalArticlepeer-review

1 Scopus citations

Abstract

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.

Original languageEnglish
Article number104547
JournalDigital Signal Processing: A Review Journal
Volume151
DOIs
StatePublished - 1 Aug 2024
Externally publishedYes

Keywords

  • Convolutional neural networks
  • Filter pruning
  • Filter redundancy
  • Filter relevance
  • Mutual information

ASJC Scopus subject areas

  • Signal Processing
  • Computer Vision and Pattern Recognition
  • Statistics, Probability and Uncertainty
  • Computational Theory and Mathematics
  • Artificial Intelligence
  • Applied Mathematics
  • Electrical and Electronic Engineering

Fingerprint

Dive into the research topics of 'Enhancing CNN efficiency through mutual information-based filter pruning'. Together they form a unique fingerprint.

Cite this