跳到主要导航 跳到搜索 跳到主要内容

Accelerating convolutional networks via global & dynamic filter pruning

  • Shaohui Lin
  • , Rongrong Ji*
  • , Yuchao Li
  • , Yongjian Wu
  • , Feiyue Huang
  • , Baochang Zhang
  • *此作品的通讯作者
  • Xiamen University
  • Tencent

科研成果: 书/报告/会议事项章节会议稿件同行评审

摘要

Accelerating convolutional neural networks has recently received ever-increasing research focus. Among various approaches proposed in the literature, filter pruning has been regarded as a promising solution, which is due to its advantage in significant speedup and memory reduction of both network model and intermediate feature maps. To this end, most approaches tend to prune filters in a layer-wise fixed manner, which is incapable to dynamically recover the previously removed filter, as well as jointly optimize the pruned network across layers. In this paper, we propose a novel global & dynamic pruning (GDP) scheme to prune redundant filters for CNN acceleration. In particular, GDP first globally prunes the unsalient filters across all layers by proposing a global discriminative function based on prior knowledge of each filter. Second, it dynamically updates the filter saliency all over the pruned sparse network, and then recovers the mistakenly pruned filter, followed by a retraining phase to improve the model accuracy. Specially, we effectively solve the corresponding non-convex optimization problem of the proposed GDP via stochastic gradient descent with greedy alternative updating. Extensive experiments show that the proposed approach achieves superior performance to accelerate several cutting-edge CNNs on the ILSVRC 2012 benchmark, comparing to the state-of-the-art filter pruning methods.

源语言英语
主期刊名Proceedings of the 27th International Joint Conference on Artificial Intelligence, IJCAI 2018
编辑Jerome Lang
出版商International Joint Conferences on Artificial Intelligence
2425-2432
页数8
ISBN(电子版)9780999241127
DOI
出版状态已出版 - 2018
活动27th International Joint Conference on Artificial Intelligence, IJCAI 2018 - Stockholm, 瑞典
期限: 13 7月 201819 7月 2018

出版系列

姓名IJCAI International Joint Conference on Artificial Intelligence
2018-July
ISSN(印刷版)1045-0823

会议

会议27th International Joint Conference on Artificial Intelligence, IJCAI 2018
国家/地区瑞典
Stockholm
时期13/07/1819/07/18

指纹

探究 'Accelerating convolutional networks via global & dynamic filter pruning' 的科研主题。它们共同构成独一无二的指纹。

引用此