文章目录
博客: blog.shinelee.me | 博客园 | CSDN
写在前面
开篇先上图,图为deconvolution在像素级语义分割中的一种应用,直观感觉deconvolution是一个upsampling的过程,像是convolution的对称过程。
本文将深入deconvolution的细节,并通过如下方式展开:
- 先回答 什么是deconvolution?为什么会有transposed convolutionon、subpixel or fractional convolution这样的名字?
- 再介绍 各种情形下 transposed convolution是如何进行的,并提供一种统一的计算方法。
什么是deconvolution
首先要明确的是,deconvolution并不是个好名字,因为它存在歧义:
- deconvolution最初被定义为“inverse of convolution”或者“inverse filter”或者“解卷积”,是指消除先前滤波作用的方法。比如,我们认为原始图像是清晰的,但是通过透镜观测到的图像却变得模糊,如果假设透镜的作用相当于以某个kernel作用在原始图像上,由此导致图像变得模糊,那么根据模糊的图像估计这个kernel或者根据模糊图像恢复原始清晰图像的过程就叫deconvolution。
- 后来论文Adaptive Deconvolutional Networks for Mid and High Level Feature Learning和Visualizing and Understanding Convolutional Networks又重新定义了deconvolution,实际上与transposed convolution、sub-pixel or fractional convolution指代相同。transposed convolution是一个更好的名字,sub-pixel or fractional convolution可以看成是transposed convolution的一个特例。对一个常规的卷积层而言,前向传播时是convolution,将input feature map映射为output feature map,反向传播时则是transposed convolution,根据output feature map的梯度计算出input feature map的梯度,梯度图的尺寸与feature map的尺寸相同。
本文谈论的是deconvolution的第2个含义,后面统一使用transposed convolution这个名字。
什么是transposed convolution?A guide to convolution arithmetic for deep learning中有这样一段话:
看完好像仍不是很直观,transposed convolution到底对应的是什么操作?等到文章的后面,这个问题的答案会逐渐清晰起来。
下面先以1个例子来对比convolution过程和transposed convolution过程,采用与A guide to convolution arithmetic for deep learning相同的设置:
- 2-D transposed convolutions (N = 2 N=2N=2)
- square inputs (i 1 = i 2 = i i_1=i_2=ii1=i2=i)
- square kernel size (k 1 = k 2 = k k_1=k_2=kk1=k2=k)
- same strides along both axes (s 1 = s 2 = s s_1=s_2=ss1=s2=s)
- same zero padding along both axes (p 1 = p 2 = p p_1=p_2=pp1=p2=p)
- square outputs (o 1 = o 2 = o o_1=o_2=oo1=o2=o)
若令i = 4 i=4i=4、s = 1 s=1s=1、p = 0 p=0p=0、k = 3 k=3k=3,输出尺寸o = 2 o=2o=2,则convolution过程是将4 × 4 4\times 44×4的map映射为2 × 2 2\times 22×2的map,而transposed convolution过程则是将2 × 2 2\times 22×2的map映射为4 × 4 4\times 44×4的map,两者的kernel size均为3,如下图所示:
可以看到,convolution过程zero padding的数量与超参数p pp一致,但是transposed convolution实际的zero padding的数量为2,为什么会这样?是为了保持连接方式相同,下面具体看一下。
convolution过程
先看convolution过程,连接方式 如下图所示,绿色表示输出,蓝色表示输入,每个绿色块具与9个蓝色块连接。
令卷积核w = ( w 0 , 0 w 0 , 1 w 0 , 2 w 1 , 0 w 1 , 2 w 1 , 2 w 2 , 0 w 2 , 1 w 2 , 2 ) \mathbf{w} = \left(\begin{array}{ccc} {w_{0,0}} & {w_{0,1}} & {w_{0,2}} \\ {w_{1,0}} & {w_{1,2}} & {w_{1,2}} \\ {w_{2,0}} & {w_{2,1}} & {w_{2,2}} \end{array}\right)w=⎝⎛w0,0w1,0w2,0w0,1w1,2w2,1w0,2w1,2w2,2⎠⎞,为了便于理解,将卷积写成矩阵乘法形式,令x \mathbf{x}x为4 × 4 4\times 44×4输入矩阵以行优先方式拉成的长度为16的向量,y \mathbf{y}y为2 × 2 2\times 22×2输出矩阵以同样方式拉成的长度为4的向量,同时将w \mathbf{w}w表示成4 × 16 4\times 164×16的稀疏矩阵C \mathbf{C}C,
C = ( w 0 , 0 w 0 , 1 w 0 , 2 0 w 1 , 0 w 1 , 1 w 1 , 2 0 w 2 , 0 w 2 , 1 w 2 , 2 0 0 0 0 0 0 w 0 , 0 w 0 , 1 w 0 , 2 0 w 1 , 0 w 1 , 1 w 1 , 2 0 w 2 , 0 w 2 , 1 w 2 , 2 0 0 0 0 0 0 0 0 w 0 , 0 w 0 , 1 w 0 , 2 0 w 1 , 0 w 1 , 1 w 1 , 2 0 w 2 , 0 w 2 , 1 w 2 , 2 0 0 0 0 0 0 w 0 , 0 w 0 , 1 w 0 , 2 0 w 1 , 0 w 1 , 1 w 1 , 2 0 w 2 , 0 w 2 , 1 w 2 , 2 ) \mathbf{C} = \left(\begin{array}{cccccccccccccccc}{w_{0,0}} & {w_{0,1}} & {w_{0,2}} & {0} & {w_{1,0}} & {w_{1,1}} & {w_{1,2}} & {0} & {w_{2,0}} & {w_{2,1}} & {w_{2,2}} & {0} & {0} & {0} & {0} & {0} \\ {0} & {w_{0,0}} & {w_{0,1}} & {w_{0,2}} & {0} & {w_{1,0}} & {w_{1,1}} & {w_{1,2}} & {0} & {w_{2,0}} & {w_{2,1}} & {w_{2,2}} & {0} & {0} & {0} & {0} \\ {0} & {0} & {0} & {0} & {w_{0,0}} & {w_{0,1}} & {w_{0,2}} & {0} & {w_{1,0}} & {w_{1,1}} & {w_{1,2}} & {0} & {w_{2,0}} & {w_{2,1}} & {w_{2,2}} & {0} \\ {0} & {0} & {0} & {0} & {0} & {w_{0,0}} & {w_{0,1}} & {w_{0,2}} & {0} & {w_{1,0}} & {w_{1,1}} & {w_{1,2}} & {0} & {w_{2,0}} & {w_{2,1}} & {w_{2,2}}\end{array}\right)C=⎝⎜⎜⎛w0,0000w0,1w0,000w0,2w0,1000w0,200w1,00w0,00w1,1w1,0w0,1w0,0w1,2w1,1w0,2w0,10w1,20w0,2w2,00w1,00w2,1w2,0w1,1w1,0w2,2w2,1w1,2w1,10w2,20w1,200w2,0000w2,1w2,000w2,2w2,1000w2,2⎠⎟⎟⎞
则convolution过程可以描述为C x = y \mathbf{C} \mathbf{x} = \mathbf{y}Cx=y,若**C i , j = 0 \mathbf{C}_{i,j}=0Ci,j=0表示x j \mathbf{x}_jxj和y i \mathbf{y}_iyi间没有连接**。
transposed convolution过程
再看transposed convolution过程,如何将长度为4的向量y \mathbf{y}y映射为长度为16的向量且保持连接方式相同?只需将C \mathbf{C}C转置,令C T y = x ′ \mathbf{C}^T \mathbf{y} = \mathbf{x}'CTy=x′,同样地,C j , i T = 0 \mathbf{C}^T_{j,i}=0Cj,iT=0表示x j ′ \mathbf{x}'_jxj′和y i \mathbf{y}_iyi间没有连接。
此时,C T \mathbf{C}^TCT对应的卷积操作恰好相当于将kernel中心对称,FULL zero padding,然后卷积,此时,1个蓝色块与9个绿色块连接,且权重与Convolution过程相同
需要注意的是,transposed convolution的kernel与convolution的kernel可以有关,也可以无关,需要看应用在什么场景,
- 在特征可视化、训练阶段的反向传播中应用的transposed convolution,并不是作为一个真正的layer存在于网络中,其kernel与convolution共享(但要经过中心对称后再卷积,相当于上面的 $ \mathbf{C} ^T $)。
- 在图像分割、生成模型、decoder中使用的transposed convolution,是网络中真实的layer,其kernel经初始化后需要通过学习获得(所以卷积核也就无所谓中心对称不对称了)。
- 前向传播为convolution/transposed convolution,则反向传播为transposed convolution/convolution。
在上面举的简化的例子中,我们可以通过分析得知transposed convolution该如何进行,但是,对于更一般情况应该怎么做?
transposed convolution的计算
对于一般情况,只需把握一个宗旨:transposed convolution将output size恢复为input size且保持连接方式相同。
对于convolution过程,我们知道其output map与input map的尺寸关系如下:
o = ⌊ i + 2 p − k s ⌋ + 1 o=\left\lfloor \frac{i+2p-k}{s} \right\rfloor + 1o=⌊si+2p−k⌋+1
若要将o oo恢复为i ii,需考虑2种情况,i + 2 p − k s \frac{i+2p-k}{s}si+2p−k整除以及不整除,先看整除的情况。
整除的情况
如果i + 2 p − k s \frac{i+2p-k}{s}si+2p−k可以整除,则由上式可得
i = s o − s + k − 2 p = [ o + ( s − 1 ) ( o − 1 ) ] + ( k − 2 p − 1 ) i = so-s+k-2p = [o+(s-1)(o-1)]+(k-2p-1)i=so−s+k−2p=[o+(s−1)(o−1)]+(k−2p−1)
因为transposed convolution也是卷积,为了符合上面卷积操作尺寸关系的数学形式,可进一步整理成
i = [ o + ( s − 1 ) ( o − 1 ) ] + [ ( k − 1 ) + ( k − 2 p − 1 ) ] − k 1 + 1 i = \frac{[o+(s-1)(o-1)] + [(k-1)+(k-2p-1)] - k}{1} + 1i=1[o+(s−1)(o−1)]+[(k−1)+(k−2p−1)]−k+1
令i ′ = o + ( s − 1 ) ( o − 1 ) i'=o+(s-1)(o-1)i′=o+(s−1)(o−1)、$p’=\frac{(k-1)+(k-2p-1)}{2} = k-p-1 、 、、s’=1、 、、k’=k$,即transposed convolution实际卷积时使用的超参数,可以这样理解:
i ′ = o + ( s − 1 ) ( o − 1 ) i'=o+(s-1)(o-1)i′=o+(s−1)(o−1):convolution的输出为o × o o\times oo×o,每行每列都是o oo个元素,有o − 1 o-1o−1个间隔,transposed convolution时在每个间隔处插入s − 1 s-1s−1个0,整体构成transposed convolution的input map;
p ′ = ( k − 1 ) + ( k − 2 p − 1 ) 2 = k − p − 1 p'=\frac{(k-1)+(k-2p-1)}{2} = k-p-1p′=2(k−1)+(k−2p−1)=k−p−1:在上一步input map的基础上再进行padding,考虑convolution常用的几种padding情况:
- VALID:p = 0 p=0p=0,transposed convolution则需padding p ′ = k − 1 p'=k-1p′=k−1,即FULL padding
- SAME:p = k − 1 2 = r p=\frac{k-1}{2}=rp=2k−1=r,这里考虑k = 2 r + 1 k=2r+1k=2r+1为奇数的一般情况,此时p ′ = r p'=rp′=r,即SAME padding
- FULL:p = k − 1 p=k-1p=k−1,则p ′ = 0 p'=0p′=0,即VALID padding
可见,convolution和transposed convolution的padding也具有某种对称性p ′ + p = k − 1 p'+p=k-1p′+p=k−1;
k ′ = k k'=kk′=k:transposed convolution的kernel size与convolution相同;
s ′ = 1 s'=1s′=1:transposed convolution的stride均为1,但也可以换个角度理解,如果认为o × o o\times oo×o相邻元素间的距离为1个像素,那么在间隔处插入s − 1 s-1s−1个0后(s > 1 s > 1s>1),得到的input map相邻元素间的距离就是亚像素的(sub-pixel),所以此时也可以称之为 sub-pixel or fractional convolution;
o ′ = i = i ′ + 2 p ′ − k ′ s ′ + 1 o'=i=\frac{i'+2p'-k'}{s'}+1o′=i=s′i′+2p′−k′+1:transposed convolution的输出与convolution的输入具有相同尺寸。
不整除的情况
接下来再看i + 2 p − k s \frac{i+2p-k}{s}si+2p−k不整除的情况,此时再按上面的方式计算得到的o ′ = i ′ + 2 p ′ − k ′ s ′ + 1 o'=\frac{i'+2p'-k'}{s'}+1o′=s′i′+2p′−k′+1将小于i ii,小多少呢?不难得出少a = [ ( i + 2 p − k ) m o d s ] a = [(i+2p-k) \mod s]a=[(i+2p−k)mods],即
o ′ = i ′ + 2 p ′ − k ′ s ′ + 1 = i − a o'=\frac{i'+2p'-k'}{s'}+1=i-ao′=s′i′+2p′−k′+1=i−a
为了让o ′ = i o'=io′=i,可写成
o ′ = i ′ + 2 p ′ + a − k ′ s ′ + 1 o'= \frac{i'+2p'+a-k'}{s'}+1o′=s′i′+2p′+a−k′+1
只需在padding后,在下边和右边再扩展a aa行和列0,然后进行卷积即可。注意,因为s ′ = 1 s'=1s′=1,我们可以将a aa放在分母也可以放在外面,之所以放在分母,是因为convolution过程中input map下边和右边的a aa行或列中的元素可能参与了运算,即与output map间存在连接,所以在transposed convolution时,为了保持同样的连接,最后扩展的a aa行和列也要参与卷积,所以放在分母。
至此,再看transposed convolution的各种情况,就很容易推算了,更多例子可参见A guide to convolution arithmetic for deep learning。
总结
最后,总结一下,
- convolution和transposed convolution互为对称过程,存在一个convolution,就存在一个与之对应的transposed convolution,反之亦然;
- convolution是将input size的map映射为output size的map,transposed convolution是将output size的map映射为input size的map——旨在将尺寸恢复;
- 两者均使用卷积操作,为了方便,两者使用同样的stride、padding、kernel size超参数,但实际执行时的操作不同,一般情况下,transposed convolution与convolution实际超参数关系为:i ′ = o + ( s − 1 ) ( o − 1 ) i'=o+(s-1)(o-1)i′=o+(s−1)(o−1)、$p’=\frac{(k-1)+(k-2p-1)}{2} = k-p-1 、 、、s’=1、 、、k’=k$。
- 之所以做这样的操作,是为了保证map间的连接方式相同(权重不一定相同),权重的设置需根据应用的场景,可能通过学习得到,也可能与convolution共享(但需要中心对称后再使用)。