ConvNeXt网络详解

ConvNeXt

论文名称：

A ConvNet for the 2020s

论文下载链接：https://arxiv.org/abs/2201.03545
论文对应源码链接：https://github.com/facebookresearch/ConvNeXt
视频讲解：https://www.bilibili.com/video/BV1SS4y157fu

文章目录

0 前言

自从

ViT(Vision Transformer)

在CV领域大放异彩，越来越多的研究人员开始拥入

Transformer

的怀抱。回顾近一年，在CV领域发的文章绝大多数都是基于

Transformer

的，比如2021年ICCV 的best paper

Swin Transformer

，而卷积神经网络已经开始慢慢淡出舞台中央。卷积神经网络要被

Transformer

取代了吗？也许会在不久的将来。今年(2022)一月份，

Facebook AI Research

和

UC Berkeley

一起发表了一篇文章

A ConvNet for the 2020s

，在文章中提出了

ConvNeXt

纯卷积神经网络，它对标的是2021年非常火的

Swin Transformer

，通过一系列实验比对，在相同的FLOPs下，

ConvNeXt

相比

Swin Transformer

拥有更快的推理速度以及更高的准确率，在

ImageNet 22K

上

ConvNeXt-XL

达到了

87.8%

的准确率，参看下图(原文表12)。看来

ConvNeXt

的提出强行给卷积神经网络续了口命。

如果你仔细阅读了这篇文章，你会发现

ConvNeXt

“毫无亮点”，

ConvNeXt

使用的全部都是现有的结构和方法，没有任何结构或者方法的创新。而且源码也非常的精简，100多行代码就能搭建完成，相比

Swin Transformer

简直不要太简单。之前看

Swin Transformer

时，滑动窗口，相对位置索引，不光原理理解起来有些吃力，源码多的也挺让人绝望的（但无法否认

Swin Transformer

的成功以及设计的很巧妙）。为什么现在基于

Transformer

架构的模型效果比卷积神经网络要好呢？论文中的作者认为可能是随着技术的不断发展，各种新的架构以及优化策略促使

Transformer

模型的效果更好，那么使用相同的策略去训练卷积神经网络也能达到相同的效果吗？抱着这个疑问作者就以

Swin Transformer

作为参考进行一系列实验。

In this work, we investigate the architectural distinctions between ConvNets and Transformers and try to identify the confounding variables when comparing the network performance. Our research is intended to bridge the gap between the pre-ViT and post-ViT eras for ConvNets, as well as to test the limits of what a pure ConvNet can achieve.

1 设计方案

作者首先利用训练

vision Transformers

的策略去训练原始的

ResNet50

模型，发现比原始效果要好很多，并将此结果作为后续实验的基准

baseline

。然后作者罗列了接下来实验包含哪些部分：

• macro design
• ResNeXt
• inverted bottleneck
• large kerner size
• various layer-wise micro designs

Our starting point is a ResNet-50 model. We first train it with similar training techniques used to train vision Transformers and obtain much improved results compared to the original ResNet-50. This will be our baseline. We then study a series of design decisions which we summarized as 1) macro design, 2) ResNeXt, 3) inverted bottleneck, 4) large kernel size, and 5) various layer-wise micro designs.

下图（原论文图2）展现了每个方案对最终结果的影响（Imagenet 1K的准确率）。很明显最后得到的

ConvNeXt

在相同FLOPs下准确率已经超过了

Swin Transformer

。接下来，针对每一个实验进行解析。

2 Macro design

在这个部分作者主要研究两方面：

• Changing stage compute ratio，在原ResNet网络中，一般conv4_x（即stage3）堆叠的block的次数是最多的。如下图中的ResNet50中stage1到stage4堆叠block的次数是(3, 4, 6, 3)比例大概是1:1:2:1，但在Swin Transformer中，比如Swin-T的比例是1:1:3:1，Swin-L的比例是1:1:9:1。很明显，在Swin Transformer中，stage3堆叠block的占比更高。所以作者就将ResNet50中的堆叠次数由(3, 4, 6, 3)调整成(3, 3, 9, 3)，和Swin-T拥有相似的FLOPs。进行调整后，准确率由78.8%提升到了79.4%。
• Changing stem to “Patchify”，在之前的卷积神经网络中，一般最初的下采样模块stem一般都是通过一个卷积核大小为7x7步距为2的卷积层以及一个步距为2的最大池化下采样共同组成，高和宽都下采样4倍。但在Transformer模型中一般都是通过一个卷积核非常大且相邻窗口之间没有重叠的（即stride等于kernel_size）卷积层进行下采样。比如在Swin Transformer中采用的是一个卷积核大小为4x4步距为4的卷积层构成patchify，同样是下采样4倍。所以作者将ResNet中的stem也换成了和Swin Transformer一样的patchify。替换后准确率从79.4% 提升到79.5%，并且FLOPs也降低了一点。

3 ResNeXt-ify

接下来作者借鉴了

ResNeXt

中的组卷积

grouped convolution

，因为

ResNeXt

相比普通的

ResNet

而言在FLOPs以及accuracy之间做到了更好的平衡。而作者采用的是更激进的

depthwise convolution

，即group数和通道数channel相同，之前在讲

MobileNet

时由很详细讲解过，如果不了解的可以翻翻之前的文章。这样做的另一个原因是作者认为

depthwise convolution

和

self-attention

中的加权求和操作很相似。

We note that depthwise convolution is similar to the weighted sum operation in self-attention.

接着作者将最初的通道数由64调整成96和

Swin Transformer

保持一致，最终准确率达到了

80.5%

。

4 Inverted Bottleneck

作者认为

Transformer block

中的

