arrow
Return

Network pruning via resource reallocation

delete2024-01-01
delete8
delete
OA
AI
Y
Yuenan Hou *
Z
Zheng Ma
C
Chunxiao Liu
Z
Zhe Wang
C
Chen Change Loy
DOI:10.1016/j.patcog.2023.109886delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
Channel pruning is broadly recognized as an effective approach to obtain a small compact model through eliminating unimportant channels from a large cumbersome network. Contemporary methods typically perform iterative pruning procedure from the original over-parameterized model, which is both tedious and expensive especially when the pruning is aggressive. In this paper, we propose a simple yet effective channel pruning technique, termed network Pruning via rEsource rEalLocation (PEEL), to quickly produce a desired slim model with negligible cost. Specifically, PEEL first constructs a predefined backbone and then conducts resource reallocation on it to shift parameters from less informative layers to more important layers in one round, thus amplifying the positive effect of these informative layers. To demonstrate the effectiveness of PEEL , we perform extensive experiments on ImageNet with ResNet-18, ResNet-50, MobileNetV2, MobileNetV3-small and EfficientNet-B0. Experimental results show that structures uncovered by PEEL exhibit competitive performance with state-of-the-art pruning algorithms under various pruning settings. Encouraging results are also observed when applying PEEL to compress the semantic segmentation model. Our code is available at https://github.com/cardwing/Codes-for-PEEL.
Keywords:
Network pruning
Resource reallocation
Searching cost
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

Pattern Recognition cover
Pattern Recognition
IF:
7.6
Papers:
1.3W
Citations:
4.5W

Organization

S
Shanghai Artificial Intelligence Laboratory
Scholars:
457
Papers: 256
Citations: 765
N
Nanyang Technological University
Scholars:
4.9W
Papers: 4.8W
Citations: 8.1W