轻量化神经网络模型总结:SqueezeNet、Xception、MobileNet、ShuffleNet


总结今年来的几个轻量化模型:SqueezeNet、Xception、MobileNet、ShuffleNetweb

下面给出时间轴:服务器

  • 2016.02 伯克利&斯坦福提出 SqueezeNet
  • 2016.10 google提出 Xception
  • 2017.04 google提出 MobileNet
  • 2017.07 face++提出 ShuffleNet

其次,说一下模型轻量化的一些方法:网络

  • 卷积核分解:使用1xN和NX1卷积核代替NXN卷积核;
  • 使用深度压缩deep compression方法:网络剪枝、量化、哈弗曼编码;
  • 奇异值分解;
  • 硬件加速器;
  • 低精度浮点数保存;

小模型的好处有哪些:分布式

  • 在分布式训练中,与服务器通讯需求小;
  • 参数少,从云端下载模型的数据量小;
  • 更适合在FPGA等内存首先的嵌入式、移动端设备上部署;

一、SqueezeNet

地址连接:https://arxiv.org/pdf/1602.07360.pdfsvg

1.1 核心思想

在ImageNet上实现了与Alexnet类似的效果,参数只有其1/50, 模型是0.5MB,占其1/510。函数

SqueezeNet核心内容有如下几点:性能

  1. 使用1x1卷积核代替3x3卷积核,减小参数量;
  2. 经过squeeze layer限制通道数量,减小参数量;
  3. 借鉴inception思想,将1x1和3x3卷积后结果进行concat;为了使其feature map的size相同,3x3卷积核进行了padding;
  4. 减小池化层,并将池化操做延后,给卷积层带来更大的激活层,保留更多地信息,提升准确率;
  5. 使用全局平均池化代替全链接层;

上述1-3是经过fire module实现的,fire module主要分为两部分,以下图所示优化

这里写图片描述

  • squeeze:1x1卷积核,参数s_1x1表示卷积核数量
  • expand:1x1卷积核和3x3卷积核,参数e_1x1和e_3x3分别表示两种卷积核的数量

该模块一共三参数s_1x一、e_1x一、e_3x3,关系保持s_1x1< e_1x1+e_3x3google

1.2 网络结构

这里写图片描述

1.3 实验结果

实验结果表示模型小,且准确率不降,反而有点提升;编码

这里写图片描述

参考连接:

一、https://blog.csdn.net/csdnldp/article/details/78648543

二、http://www.noobyard.com/article/p-updyahpy-qd.html

二、Xception

地址连接:https://arxiv.org/abs/1610.02357

2.1 核心思想

虽然本文中方法能够下降参数量,可是论文加宽了网络结构,所以这篇论文不在于压缩模型,旨在于提升性能,与同等参数量的inception v3相比,效果更好。

首先是inception v3的一系列延伸,见下图:

一、版本1:最初的inception v3

这里写图片描述

二、版本2:对1进行简化

这里写图片描述

三、版本3:对2简化,能够先使用一个统一的1x1卷积核,而后每一个3x3卷积核的输入只是1x1卷积后的feature map的一部分。本图中是1/3;

这里写图片描述

四、版本4:在3的基础上进一步延伸,将1x1卷积后的全部feature map按通道所有划分,每个通道对应一个3x3卷积,即3x3卷积核的数量就是1x1卷积后feature map的通道数。

这里写图片描述

而后在xception中,主要采用depthwise separable convolution思想(这个后面在mobile net中详细解释,好奇怪,明明是mobile net后出现的,反正都是一家的,估计公布前后的问题吧。)

首先xception相似于图4,可是区别有两点:

一、xception中没有relu激活函数;
二、图4是先1x1卷积,后通道分离;xception是先进行通道分离,即depthwise separable convolution,而后再进行1x1卷积。

此外,进行残差链接时,再也不是concat,而是采用加法操做。

2.2 网络结构

这里写图片描述
这里写图片描述

2.3 实验结果

这里写图片描述

参考连接:

一、http://www.noobyard.com/article/p-msrksyns-qb.html

二、https://www.baidu.com/link?url=ERPMsc_io0x0usMhZdLD5POp-4p3dHyNtg4z92eeNsIpzxbKJMtmEH39A5op8p2XiQ4CWDPMu03Ygbrs8GAOAK&wd=&eqid=e517c5270000c4ed000000035b7a2a44

三、MobileNet

地址连接:https://arxiv.org/pdf/1704.04861.pdf

