Bag of Tricks for Image Classification with Convolutional Neural Networks笔记

以下内容摘自《Bag of Tricks for Image Classification with Convolutional Neural Networks》。

1 高效训练

1.1 大batch训练

当我们有一定资源后,当然希望能充分利用起来,所以通常会增加batch size来达到加速训练的效果。但是,有不少实验结果表明增大batch size可能降低收敛率,所以为了解决这一问题有人以下方法可供选择:

1.1.1 线性增加学习率

一句话概括就是batch size增加多少倍,学习率也增加多少倍

例如最开始batch size=256时, lr=0.1。后面batch size变成b后,lr=\(0.1\times \frac{b}{256}\)

1.1.2 学习率warmup

在训练最开始,所有的参数都是随机初始化的,因此离最终的解还需要很多次迭代才能到达。一开始使用一个太大的学习率会导致不稳定,所以可以在最开始使用一个很小的学习率,然后逐步切换到默认的初始学习率。

例如我们可以设定在前面100个batch学习率由0线性增加到默认初始值0.1,也就是说每个batch学习率增加0.001

1.1.3 \(\gamma=0\) for BN layer

现在很多网络结构中都会用到BatchNorm层,而我们知道BN层会对数据作如下操作:

1.标准化输入: \(\hat{x_1}=\text{standardize}(x)\)
2.伸缩变换(scale transformation): \(\hat{x_2}=\gamma\hat{x_1}+\beta\)

上面步骤中的\(\gamma,\beta\)都是可学习的参数,分别初始化为0,1。

原文解释这样做的好处如下(没太懂什么意思,懂得麻烦解释一下):

Therefore, all residual blocks just return their inputs, mimics network that has less number of layers and is easier to train at the initial stage.

1.1.4 No bias decay

意思就是我们可以weight decay,但是不用做bias decay

1.2 低精度训练

P. Micikevicius,et al. Mixed precision training.中提出将所有的参数和激活值用FP16存储,并且使用FP16来计算梯度。另外,所有的参数都需要一份FP32的拷贝用来做参数更新。另外将loss乘上一个标量以便求得的梯度范围可以与FP16对齐。

2. 模型设计Trick

下面以ResNet为例进行介绍有哪些模型设计的Trick。

下图示何凯明提出的ResNet的示意图,可以看到主要由三部分组成:

  • input stem:主要是先用一个7*7的卷积核,后接一个最大池化
  • stage2-4:每个stage最开始是先用下采样模块对输入数据维度做减半操作,注意不是直接使用池化,而是通过设置卷积stride=2实现的。下采样之后后接若干个残差模块。
  • output:最后接上预测输出模块

ikyf7dhzwr.png

下面介绍一下可以对上面的ResNet做哪些魔改:

  • input stem改!

可以看到主要将7*7的卷积核改成了3个3*3的卷积核,这样效果类似,而且参数更少。

image.png

  • stage中的down sampling部分的Path A和Path B都可以改:
    • Path A:将stride=2从原来的1*1卷积核部分改到了3*3卷积核内
      image.png
    • Path B:该条路径只有一个卷积操作,没法像上面那样,所以加入了一个池化层。
      image.png

实验结果:

Model#paramsFLOPsTop-1Top-5
ResNet-5025 M3.8 G76.2192.97
ResNet-50-B25 M4.1 G76.6693.28
ResNet-50-C25 M4.3 G76.8793.48
ResNet-50-D25 M4.3 G77.1693.52

3. 训练Trick

3.1 Cosine learning rate decay

计算公式如下:

\[ \eta_{t}=\frac{1}{2}\left(1+\cos \left(\frac{t \pi}{T}\right)\right) \eta \]
其中\(\eta\)是初始化学习率。

示意图如下:

image.png

3.2 label smoothing

这个方法我在之前的一个项目中用过,但是感觉没什么效果,所以各种Trick还是得视情况而定,并不是万能的。

通常分类任务中每张图片的标签是one hot形式的,也就是说一个向量在其对应类别索引上设置为1,其他位置为0,形如[0,0,0,1,0,0]。

label smoothing就是将类别分布变得平滑一点,即

\[ q_{i}=\left\{\begin{array}{ll}{1-\varepsilon} & {\text { if } i=y} \\ {\varepsilon /(K-1)} & {\text { otherwise }}\end{array}\right. \]
其中\(q_{i}\)就代表某一类的ground truth,例如如果\(i==y\),那么其最终真实值就是\(1-\varepsilon\),其它位置设置为\(\varepsilon /(K-1)\),而不再是。这里的\(\varepsilon\)是一个很小的常数。

3.3 Mixup

方法如下:

每次随机抽取两个样本进行加权求和得到新的样本,标签同样做加权操作。公式中的\(\lambda\in[0,1]\)是一个随机数,服从\(\text{Beta}(\alpha,\alpha)\)分布。

\[ \begin{aligned} \hat{x} &=\lambda x_{i}+(1-\lambda) x_{j} \\ \hat{y} &=\lambda y_{i}+(1-\lambda) y_{j} \end{aligned} \]



MARSGGBO原创





2019-9-4



转载于:https://www.cnblogs.com/marsggbo/p/11459951.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值