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

Network Adjustment: Channel Search Guided by FLOPs Utilization Ratio

Beihang University(北京航空航天大学);Johns Hopkins University(约翰斯·霍普金斯大学);Peking University(北京大学);Xidian University(西安电子科技大学)

Zhengsu Chen, Jianwei Niu, Lingxi Xie, Xuefeng Liu, Longhui Wei, Qi Tian

英文摘要:

Automatic designing computationally efficient neural networks has received much attention in recent years. Existing approaches either utilize network pruning or leverage the network architecture search methods. This paper presents a new framework named network adjustment, which considers network accuracy as a function of FLOPs, so that under each network configuration, one can estimate the FLOPs utilization ratio (FUR) for each layer and use it to determine whether to increase or decrease the number of channels on the layer. Note that FUR, like the gradient of a non-linear function, is accurate only in a small neighborhood of the current network. Hence, we design an iterative mechanism so that the initial network undergoes a number of steps, each of which has a small `adjusting rate' to control the changes to the network. The computational overhead of the entire search process is reasonable, i.e., comparable to that of re-training the final model from scratch. Experiments on standard image classification datasets and a wide range of base networks demonstrate the effectiveness of our approach, which consistently outperforms the pruning counterpart. The code is available at https://github.com/danczs/NetworkAdjustment.

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

↑