MLP

模块非常像

MobileNetV2

中的

Inverted Bottleneck

模块，即两头细中间粗。下图a是

ReNet

中采用的

Bottleneck

模块，b是

MobileNetV2

采用的

Inverted Botleneck

模块（图b的最后一个

1x1

的卷积层画错了，应该是

384->96

，后面如果作者发现后应该会修正过来），c是

ConvNeXt

采用的是

Inverted Bottleneck

模块。关于

MLP

模块可以参考之前讲的Vision Transformer博文，关于

Inverted Bottleneck

模块可以参考之前讲的MobileNetv2博文。

作者采用

Inverted Bottleneck

模块后，在较小的模型上准确率由

80.5%

提升到了

80.6%

，在较大的模型上准确率由

81.9%

提升到

82.6%

。

Interestingly, this results in slightly improved performance (80.5% to 80.6%). In the ResNet-200 / Swin-B regime, this step brings even more gain (81.9% to 82.6%) also with reduced FLOPs.

5 Large Kernel Sizes

在

Transformer

中一般都是对全局做

self-attention

，比如

Vision Transformer

。即使是

Swin Transformer

也有

7x7

大小的窗口。但现在主流的卷积神经网络都是采用

3x3

大小的窗口，因为之前

VGG

论文中说通过堆叠多个

3x3

的窗口可以替代一个更大的窗口，而且现在的GPU设备针对

3x3

大小的卷积核做了很多的优化，所以会更高效。接着作者做了如下两个改动：

• Moving up depthwise conv layer，即将depthwise conv模块上移，原来是1x1 conv -> depthwise conv -> 1x1 conv，现在变成了depthwise conv -> 1x1 conv -> 1x1 conv。这么做是因为在Transformer中，MSA模块是放在MLP模块之前的，所以这里进行效仿，将depthwise conv上移。这样改动后，准确率下降到了79.9%，同时FLOPs也减小了。
• Increasing the kernel size，接着作者将depthwise conv的卷积核大小由3x3改成了7x7（和Swin Transformer一样），当然作者也尝试了其他尺寸，包括3, 5, 7, 9, 11发现取到7时准确率就达到了饱和。并且准确率从79.9% (3×3) 增长到 80.6% (7×7)。

6 Micro Design

接下来作者在聚焦到一些更细小的差异，比如激活函数以及Normalization。

• Replacing ReLU with GELU，在Transformer中激活函数基本用的都是GELU，而在卷积神经网络中最常用的是ReLU，于是作者又将激活函数替换成了GELU，替换后发现准确率没变化。
• Fewer activation functions，使用更少的激活函数。在卷积神经网络中，一般会在每个卷积层或全连接后都接上一个激活函数。但在Transformer中并不是每个模块后都跟有激活函数，比如MLP中只有第一个全连接层后跟了GELU激活函数。接着作者在ConvNeXt Block中也减少激活函数的使用，如下图所示，减少后发现准确率从80.6%增长到81.3%。
• Fewer normalization layers，使用更少的Normalization。同样在Transformer中，Normalization使用的也比较少，接着作者也减少了ConvNeXt Block中的Normalization层，只保留了depthwise conv后的Normalization层。此时准确率已经达到了81.4%，已经超过了Swin-T。
• Substituting BN with LN，将BN替换成LN。Batch Normalization（BN）在卷积神经网络中是非常常用的操作了，它可以加速网络的收敛并减少过拟合（但用的不好也是个大坑）。但在Transformer中基本都用的Layer Normalization（LN），因为最开始Transformer是应用在NLP领域的，BN又不适用于NLP相关任务。接着作者将BN全部替换成了LN，发现准确率还有小幅提升达到了81.5%。
• Separate downsampling layers，单独的下采样层。在ResNet网络中stage2-stage4的下采样都是通过将主分支上3x3的卷积层步距设置成2，捷径分支上1x1的卷积层步距设置成2进行下采样的。但在Swin Transformer中是通过一个单独的Patch Merging实现的。接着作者就为ConvNext网络单独使用了一个下采样层，就是通过一个Laryer Normalization加上一个卷积核大小为2步距为2的卷积层构成。更改后准确率就提升到了82.0%。

7 ConvNeXt variants

对于

ConvNeXt

网络，作者提出了

T/S/B/L

四个版本，计算复杂度刚好和

Swin Transformer

中的

T/S/B/L

相似。

We construct different ConvNeXt variants, ConvNeXt-T/S/B/L, to be of similar complexities to Swin-T/S/B/L.

这四个版本的配置如下：

• ConvNeXt-T: C = (96, 192, 384, 768), B = (3, 3, 9, 3)
• ConvNeXt-S: C = (96, 192, 384, 768), B = (3, 3, 27, 3)
• ConvNeXt-B: C = (128, 256, 512, 1024), B = (3, 3, 27, 3)
• ConvNeXt-L: C = (192, 384, 768, 1536), B = (3, 3, 27, 3)
• ConvNeXt-XL: C = (256, 512, 1024, 2048), B = (3, 3, 27, 3)

其中C代表4个

stage

中输入的通道数，B代表每个

stage

重复堆叠block的次数。

8 ConvNeXt-T 结构图

下图是我根据源码手绘的

ConvNeXt-T

网络结构图，仔细观察

ConvNeXt Block

会发现其中还有一个

Layer Scale

操作（论文中并没有提到），其实它就是将输入的特征层乘上一个可训练的参数，该参数就是一个向量，元素个数与特征层channel相同，即对每个channel的数据进行缩放。

Layer Scale

操作出自于

Going deeper with image transformers. ICCV, 2021

这篇文章，有兴趣的可以自行了解。