CNN 结构设计技巧-兼顾速度精度与工程实现

技术讨论 Admin ⋅ 于 1周前 ⋅ 97 阅读

目录:

  • 总纲

  • 分割篇

  • low-level篇

  • 检测篇

  • metric learning篇

  • 分类篇

  • landmark篇

  • 视频理解篇

  • 双目篇

  • 3D篇

  • 数据增强篇

总纲

cnn中各个参数的辩证矛盾。

深度决定了网络的表达能力,网络越深学习能力越强。

宽度(通道数)决定了网络在某一层学到的信息量,另外因为卷积层能重组通道间的信息,这一操作能让有效信息量增大(这也是1x1卷积的作用,它能学习出重组信息,使得对于任务更友好,所以这里不能和分离卷积一起比较,传统卷积的有效卷积数更多,正比于输入通道乘以输出通道,分离卷积的有效卷积正比于输入通道数,传统卷积相当于分离卷积前升维再做分离卷积)。

感受野决定了网络在某一层看到多大范围,一般说来最后一层一定至少要能看到最大的有意义的物体,更大的感受野通常是无害的。

在达到相同感受野的情况下,多层小卷积核的性能一定比大卷积核更好,因为多层小卷积核的非线性更强,而且更有利于特征共享。

file

多层小卷积核参数量少,非线性强,有利于特征共享

分辨率很重要,尽量不要损失分辨率,为了保住分辨率,在使用下采样之前要保证在这一层上有足够的感受野,这个感受野是相对感受野,是指这一个下采样层相对于上一个下采样层的感受野,把两个下采样之间看成一个子网络的话,这个子网络必须得有一定的感受野才能将空间信息编码到下面的网络去,而具体需要多大的相对感受野,只能实验,一般说来,靠近输入层的层空间信息冗余度最高,所以越靠近输入层相对感受野应该越小。同时在靠近输入层的层,这里可以合成一个大卷积核来降低计算量。

这种矛盾决定了下面的做法:

前面几层下采样频率高一点,中间层下采样频率降低,并使用不下采样的方法提高深度。

网络能深则深,在保持比较小宽度的时候,要想办法加深网络,变深的过程中网络慢慢变胖。

使用小卷积核(不包括1x1,因为它对于增加感受野无意义),小卷积核有利于网络走向更深,并且有更好的识别鲁棒性,尤其是在分辨率更小的特征图上,因为卷积核的尺寸是相当于特征图的分辨率来说的,大特征图上偏大的卷积核其实也并不大。

下采样在网络前几层的密度大一些,(这样能尽可能用微弱精度损失换取速度提升) 越往后下采样的密度应该更小,最终能够下采样的最大深度,以该层的感受野以及数据集中最大的有意义物体尺寸决定(自然不可能让最大有意义的物体在某一层被下采样到分辨率小于1,但是网络依然可以work,只不过最后几层可能废弃了(要相信cnn的学习能力,因为最大不了它也能学出单位卷积,也就是只有中心元素不为0的卷积核),更准确的说这是最大感受野的极限,最大感受野应该覆盖数据集中最大有意义的物体)。

第一层下采样的时候大卷积核能尽可能保住分辨率(其实相当于合成了两三层小卷积核,另外,这和插值是类似的,类比于最近邻插值,双线性插值,双三次插值,这其实和感受野理论一致,更远的插值意味着更大的感受野)。

file
越靠前分辨率降低越快,中间一定要加深

shortcut connection里,找不到concat,用add凑合吧,在需要量化的场合,add会更好,反之亦然。

先训一个大模型然后裁剪,也许比直接训一个小模型性能好。

能用可分离卷积替代的卷积一定要替代,一般除了第一个卷积,都能替代,替代完后考虑给替代可分离的通道数乘以2,因为可分离卷积的参数和计算量都是线性增长的,这样做依然有速度增益。同样的道理适用于2+1分离卷积。

file

计算量线性增长,可以控制通道数和depth multiplier,代价小

这里稍微提醒一下分离卷积在大型gpu上性能不好原因是gpu利用率,因为大型gpu核心数太多,如果计算量不够大核心没有用完,那么不同卷积层比较不出差异,反而层数带来的延迟会更严重,但是我个人表示相信分离卷积在未来一定会被合成一个层,这只是工程问题。

inception或者shortcut connection或者dense connection其实都相当于ensemble模型,考虑结合使用,shortcut connection的使用几乎是无痛的。

file

各种inception结构,尤其是后面两种,使用空间分离卷积,计算量小且性能好

使用通道注意力,比如squeeze and excitation attention:

file

通道注意力能使通道信息对于任务更友好,尤其是分类任务,semodule能被在任何原网络任何地方插入,计算量只有小幅上涨

