本文主要内容是 CNN 的 BP 算法,看此文章前请保证对CNN有初步认识。
网络表示
CNN相对于传统的全连接DNN来说增加了卷积层与池化层,典型的卷积神经网络中(比如LeNet-5 ),开始几层都是卷积和池化的交替,然后在靠近输出的地方做成全连接网络,这时候已经将所有两维2D的特征maps转化为全连接的一维网络的输入。在前向传导或城中中,也只有两处与传统的 MLP 有所不同,分别是卷积层前向传导,与 pooling 传导到卷积层,如下图所示:
在上图中,层 (l−1)(l−1) 可以为pooling 层或是输入层,(l)(l) 层对 (l−1)(l−1) 层进行卷积操作,(l+1)(l+1) 对 (l)(l) 进行pooling操作。
符号表示
K(l−1)K(l−1):(l−1)(l−1) 层用到的卷积核,即(l−1)(l−1) 层的权值参数;
K(l−1)ijKij(l−1):从 (l−1)(l−1)层映射到 (l)(l) 层Feature Map j 所用到的卷积核 jj 的第 ii 个卷积模板;
b(l−1)b(l−1):(l−1)(l−1) 层的偏置参数;
b(l−1)jbj(l−1):映射到 (l)(l) 层中的 Feature Map j 的偏置参数;
w(l)w(l):表示 (l)(l) 层的参数;
w(l)jwj(l):卷积层第 j 个Feature Map 到 pooling 层第 j 个 Feature Map 是一一对应关系,即每个Feature Map 对应一个w(l)jwj(l);
b(l)b(l):(l)(l) 层的偏置参数;
b(l)jbj(l):卷积层第 j 个Feature Map 到pooling层第 j 个 Feature Map的偏置参数;
z(l)z(l):(l)(l) 层的输入;
a(l)a(l):(l)(l) 层的输出.
前向传导
CNN 的前向传导,首先从 (l−1)(l−1) 层到 (l)(l) 层,此处一般为多对多的形式,即对多个通道进行卷积操作,生成多个Feature Map,(l)(l)层第 j 个Feature Map的输入 z(l)jzj(l) 的计算如下:
需要注意:这里 z(l)j,a(l−1)i,K(l−1)ijzj(l),ai(l−1),Kij(l−1) 均为向量形式," * "表示卷积操作,z(l)jzj(l) 代表 第j 个 Feature Map 的输入,a(l−1)iai(l−1) 代表第 (l−1)(l−1) 层第 i 个 Feature Map的输出,Kl−1ijKijl−1 代表第 j 个卷积核中第 i 个卷积模板,要生成 (l)(l) 层的第 j 个Feature Map,需要对 (l−1)(l−1) 层的 MjMj 个通道同时进行卷积操作,对得到的结果加偏置求和即可。
第 (l)(l) 层的输出只需要对输入z(l)jzj(l) 做一个激活函数即可:
综上,从层 到层可以用下图描述:
接下来从到传导到第层,此处对应pooling 操作,一般为一对一的形式,层的第 N 个 Feature Map 生成层的N个 Feature Map ,计算过程如下:
这里down(.)对应的是一个下采样(sub-sampling) 函数,这个函数会对层的输出中的n-n的像素块进行pooling操作,使得输出在横纵方向都缩小n倍。经过down(.)操作后,与有相同的 size 根据以上公式计算即可。同理,输出继续做一个映射:
至此,前向传导过程已经可以求得,接下来与MLP类似,就是根据链式求导法则,求得残差的反向传导。
前馈 (Back Propagation)
对于 pooling 层,假设已知其第 j 个Feature Map 的残差向量 ,将其传导到,即pooling层的误差传导到卷积层:
比如对于,即层的第 j 个 Feature Map 的残差项如下图左,对其做 操作,得到下图右:
后
操作与相反, 操作会还原的残差map,使其与卷积层的map大小一致,即还原到down(.)之前的大小。这里残差为一对一传递,将层 的map的激活函数的偏导数与从第 层的得到的残差map逐元素相乘,最后乘以参数即可,注意这里为一个常量,每个Feature Map j 对应一个参数。
卷积层与层一般为多对多的关系,根据以上分析求得了卷积层的残差项,接下来就是卷积层的残差反向传导的过程。卷积层前面分两种情况,1)若卷积层前面为pooling层,则上面假设已知的就是根据这里的推导所求得,并不是凭空出现。2)卷积层前面为初始输入。
首先回忆卷积过程,卷积核 j 会对层多个通道进行卷积(每个通道对应 j 的一个卷积模板)求和,才会得到得到层 的第 j 个 Feature Map,所以层的通道 i 的残差项应该由与之相关的层 的所有Feature Map j 共同前向传导。假设与层的通道 i 相关的 的所有Feature Map 共有 M 个,则有:
后做卷积的意思是完全卷积 假设输入图像 A 大小为 a x b,卷积核 B 大小为c x d,操作会对A进行填充,在 A 左右各填充(d-1)列,上下各填充(c-1)行,即 后 A 的大小变为( a+2c-2)x(b+2d-2),用B对后的A做卷积,则卷积返回的大小为(a+b-1)x(a+b-1)。即为旋转180度。
此处公式设计到卷积操作,形式比较复杂,下面引入一个示例来讲解上述公式:
层的大小为 3 x 3,对层的通道 i (为了便于说明,假设层为单通道),通过如下两个卷积核 、 ,得到 层的两个Feature Map,假设其残差已经求得(前边已经讲过如何求卷积层的残差),残差分别为 。
分别进行操作与操作:
将与进行卷积,与进行卷积:
注意此处不考虑项(或者认为是线性激活 即 = 1),对结果求和有:
以上过程即完成了整个公式的计算,为什么这么计算呢?可以把以上过程展开,把与 展开为全连接的形式,(此处清楚起见,只写了),根据MLP中误差按权值平均反向传导的形式 可求得层的残差,其实归根到底还是MLP得形式。
求解网络参数的导数
至此,关于CNN中的输入前向传导与残差的后向反馈全部计算完成,下面可以根据来对网络的权值 进行求导。
假设当前损失函数为
1)对于 ,根据链式求导法则:
根据残差的定义与前边的计算公式:
可得到如下公式,公式中的累加是因为 conv 层中 n *n 的区域变为 1个 1*1 的 pooling 元素,但是 这 n*n 的权值取值是一样的,所以对该权值求导是 pooling 层所有节点的累加:
这里均为实数,且中的所有元素的参数均为,所以对结果向量进行求和操作。这里的代表对层Feature Map j 中的所有元素求和
2)对于 ,根据链式求导法则:
根据残差的定义与前边的计算公式:
可得:
同样通过一个实例说明关于卷积核的偏导数求解过程,对层的通道 i 用卷积核进行卷积操作,得到层的 Feature Map j,其残差已经求得,对于下图左的卷积过程展开,下图右为展开后:
注意这里层的节点① ② ④ ⑤ 用卷积核卷积生成生成层的节点①,即层的节点① ② ④ ⑤ 与层的节点①通过权值相连,分别为回忆传统的MLP网络的参数求导公式:
所以这里有对于单个层的节点①对应的权值求导有:
考虑层的节点② ③ ④对应的权值与节点①对应的权值相等(用的同一卷积核):所以可以忽略节点对应的下标:
同样分别对节点层的节点② ③ ④对应的权值求导,比如对于节点 ④:
最终将偏导数项相加有:
下图描述了以上过程,等等,仔细观察一下以上公式,这不就是用层残差Map 对层的输出做卷积吗?oh my god,的导数可以通过对层的输出与卷积层的残差矩阵卷积得到!!
、
总结以上公式综上所述:
1)任意层前向传到到卷积层与卷积层前向传导到pooling层
2)pooling层前馈到卷积层,与卷积层前馈到任意层
3)任意层与卷积层之间参数的求导
4)pooling层与卷积层之间参数的求导
一些复杂的卷积操作matlab函数('valid' 与 'full' 表示的卷积模式不同):
至此,CNN中的前后向传导的过程全部推倒完毕,其实万变不离其宗,归根到底还是MLP那一套BP算法,接下来就是实现CNN了,也就是cs231n的作业。
参考:
Notes on Convolutional Neural Networks, Jake Bouvrie