Efficient tensor decomposition-based filter pruning - Archive ouverte HAL
Article Dans Une Revue Neural Networks Année : 2024

Efficient tensor decomposition-based filter pruning

Résumé

In this paper, we present CORING, which is short for effiCient tensOr decomposition-based filteR prunING, a novel filter pruning methodology for neural networks. CORING is crafted to achieve efficient tensor decomposition-based pruning, a stark departure from conventional approaches that rely on vectorized or matricized filter representations. Our approach represents a significant leap forward in the field by introducing tensor decompositions, specifically the HOSVD, which preserves the multidimensional nature of filters while providing a low-rank approximation, thus substantially reducing complexity. Furthermore, we introduce a versatile method for calculating filter similarity by using the low-rank approximation offered by the HOSVD. This obviates the need for using full filters or reshaped versions and enhances the overall efficiency and effectiveness of our approach. Extensive experimentation across diverse architectures and datasets spanning various vision tasks, including image classification, object detection, instance segmentation, and keypoint detection, validates CORING's prowess. Remarkably, it outperforms state-of-the-art methods in reducing MACs and parameters, consistently enhancing validation accuracy. Furthermore, we supplement our quantitative results with a comprehensive ablation study, providing substantial evidence of the efficiency of our tensor-based approach. Beyond quantitative outcomes, qualitative results vividly illustrate CORING's ability to retain essential features within pruned neural networks. Our code is available for research purposes.
Fichier principal
Vignette du fichier
_NN__CORING__Manuscript_v2_0.pdf (5.03 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)
Licence
Domaine public

Dates et versions

hal-04577465 , version 1 (16-05-2024)

Licence

Domaine public

Identifiants

Citer

Van Tien Pham, Yassine Zniyed, Thanh Phuong Nguyen. Efficient tensor decomposition-based filter pruning. Neural Networks, In press, pp.106393. ⟨10.1016/j.neunet.2024.106393⟩. ⟨hal-04577465⟩
90 Consultations
219 Téléchargements

Altmetric

Partager

More