事实上,senet是hrnet的一个特例,hrnet不仅有通道注意力,同时也有空间注意力

file

hrnet的基本单元

file

分割篇

网络最后面完全可以使用不采样的空洞卷积来做,对于分割尤其如此。

file

最后面不要再下采样了,会损失很多分辨率,考虑使用aspp来融合不同尺度的感受野

转置卷积完全可以使用上采样+卷积来替代。

file

几种不同的上采样:(a)先用upsamling或者unpooling然后接两个conv,并使用跳转连接,(b)使用upsampling或者unpooling然后接普通的conv,(c)接转置卷积(d)卷积后使用upsampling或者unpooling

这四种上采样方法里面,a的性能最好,但是计算量最大,b的计算量也很大,c和d的计算量相当,但是c很容易产生棋盘格现象。

a和b还有更加快速的版本,可见

file

更快版本的upconv和upproj,具体来说就是用四个小卷积核,分别是3x3,2x3,3x2和2x2来代替大卷积核,然后用interleaving来代替unpooling

interleaving的实现如下,可以参见tensorflow的depth_tospace或者pytorch的pixel_shuffle,这就是子像素卷积。

file

interleaving也就是子像素卷积,它也是一种上采样方法

对于分割模型而言,在encoder和decoder之间使用长跳转连接,在encoder和decoder内部使用短跳转连接,尤其是encoder比较深的情况下。

file

需要短跳转连接,否则中间层无法被有效更新

low-level篇

感受野也不是越大越好(大感受野意味着更多的语义信息),对于某些low-level的任务而言,比如图像降噪,感受野可能也就局部patch大小,对于low-level而言,下采样的意义更小,尽可能少使用下采样。

file

对于low-level而言,感受野不用太大,具体指去噪,去马赛克,图像增强,和关键点等任务


检测篇

======

对于检测,anchor层不一定要有三层,同一层按照需求可以设定的aspect ratio和size都是可选的,比如blazeface使用两层anchor,这个要按照需求设计,事实上不同的层其实代表着不用的scale,三层就是三个scale,两层就是两个scale,一般说来scale分为大中小三种,但是依具体情况而变,主要取决于数据集中的scale分布(这点可能需要修改了,因为根据最新的论文,可能不需要铺更密的anchor,但是需要新的采样算法)。

file
更密集的anchor设置对于密集检测更有利

file

blazeface使用两层anchor

bifpn真的有用,性能有提升且计算量小。

file

检测中各种融合不同scale的方法,bifpn取得了性能和速度的最好平衡

batchnorm一定要用,如果你是多机多卡,也可以考虑同步的batchnorm。

如果你的一阶段检测模型得不到好的分类结果,考虑两阶段,先检测再分类。

检测模型里的预训练模型是有用的,至少能提升前景背景的检测区分度。

anchor free暂不讨论,以后再添加,因为anchor based方法效果已经不错了,并且工程上非常成熟。另外请记住,密集检测没法完全做到anchor free,因为事实上检测目标的可能的位置和宽度高度的分布范围太广 ,所以必须要加上限制,而这个限制本质上就是anchor,另外anchor很容易推广到3d检测。

file

anchor free方法里面需要预测bbox,也需要一定限制,这个限制本质上也是一个anchor


metric learning篇

metric learning(图像比对)一般说来是batchsize更大会性能更好,因为这样能采样到的正负样本对的范围更大。

file

实际使用中,负样本的空间是非常大的,比正样本大很多,所以需要比较大的采样空间



分类篇

如果你的分类精度不够是因为有两类或者多类太相近造成的,考虑使用其他softmax,比如amsoftmax。

file

各种魔改的softmax能更好的增大类间差距,能够更好的分开softmax分不开的类别

如果你的分类精度不够是样本不均衡造成的,考虑使用focal loss。

不要只看精度,考虑其他metrics,并始终以上线后的可视化效果为最终评判标准。

landmark篇

尽可能使用全卷积网络来做landmark,不要直接用fc回归,回归真的不太稳定。另外,数学意义上的cnn是平移不变的,位置信息是通过padding泄露出来,所以fc直接回归landmark相当于用padding去拟合位置信息。

全卷积回归landmark如有必要一定要考虑part affinity fields的做法。

file

全卷积计算landmark,中间两幅图是heatmap高斯分布,以landmark为中心,下面是part affinity map

part affinity fields是替代picturiol model最好的选择,并且它也是全卷积的,它能极大的提高关节点检测的鲁棒性

file

两个全卷积分支,上面预测heatmap,小面预测paf,最后过一个parsing把两支结合起来


以下内容涉及3d卷积:

file

这就是3d卷积,可以看到和2d卷积没有本质差异,只是输入输出都变成了3d