3.1 核心思想

主要是两个策略:

  1. 采用depthwise separable convolution,就是分离卷积核;
  2. 设置宽度因子width multipler和分辨率因子resolution multiplier;

3.1.1 depthwise separable convolution

假设上一层获得的feature map的size为D_K * D_K * M,本层的卷积核大小为D_K * D_K,卷积核个数为M。

一、首先介绍传统卷积核的操做方式,以下图。

这里写图片描述

卷积核D_K * D_K须要在feature map的每一个通道上进行D_K * D_K次卷积,而后一共M个卷积核,所以计算量为:

这里写图片描述

二、介绍depthwise separable convolution,以下图

这里写图片描述

将卷积核进行拆解,分为两步,首先用M个D_K * D_K卷积核在feature map进行卷积,计算量为

这里写图片描述

而后再使用N个1 * 1 * M卷积核在前面获得的结果上进行feature map,计算量为:

这里写图片描述

因此,进行分解后的总计算量为:

这里写图片描述

三、计算量比较

这里写图片描述

能够看到,随着卷积核个数的增长,即通道数变多,feature map的大小,传统方式的计算量比分解要大得多。

3.1.2 宽度因子和分辨率因子

怎么才能使网络进一步压缩呢?能够进一步减小feature map的通道数和size,经过宽度因子减小通道数,分辨率因子减小size。

一、宽度因子α

这里写图片描述

二、分辨率因子β

这里写图片描述

两个参数都属于(0,1]之间,当为1时则是标准mobileNet。

3.2 网络结构

3.2.1 基本结构块

这里写图片描述

3.2.2 网络结构

这里写图片描述

3.2.3 训练细节

一、使用RMSprop优化器;

二、未作大量数据加强,由于参数量小过拟合不严重;

三、采用了随机图像裁剪输入;

四、使用较小的weight decay,或者不使用;

3.3 实验结果

给出几个基本的实验比较结果。

这里写图片描述

参考连接:

一、https://blog.csdn.net/t800ghb/article/details/78879612

二、http://www.noobyard.com/article/p-blrswdtr-bk.html

三、http://www.noobyard.com/article/p-wxjuartj-mp.html

四、ShuffleNet

地址连接:https://arxiv.org/pdf/1707.01083.pdf

4.1 核心思想

核心思想有两点:

  1. 借鉴resnext分组卷积思想,但不一样的是采用1x1卷积核;
  2. 进行通道清洗,增强通道间的信息流通,提升信息表示能力。

此外本篇论文中也采起了mobilenet的depthwise separasable convolution的方式。

4.1.1 逐点群卷积pointwise group convolution

这个就是采用resnext的思想,将通道分组,每组分别进行卷积操做,而后再把结果进行concat。可是不一样于resnext的是,shufflenet采用的是1x1卷积核。

4.1.2 通道清洗channel shuffle

什么是通道shuffle,就是在分组卷积后获得的feature map不直接进行concat,先将每组feature map按通道打乱,从新concat,以下图所示:

这里写图片描述

如何进行shuffle,这里参考连接

对于一个卷积层分为g组,

  1. 卷积后一共获得g×n个输出通道的feature map;
  2. 将feature map 进行 reshape为(g,n);
  3. 进行转置为(n,g);
  4. 对转置结果flatten,再分回g组做为下一层的输入。

这里写图片描述

4.2 网络结构

4.2.1 shuffle unit

下图中,a是标准的残差结构,不过是3x3卷积核使用了mobilenet中的depthwise convolution操做;

b是在a的基础上加了本文的通道shuffle操做,先对1x1卷积进行分组卷积操做,而后进行channel shuffle;

c是在旁路加了一步长为2的3x3的平均池化,并将前二者残差相加的操做改成了通道concat,增长了通道数量。

这里写图片描述

4.2.2 网络结构

这里写图片描述

4.3 实验结果

一、评估逐点组卷积:分组的效果均比没有分组的效果好,可是某些模型随着组数增长,性能有降低,这就是通道间失去联系带来的问题;

这里写图片描述

二、评估channel shuffle,shuffle会比没有shuffle效果好,并且对于组数越大,效果越好,说明了shuffle的重要性,也说明了上图中组数增长性能降低的问题。

这里写图片描述

三、与mobilenet的比较

这里写图片描述

参考连接:

一、https://blog.csdn.net/u011974639/article/details/79200559

二、https://blog.csdn.net/u014380165/article/details/75137111