Deformable Convolution Networks[

2017-04-15 本文已影响2129人 Zliang_hhh

参考资料：

原paper地址：https://arxiv.org/abs/1703.06211

知乎讨论地址：https://www.zhihu.com/question/57493889

csdn blog地址：http://blog.csdn.net/yaoqi_isee/article/details/65963253

Notes地址：https://medium.com/@phelixlau/notes-on-deformable-convolutional-networks-baaabbc11cf3

Abstract

由于构造卷积神经网络 (CNN) 所用的模块中几何结构是固定的，其几何变换建模的能力本质上是有限的。在我们的工作中，我们引入了两种新的模块来提高卷积神经网络 (CNN) 对变换的建模能力，即可变形卷积 (deformable convolution) 和可变形兴趣区域池化 (deformable ROI pooling)。它们都是基于在模块中对空间采样的位置信息作进一步位移调整的想法，该位移可在目标任务中学习得到，并不需要额外的监督信号。新的模块可以很方便在现有的卷积神经网络 (CNN) 中取代它们的一般版本，并能很容易进行标准反向传播端到端的训练，从而得到可变形卷积网络 (deformable convolutional network)。大量的实验验证了我们的方法在目标检测和语义分割这些复杂视觉任务上的有效性。代码地址：这里

1.Introduction

在这项工作中，我们引进了新的模块，大大增强了CNN的几何变换建模能力。首先是可变形卷积。它在标准卷积中向常规采样网格添加了2D偏移。

Figure 1 展示了卷积核大小为 3x3 的正常卷积和可变形卷积的采样方式：(a) 所示的正常卷积规律的采样 9 个点（绿点），(b)(c)(d) 为可变形卷积，在正常的采样坐标上加上一个位移量（蓝色箭头），其中 (c)(d) 作为 (b) 的特殊情况，展示了可变形卷积可以作为尺度变换，比例变换和旋转变换的特殊情况

二是deformable RoI pooling。它在以前的RoI pooling中的常规bin partition中为每个bin添加了一个offset。类似地，从先前的特征图和RoI学习偏移，使得能够对具有不同形状的对象进行自适应部分定位。

两个模块的权值都不多。它们因为加入了偏移的学习添加了少量参数和计算。它们可以轻松集成到深CNN架构中，并通过反向传播进行端到端的训练。这种结构叫做deformable convolutional networks或deformable ConvNets.

两者原理是一样的，就是在这些卷积或者ROI采样层上，添加了位移变量，这个变量根据数据的情况学习，偏移后，相当于卷积核每个方块可伸缩的变化，从而改变了感受野的范围，感受野成了一个多边形。

2.Deformable Convolutional Networks

虽然CNN中的特征映射图和卷积核是3D tensor，但可变卷积和RoI池都在2D空间域上运行的，并且在通道维度上保持不变。

2.1Deformable Convolution

一个2D的卷积由两步组成：

1）使用框架R在输入的特征图上采样

2）将采样值与对应的权重相乘，再求和

R定义了感受野的大小和dilation。例如：

R = {(−1, −1),(−1, 0), . . . ,(0, 1),(1, 1)} 定义了3×3的内核，dilation为1

对于输出特征图y上的每个位置p0，我们有

Figure 2 正常卷积公式

Deformable Conv操作并没有改变卷积的计算操作，而是在卷积操作的作用区域上，加入了一个可学习的参数∆pn。对于变形的卷积，增加了一个参数，即偏移量 {∆pn|n = 1, ..., N}, where N = |R|.

Figure 3 变形卷积公式

然而，这样的操作引入了一个问题，即需要对不连续的位置变量求导。作者在这里借鉴了之前Spatial Transformer Network和若干Optical Flow中warp操作的想法，使用了bilinear插值将任何一个位置的输出，转换成对于feature map的插值操作。

Figure 4 p =p0 + pn + ∆pn for Eq. (2), q enumerates all integral spatial locations in the feature map x

Figure 5 G(·, ·) is the bilinear interpolation kernel.

上式中的g(a, b) = max(0, 1 − |a − b|)，等式3是很容易计算的，大多数的q位置G为0。

总结一下上面四个公式就是，公式1是正常的卷积计算公式，公式2是添加了偏移后的公式，然后对这个公式做线性插值变化就成了3,3的左边是原始图的位置，x(q)是卷积后的特征图，G就是需要学习的变换核。这样就可以求偏导进行梯度下降学习了。4就是这个需要学习的核的表示。

Figure 6 说明了一个3X3的变形卷积，首先通过一个小卷积层（绿色）的输出得到可变形卷积所需要的位移量，然后将其作用在卷积核（蓝色）上，达到可变形卷积的效果

对于输入的一张feature map，假设原来的卷积操作是3*3的，那么为了学习offset，我们定义另外一个3*3的卷积层，输出的offset field其实就是原来feature map大小，channel数等于2（分别表示x,y方向的偏移）。这样的话，有了输入的feature map，有了和feature map一样大小的offset field，我们就可以进行deformable卷积运算。所有的参数都可以通过反向传播学习得到。

2.2 Deformable ROI Pooling

RoI池模块将任意大小的输入矩形区域转换为固定大小的特征。给定一个ROI，大小为w*h，它最后会被均匀分为K*K块，k是个自由参数。标准的ROI pooling是从输入的特征图x中生成k*k个输出特征图y.第（i，j）个块的pooling操作可以被定义为：

Figure 7 standard ROI pooling

p0是左上方的角落块，nij是这个块内的像素值。类似的定义变形的ROI pooling，增加一个偏移量∆pij,如下定义

Figure 8 deformable ROI pooling

Figure 9 描述了一个3*3的deformable ROI pooling

2.3. Deformable ConvNets

可变卷积和RoI池模块都具有与其简单的额版本有着相同的输入和输出。因此，他们可以取代在现有的CNNs中原有的模块的位置。在训练的时候，用于偏移学习的附加conv / fc层初始化为0，他们的学习速率是现有layers的β倍（默认为β= 1）。他们的值是通过bp训练，由双线性插值运算，见公式（3）和（4）。所得到的CNN称为deformable ConvNets。

我们可以把deformable ConvNets分成两步，一是用一个深度全卷积网络从输入的图上得到特征图，二是，用一层具有特定功能的网络从特征图中产生结果。

Figure 10：在两层的标准卷积（a）中使用固定感受野和在可变形卷积（b）中使用可变形的感受野。顶部：顶部功能图上的两个激活单元，在不同尺寸和形状的两个对象上。激活来自3×3过滤器。中间：3×3过滤器在以前的特征图上的采样位置。另外两个激活单元被突出显示。底部：上一个特征图上两个级别的3×3滤波器的采样位置。两组位置突出显示，对应于上面突出显示的单位。