YOLOV4所用到的一些tricks

原文链接http://arxiv.org/abs/2004.10934

整体框架

      

Bag of Freebies(BoF) & Bag  of Specials (BoS)

  Bag of Freebies(BoF)  指那些能够提高精度而不增加推断时间的技术。比如数据增广的方法图像几何变换、CutOut、grid mask等,网络正则化的方法DropOut、DropBlock等,类别不平衡的处理方法、难例挖掘方法、损失函数的设计等。

  Bag  of Specials (BoS)是指那些增加稍许推断代价,但可以提高模型精度的方法,比如增大模型感受野的SPP、ASPP、RFB等,引入注意力机制Squeeze-and-Excitation (SE) 、Spatial Attention Module (SAM)等 ,特征集成方法SFAM , ASFF , BiFPN等,改进的激活函数Swish、Mish等,或者是后处理方法如soft NMS、DIoU NMS等。

Backbone: CSPDarknet53

source: CSPNet: A new backbone that can enhance learning capability of cnn论文;

CSPNet提出主要是为了解决三个问题: a.增强CNN的学习能力,能够在轻量化的同时保持准确性。

                  b.降低计算瓶颈。

                  c.降低内存成本。 

CSP 在 DenseNet的使用

Neck: SPP(Spatial pyramid pooling) 、PANet(Path Aggregation Network)

source: 

    SPP:Spatial pyramid pooling in deep convolutional networks for visual recognition。

       PANet: Path Aggregation Network for Instance Segmentation .(实力分割中提出)

PANet 网络框架

  关键:更好的利用特征融合 1.为了提高低层信息的利用率,加快低层信息的传播效率,提出了Bottom-up Path Augmentation; 2.通常FPN在多层进行选anchors时,根据anchors的大小,将其分配到对应的层上进行分层选取。这样做很高效,但同时也不能充分利用信息了,提出了Adaptive Feature Pooling。 3.为了提高mask的生成质量,将卷积-上采样和全连接层进行融合,提出了Fully-connected Fusion。

CutMix and Mosaic data augmentation, DropBlock regularization, Class label smoothing

source:

    CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Features (开源)

Mixup:将随机的两张样本按比例混合,分类的结果按比例分配(个人看法:如果是使用一个随机数来控制权重,可能AP会增);

Cutout:随机的将样本中的部分区域cut掉,并且填充0像素值,分类的结果不变;

CutMix:就是将一部分区域cut掉但不填充0像素而是随机填充训练集中的其他数据的区域像素值,分类结果按一定的比例分配

Mosaic data augmentation:可以见论文给的示例图,提出了一种混合四幅训练图像的数据增强方法。

DropBlock regularization(来自知乎):

                                  各种Dropout 组合

Mish activation, Cross-stage partial connections (CSP), Multi- input weighted residual connections (MiWRC)

source:

    Mish: A Self Regularized Non-Monotonic Neural Activation Function论文(开源)

激活函数公式:Mish=x * tanh(ln(1+e^x))

  描述:x轴无边界(即正值可以达到任何高度)避免了由于封顶而导致的饱和。理论上对负值的轻微允许允许更好的梯度流,而不是像ReLU中那样的硬零边界。最后,平滑的激活函数允许更好的信息深入神经网络,从而得到更好的准确性和泛化。

 

Mish 激活函数的曲线

CSP、MiWRC 都是一种让网络更好训练的跳接方式。

CIoU-loss, CmBN

发展历程: IoU loss -> Giou loss -> Diou loss & Cious loss

主要依据:geometric factors:overlap area, central point distance and aspect ratio

IOU loss :依据交并比的loss, 不做介绍了。

Giou loss:

                                   GIOU核心算法

Diou loss

切入点:

第一:直接最小化预测框与目标框之间的归一化距离是否可行,以达到更快的收敛速度。

第二:如何使回归在与目标框有重叠甚至包含时更准确、更快。

 

R(B, Bgt)为预测框与真实框的惩罚项。

b, bgt分别为框的中心,P^2即两个点的欧式距离,c^2为包含两个框的最小框的对角线长度。

 Ciou loss 

在Diou loss 的基础上考虑了—— the consistency of aspect ratio   即公式中的 ν参数,α 为权衡参数。 

CmBN

CMBN示意图

Self-Adversarial Training (SAT)

  SAT 为一种新型数据增强方式。在第一阶段,神经网络改变原始图像而不是网络权值。通过这种方式,神经网络对其自身进行一种对抗式的攻击,改变原始图像,制造图像上没有目标的假象。在第二阶段,训练神经网络对修改后的图像进行正常的目标检测。

注意力机制——SAM、PAN 

所做改进: 将SAM从空间注意力机制修改为点上的注意力机制,并将PAN的相加模块改为级联。

读完结论

一系列堆料,结果建立在好的backbone 上面,好的backbone 再加入一系列训练提高方式, 必然长点。

 

原文地址:https://www.cnblogs.com/E-Dreamer-Blogs/p/12814665.html