HRel
HRel computes Mutual Information between convolutional filter activation maps and class labels under the Information Bottleneck framework and prunes low-relevance filters to reduce FLOPs and parameters while preserving classification performance.
Key Features:
- Information Bottleneck and Mutual Information (MI): Employs the Information Bottleneck principle to compute MI between filter activation maps and class labels, defining this quantity as Relevance.
- Relevance-based filter identification: Quantifies filter importance via MI to mark high-Relevance filters as essential and lower-MI filters as candidates for pruning.
- Activation map–label focus: Uniquely evaluates the relationship between activation maps and class labels rather than other MI proxies to determine filter significance.
- Performance metrics: Demonstrated reductions include LeNet-5: 97.98% FLOP reduction with a 0.52% accuracy drop; VGG-16: 94.98% parameter reduction with a 0.36% top-1 accuracy decrease; ResNet-50: 66.42% FLOP pruning with a 1.17% top-5 accuracy decline.
- Information Plane analysis: Analyzes Information Plane dynamics to provide insights into the effects of pruning on network performance.
Scientific Applications:
- CNN architectures: Applied to LeNet-5, VGG-16, ResNet-56, ResNet-110, and ResNet-50 for pruning and efficiency evaluation.
- Datasets: Evaluated on MNIST, CIFAR-10, and ImageNet for performance and generalization assessment.
- Domains: Suitable for image recognition, bioinformatics, and scenarios requiring deployment of CNNs on resource-constrained devices or large-scale data processing.
Methodology:
Compute the mutual information between each filter's activation map and class labels under the Information Bottleneck framework to determine Relevance, then prune filters with lower relevance.
Topics
Details
- Cost:
- Free of charge
- Tool Type:
- desktop application
- Programming Languages:
- Python
- Added:
- 6/14/2022
- Last Updated:
- 6/14/2022
Operations
Data Inputs & Outputs
Filtering
Inputs
Publications
Sarvani C, Ghorai M, Dubey SR, Basha SS. HRel: Filter pruning based on High Relevance between activation maps and class labels. Neural Networks. 2022;147:186-197. doi:10.1016/j.neunet.2021.12.017. PMID:35042156.
PMID: 35042156