CVPR2020 | 超越MobileNetV3的轻量级网络(文末论文下载)

最新推荐文章于 2024-07-17 14:22:21 发布

计算机视觉研究院

最新推荐文章于 2024-07-17 14:22:21 发布

阅读量2.1k

点赞数

本文链接：https://blog.csdn.net/gzq0723/article/details/104765214

版权

CVPR2020将在6月份举行，现在已经公布了收录文章，虽然有很多低质量的文章，但是，“计算机视觉战队”从其中找一些好的算法框架与大家分享，今天分享的就是超越Mobilenet_V3的轻量级网络——GhostNet。

虽然说mobileNet或者是shuffleNet提出了使用depthwise或者是shuffle等操作，但是引入的1x1卷积依然会产生一定的计算量。

为什么1x1依然会产生较大的计算量?

看卷积计算量的计算公式n∗h∗w∗c∗k∗kn*h*w*c*k*kn∗h∗w∗c∗k∗k,可以发现，由于c和n都是比较大的，所以会导致这个计算量也是比较大的，但是作者在分析输出的特征图的时候发现，其实有些特征图是比较相似的。

Introduction

目前，神经网络的研究趋向于移动设备上的应用，一些研究着重于模型的压缩方法，比如剪枝，量化，知识蒸馏等，另外一些则着重于高效的网络设计，比如MobileNet，ShuffleNet等。作者在分析输出的特征图的时候发现，其实有些特征图是比较相似的，如下图所示，作者认为可以通过简单的变换得到。

训练好的网络一般都有丰富甚至冗余的特征图信息来保证对输入的理解，如上图ResNet-50的特征图，相似的特征图类似于对方的ghost。冗余的特征是网络的关键特性，论文认为与其避免冗余特征，不如以一种cost-efficient的方式接受，获得很不错的性能提升，论文主要有两个贡献：

提出能用更少参数提取更多特征的Ghost模块，首先使用输出很少的原始卷积操作(非卷积层操作)进行输出，再对输出使用一系列简单的线性操作来生成更多的特征。这样，不用改变其输出的特征图，Ghost模块的整体的参数量和计算量就已经降低了；
基于Ghost模块提出GhostNet，将原始的卷积层替换为Ghost模块。

基于此，作者得到启发，是不是每张图都需要用这么大的计算量去得到？是否可以通过cheap transformation得到这些相似的特征图？

然后就出现了GhostNet。Ghost就是说相似的特征图，犹如另一个的幽灵，可以通过简单的线性变换得到。

Ghost Module for More Features

对于输入数据X∈ℝc×h×w，卷积层操作如上公式，Y∈ℝh′×w′×n为输出的n维特征图，f∈ℝc×k×k×nf为该层的卷积核，可得该层的计算量为n⋅h′⋅w′⋅c⋅k⋅k，这个结果一般较大，是由于n和c一般都很大。上面公式的参数量与输入和输出的特征图数息息相关，而从图1可以看出中间特征图存在大量冗余，且存在相似的特征(Ghost)，所以完全没必要占用大量计算量来计算这些Ghost。

假设原输出的特征为某些内在特征进行简单的变换得到Ghost，通常这些内在特征数量都很少，并且能通过原始卷积操作如下公式获得，Y′∈ℝh′×w′×m为原始卷积输出，f′∈ℝc×k×k×m为使用的卷积核，m≤n，bias直接简化了：

为了获得原来的n维特征，对Y′的内在特征分别使用一系列简单线性操作来产生s维ghost特征，Φi,j为生成y′i的j-th ghost特征图的线性变换函数，最后的Φi,s为保存内在特征的identity mapping，整体计算如上图2b。

简单的来说就是：

首先，假设我们输入特征图的尺寸是h*w*c,输出特征图的尺寸是h’*w’*n,卷积核大小为k*k。

在cheap operation变换中，我们假设特征图的channel是m，变换的数量是s，最终得到的新的特征图的数量是n,那么我们可以得到等式：

n=m∗sn = m * s

n=m∗s

由于Ghost的变换过程中最后存在一个恒等变换（Identity），所以实际有效的变换数量是s-1,所以上式可以得到如下公式：