视频理解篇

file

视频理解的常用架构,分为lstm,单流和双流,其中双流使用稠密光流作为一支输入


双目篇(立体匹配)

这里提醒作立体匹配前必须做畸变校正和极线校正

file

原始双目图像的极线都是倾斜的,立体匹配一定要纠正为图像基线平行的平行线

file

视差和深度的关系

视差估计一般分为两个阶段,代价初始值计算构建cost volume(cost-volume也是3d的)和代价聚合,在最简单的架构里面,直接在feature-map使用L1或者L2构建cost volume(也叫做distance-based cost volume)

file

file

在feature map上使用距离度量作为cost

file

多级预测,最深层次预测一个粗视差,然后不断预测residual,disparity网络的主体是3d卷积

file

视差估计网络

3d卷积负责代价匹配的代价其实很大,因为3d卷积的计算量其实比较大。

GANet借鉴sgm(semi-global matching)算法的思想,使用如下公式来进行代价聚合,

file
动态规划进行代价聚合,这个公式可以看成可微分的sgm

file

GA-Net使用GA来进行代价聚合,代替3d卷积,计算量小了非常多


3D篇

lidar篇

lidar最为特殊,因为它的输入是分布不均匀的空间点集,也就是点云,不像2d图像或者3d视频那样,均匀分布。

file

点云存在的问题

  1. 无序性:点云本质上是一长串点(nx3矩阵,其中n是点数)。在几何上,点的顺序不影响它在空间中对整体形状的表示,例如,相同的点云可以由两个完全不同的矩阵表示。

  2. 相同的点云在空间中经过一定的刚性变化(旋转或平移),坐标发生变化,我们希望不论点云在怎样的坐标系下呈现,网络都能得到相同的结果。

也就是说我们希望点云上的神经网络同时具有点集置换不变性和刚性不变性

pointnet的做法,保证置换不变性,去拟合一个对称函数,对称函数有置换不变性,比如x1+x2,同理x1-x2没有置换不变性

file

为了保证置换不变性去拟合一个对称函数

为了保证刚性不变性,类似于spatial transformer network的做法,用一个旁支去拟合一个刚性变换,

file

拟合一个刚性变换,作用在原始点集上,注意升维之后,就不止3个分量了

所有的拟合都是通过mlp进行的,

file

所以本质上它不是一个卷积神经网络,pointnet的通用架构如下

file

pointconv

pointnet不是卷积神经网络,在19年的cvpr上,有这样一篇论文,叫pointconv,它实现了点集上的卷积

file

注意到的是输出的是一个定义在三维平面上的函数,xyz是它的定义域,原始点集是常函数

其中,W 和 F 均为连续函数,(x,y,z)(x,y,z) 是 3D 参考点的坐标,(δx,δy,δz)(δx,δy,δz) 表示邻域 G 中的 3D 点的相对坐标。上式可以离散化到一个离散的 3D 点云上。同时,考虑到 3D 点云可能来自于一个不均匀采样函数,为了补偿不均匀采样,使用逆密度对学到的权重进行加权。PointConv 可以由下式表示,

file

离散化后如下所示

file

其中,S 表示逆密度系数函数。连续函数 W 可以用多层感知器(MLP)近似。函数 W 的输入是以 (x, y, z) 为中心的 3D 邻域内的 3D 点的相对坐标,输出是每个点对应的特征 F 的权重。S 是一个关于密度的函数,输入是每个点的密度,输出是每个点对应的逆密度系数。这个非线性的函数同样可以用一个多层感知机近似。

file

pointconv是真正的卷积操作,可以期待以后在其上可以做3d分类,分割,检测,和2d卷积网络使用方法几乎一致。

数据增强篇

一定要做图像亮度变换增强,亮度鲁棒性会更好,但是要结合你的数据的亮度分布自适应的调整。

最后,在不改动网络backbone的基础上,多尝试一些新的loss,工程上的代价并不大。

来源:https://zhuanlan.zhihu.com/p/100609339

作者:akkaze-郑安坤

本文来自知乎专栏,仅供学习参考使用,著作权归作者所有。如有侵权,请私信删除。



-End-


*延伸阅读
相关文章:
CNN 的平移不变性详解
ECCV2018|ShuffleNetV2:轻量级 CNN 网络中的桂冠


file
△ 关注极市平台
获得最新CV干货art.net/uploads/images/202001/14/3/hS2PC78VBf.gif?imageView2/2/w/1240/h/0)

微信公众号: 极市平台(ID: extrememart )
每天推送最新CV干货

成为第一个点赞的人吧 :bowtie:
回复数量: 0
暂无回复~
您需要登陆以后才能留下评论!