计算机科学与探索 ›› 2020, Vol. 14 ›› Issue (9): 1456-1470.DOI: 10.3778/j.issn.1673-9418.1912079

• 综述·探索 • 上一篇    下一篇

紧凑的神经网络模型设计研究综述

郎磊,夏应清   

  1. 华中师范大学 物理科学与技术学院,武汉 430079
  • 出版日期:2020-09-01 发布日期:2020-09-07

Survey on Compact Neural Network Model Design

LANG Lei, XIA Yingqing   

  1. College of Physical Science and Technology, Central China Normal University, Wuhan 430079, China
  • Online:2020-09-01 Published:2020-09-07

摘要:

近年来卷积神经网络在广泛的应用中取得了优秀的表现,但巨大的资源消耗量使得其应用于移动端和嵌入式设备成为了挑战。为了解决此类问题,需要对网络模型在大小、速度和准确度方面做出平衡。首先,从模型是否预先训练角度,简要介绍了网络压缩与加速的两类方法——神经网络压缩和紧凑的神经网络。具体地,阐述了紧凑的神经网络设计方法,展示了其中不同运算方式,强调了这些运算的特点,并根据基础运算的不同,将其分为基于空间卷积的模型设计和基于移位卷积的模型设计两大类,然后每类分别选取三个网络模型从基础运算单元、核心构建块和整体网络结构进行论述。同时,分析了各网络以及常规网络在ImageNet数据集上的性能。最后,总结了现有的紧凑神经网络设计技巧,并展望了未来的发展方向。

关键词: 卷积神经网络(CNN), 轻量化, 移位操作, 卷积方式

Abstract:

In recent years, convolutional neural network has achieved excellent performance in a wide range of applications, but it consumes huge resources, which is a challenge for its application to mobile terminals and embedded devices. To this end, network models need to be balanced in size, speed and accuracy. Firstly, two methods of network compression and acceleration, neural network compression and compact neural network, are briefly intro-duced from the perspective of whether the model is pre-trained or not. Specifically, this paper describes the design method of compact neural network, shows the different operation modes, emphasizes the characteristics of these operations, and divides them into two categories: model design based on spatial convolution and model design based on shift convolution according to the different basic operations. Then, each class selects three network models to discuss from the basic operation unit, the core building block and the overall network structure. At the same time, the performance of each network and the conventional network on the ImageNet dataset is analyzed. Finally, this paper summarizes the existing design techniques of compact neural network and looks forward to the future develop-ment direction.

Key words: convolutional neural network (CNN), lightweight, shift operation, convolution method