修剪
滤波器(信号处理)
卷积神经网络
特征(语言学)
计算机科学
人工智能
还原(数学)
模式识别(心理学)
算法
数学
计算机视觉
语言学
哲学
农学
生物
几何学
作者
Xuming Han,Y.X. Chu,Ke Wang,Limin Wang,Yuqing Lin,Weiping Ding
标识
DOI:10.1016/j.ins.2024.120229
摘要
Filter pruning is an effective method for reducing the size of convolutional neural networks without sacrificing performance. Most filter pruning methods prioritize filters with high information content, but fail to consider that filters with low information content might capture essential features. Moreover, we have discovered that the distinctions among feature maps generated by filters can identify crucial features. Based on this insight, we propose a novel pruning method called inTer-feAture distinctIon fiLter fusiOn pRuning (TAILOR), which fuses the feature distinctions between filters. TAILOR randomly selects multiple filter sets within a convolutional layer and calculates the output feature maps of the next convolutional layer of these sets. Subsequently, an intelligent distinction optimization scheme is proposed to obtain the optimal filter set for filter pruning, which supplants the original convolutional layer. Experimental results indicated that the inter-feature distinctions among filters significantly affect filter pruning. TAILOR outperforms state-of-the-art filter-pruning methods in terms of model prediction accuracy, floating-point operations, and parameter scale. For instance, with VGG-16, TAILOR achieves a 73.89% FLOPs reduction by removing 91.85% of the parameters, while improving accuracy by 0.36% on the CIFAR-10.
科研通智能强力驱动
Strongly Powered by AbleSci AI