arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1707.08005stat.MLcs.LG

Towards Evolutional Compression

Peking University(北京大学);The University of Sydney(悉尼大学)

Yunhe Wang, Chang Xu, Jiayan Qiu, Chao Xu, Dacheng Tao

英文摘要:

Compressing convolutional neural networks (CNNs) is essential for transferring the success of CNNs to a wide variety of applications to mobile devices. In contrast to directly recognizing subtle weights or filters as redundant in a given CNN, this paper presents an evolutionary method to automatically eliminate redundant convolution filters. We represent each compressed network as a binary individual of specific fitness. Then, the population is upgraded at each evolutionary iteration using genetic operations. As a result, an extremely compact CNN is generated using the fittest individual. In this approach, either large or small convolution filters can be redundant, and filters in the compressed network are more distinct. In addition, since the number of filters in each convolutional layer is reduced, the number of filter channels and the size of feature maps are also decreased, naturally improving both the compression and speed-up ratios. Experiments on benchmark deep CNN models suggest the superiority of the proposed algorithm over the state-of-the-art compression methods.

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

↑