Logo image
Pruning Networks With Cross-Layer Ranking & k-Reciprocal Nearest Filters
期刊文章

Pruning Networks With Cross-Layer Ranking & k-Reciprocal Nearest Filters

Mingbao Lin, Liujuan Cao, Yuxin Zhang, Ling Shao, Chia-Wen LinRongrong Ji
IEEE Transactions on Neural Networks and Learning Systems
2022

摘要

Computer architecture Efficient inference filter pruning Indexes Informatics model compression network structure. Search problems Tensors Training Weight measurement Software Computer Science Applications Computer Networks and Communications Artificial Intelligence
This article focuses on filter-level network pruning. A novel pruning method, termed CLR-RNF, is proposed. We first reveal a ``long-tail'' pruning problem in magnitude-based weight pruning methods and then propose a computation-aware measurement for individual weight importance, followed by a cross-layer ranking (CLR) of weights to identify and remove the bottom-ranked weights. Consequently, the per-layer sparsity makes up the pruned network structure in our filter pruning. Then, we introduce a recommendation-based filter selection scheme where each filter recommends a group of its closest filters. To pick the preserved filters from these recommended groups, we further devise a k-reciprocal nearest filter (RNF) selection scheme where the selected filters fall into the intersection of these recommended groups. Both our pruned network structure and the filter selection are nonlearning processes, which, thus, significantly reduces the pruning complexity and differentiates our method from existing works. We conduct image classification on CIFAR-10 and ImageNet to demonstrate the superiority of our CLR-RNF over the state-of-the-arts. For example, on CIFAR-10, CLR-RNF removes 74.1&null FLOPs and 95.0&null parameters from VGGNet-16 with even 0.3&null accuracy improvements. On ImageNet, it removes 70.2&null FLOPs and 64.8&null parameters from ResNet-50 with only 1.7&null top-five accuracy drops. Our project is available at https://github.com/lmbxmu/CLR-RNF.

相關連結

指標

1 檢視次數

詳細資料

Logo image