m∗(s−1)=n/s∗(s−1)m * (s-1) = n/s * (s-1)

m∗(s−1)=n/s∗(s−1)

所以我们便可以计算得到如下结果：

当然这里还有一个条件：m << n

通过这么分析，可以体会到，其实GhostNet的方法也很简单，无外乎就是将原本的乘法变成了两个乘法相加，然后在代码实现中，其实第二个变换是用depthwise conv实现的。作者在文中也提到，前面的卷积使用pointwise效率比较高，所以网络嫣然类似一个mobilenet的反过来的版本，只不过GhostNet采用了拼接的方式，进一步减少了计算量。

Ghost module的pytorch代码如下：

class GhostModule(nn.Module):
    def __init__(self, inp, oup, kernel_size=1, ratio=2, dw_size=3, stride=1, relu=True):
        super(GhostModule, self).__init__()
        self.oup = oup
        init_channels = math.ceil(oup / ratio)
        new_channels = init_channels*(ratio-1)


        self.primary_conv = nn.Sequential(
            nn.Conv2d(inp, init_channels, kernel_size, stride, kernel_size//2, bias=False),
            nn.BatchNorm2d(init_channels),
            nn.ReLU(inplace=True) if relu else nn.Sequential(),
        )


        self.cheap_operation = nn.Sequential(
            nn.Conv2d(init_channels, new_channels, dw_size, 1, dw_size//2, groups=init_channels, bias=False),
            nn.BatchNorm2d(new_channels),
            nn.ReLU(inplace=True) if relu else nn.Sequential(),
        )


    def forward(self, x):
        x1 = self.primary_conv(x)
        x2 = self.cheap_operation(x1)
        out = torch.cat([x1,x2], dim=1)
        return out[:,:self.oup,:,:]

与现有网络的不同

与目前主流的卷积操作对比，Ghost模块有以下不同点：

对比Mobilenet、Squeezenet和Shufflenet中大量使用1×1pointwise卷积，Ghost模块的原始卷积可以自定义卷积核数量；
目前大多数方法都是先做pointwise卷积降维，再用depthwise卷积进行特征提取，而Ghost则是先做原始卷积，再用简单的线性变换来获取更多特征；
目前的方法中处理每个特征图大都使用depthwise卷积或shift操作，而Ghost模块使用线性变换，可以有很大的多样性；
Ghost模块同时使用identity mapping来保持原有特征。

框架结构

下图是Ghost bottleneck结构图，很类似resnet结构，不同的是channel是先升维再降维。

Ghost Bottleneck(G-bneck)与residual block类似，主要由两个Ghost模块堆叠二次，第一个模块用于增加特征维度，增大的比例称为expansion ration，而第二个模块则用于减少特征维度，使其与shortcut一致。

G-bneck包含stride=1和stride=2版本，对于stride=2，shortcut路径使用下采样层，并在Ghost模块中间插入stride=2的depthwise卷积。为了加速，Ghost模块的原始卷积均采用pointwise卷积。

下面是GhostNet的网络结构图，可以看到channel控制的比较小，并且引入了SE结构。

实验

作者首先采用控制变量法，测试不同的s以及d的效果。经过测试发现在s=2,d=3的情况下模型表现较好。

下表为在resnet50实验的不同网络的压缩比例以及精度对比：

可以看到使用Ghost模块不仅比其它压缩方法更能降低模型的体量，也最能保持模型准确率：

The feature maps in the 2nd layer of Ghost-VGG-16. The left-top image is the input, the feature maps in the left red box are from the primary convolution, and the feature maps in the right green box are after the depthwise transformation.

ImageNet效果对比：

目标检测的效果：

小结

为了减少神经网络的计算消耗，论文提出Ghost模块来构建高效的网络结果。该模块将原始的卷积层分成两部分，先使用更少的卷积核来生成少量内在特征图，然后通过简单的线性变化操作来进一步高效地生成ghost特征图。从实验来看，对比其它模型，GhostNet的压缩效果最好，且准确率保持也很不错，论文思想十分值得参考与学习。

论文地址：https://arxiv.org/pdf/1911.11907.pdf