arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2001.08565cs.CV

Channel Pruning via Automatic Structure Search

Xiamen University(厦门大学);Beihang University(北京航空航天大学);Tencent Youtu Lab, Tencent Technology (Shanghai) Co., Ltd(腾讯优图实验室,腾讯科技(上海)有限公司);Peking University(北京大学)

Mingbao Lin, Rongrong Ji, Yuxin Zhang, Baochang Zhang, Yongjian Wu, Yonghong Tian

英文摘要:

Channel pruning is among the predominant approaches to compress deep neural networks. To this end, most existing pruning methods focus on selecting channels (filters) by importance/optimization or regularization based on rule-of-thumb designs, which defects in sub-optimal pruning. In this paper, we propose a new channel pruning method based on artificial bee colony algorithm (ABC), dubbed as ABCPruner, which aims to efficiently find optimal pruned structure, i.e., channel number in each layer, rather than selecting "important" channels as previous works did. To solve the intractably huge combinations of pruned structure for deep networks, we first propose to shrink the combinations where the preserved channels are limited to a specific space, thus the combinations of pruned structure can be significantly reduced. And then, we formulate the search of optimal pruned structure as an optimization problem and integrate the ABC algorithm to solve it in an automatic manner to lessen human interference. ABCPruner has been demonstrated to be more effective, which also enables the fine-tuning to be conducted efficiently in an end-to-end manner. The source codes can be available at https://github.com/lmbxmu/ABCPruner.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