动手学pytorch-卷积神经网络基础

Posted 54hys

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了动手学pytorch-卷积神经网络基础相关的知识,希望对你有一定的参考价值。

卷积神经网络基础

1.二维卷积层
2.填充和步幅
3.多输入通道和多输出通道
4.卷积层与全连接层的对比
5.池化

1.二维卷积层

二维互相关(cross-correlation)运算的输入是一个二维输入数组和一个二维核(kernel)数组,输出也是一个二维数组,其中核数组通常称为卷积核或过滤器(filter)。卷积核的尺寸通常小于输入数组,卷积核在输入数组上滑动,在每个位置上,卷积核与该位置处的输入子数组按元素相乘并求和,得到输出数组中相应位置的元素。图1展示了一个互相关运算的例子,阴影部分分别是输入的第一个计算区域、核数组以及对应的输出。

技术图片
图1 二维互相关运算

特征图与感受野

二维卷积层输出的二维数组可以看作是输入在空间维度(宽和高)上某一级的表征,也叫特征图(feature map)。影响元素(x)的前向计算的所有可能输入区域(可能大于输入的实际尺寸)叫做(x)的感受野(receptive field)。

以图1为例,输入中阴影部分的四个元素是输出中阴影部分元素的感受野。将图中形状为(2 imes 2)的输出记为(Y),将(Y)与另一个形状为(2 imes 2)的核数组做互相关运算,输出单个元素(z)。那么,(z)(Y)上的感受野包括(Y)的全部四个元素,在输入上的感受野包括其中全部9个元素。可见,可以通过更深的卷积神经网络使特征图中单个元素的感受野变得更加广阔,从而捕捉输入上更大尺寸的特征。

2.填充和步幅

填充(padding)是指在输入高和宽的两侧填充元素(通常是0元素),图2里在原输入高和宽的两侧分别添加了值为0的元素。

技术图片

图2 在输入的高和宽两侧分别填充了0元素的二维互相关计算

如果原输入的高和宽是(n_h)(n_w),卷积核的高和宽是(k_h)(k_w),在高的两侧一共填充(p_h)行,在宽的两侧一共填充(p_w)列,则输出形状为:

[ (n_h+p_h-k_h+1) imes(n_w+p_w-k_w+1) ]

在互相关运算中,卷积核在输入数组上滑动,每次滑动的行数与列数即是步幅(stride)。此前使用的步幅都是1,图3展示了在高上步幅为3、在宽上步幅为2的二维互相关运算。

技术图片

图3 高和宽上步幅分别为3和2的二维互相关运算

一般来说,当高上步幅为(s_h),宽上步幅为(s_w)时,输出形状为:

$$

lfloor(n_h+p_h-k_h+s_h)/s_h floor imes lfloor(n_w+p_w-k_w+s_w)/s_w floor

$$

如果(p_h=k_h-1)(p_w=k_w-1),那么输出形状将简化为(lfloor(n_h+s_h-1)/s_h floor imes lfloor(n_w+s_w-1)/s_w floor)。更进一步,如果输入的高和宽能分别被高和宽上的步幅整除,那么输出形状将是((n_h / s_h) imes (n_w/s_w))

(p_h = p_w = p)时,称填充为(p);当(s_h = s_w = s)时,称步幅为(s)

在卷积神经网络中使用奇数高宽的核,比如(3 imes 3)(5 imes 5)的卷积核,对于高度(或宽度)为大小为(2 k + 1)的核,令步幅为1,在高(或宽)两侧选择大小为(k)的填充,便可保持输入与输出尺寸相同。

3.多输入通道和多输出通道

多输入通道

卷积层的输入可以包含多个通道,图4展示了一个含2个输入通道的二维互相关计算的例子。

技术图片

图4 含2个输入通道的互相关计算

假设输入数据的通道数为(c_i),卷积核形状为(k_h imes k_w),为每个输入通道各分配一个形状为(k_h imes k_w)的核数组,将(c_i)个互相关运算的二维输出按通道相加,得到一个二维数组作为输出。把(c_i)个核数组在通道维上连结,即得到一个形状为(c_i imes k_h imes k_w)的卷积核。

多输出通道

卷积层的输出也可以包含多个通道,设卷积核输入通道数和输出通道数分别为(c_i)(c_o),高和宽分别为(k_h)(k_w)。如果希望得到含多个通道的输出,可以为每个输出通道分别创建形状为(c_i imes k_h imes k_w)的核数组,将它们在输出通道维上连结,卷积核的形状即(c_o imes c_i imes k_h imes k_w)

对于输出通道的卷积核,提供这样一种理解,一个(c_i imes k_h imes k_w)的核数组可以提取某种局部特征,但是输入可能具有相当丰富的特征,需要有多个这样的(c_i imes k_h imes k_w)的核数组,不同的核数组提取的是不同的特征。

1x1卷积层

最后讨论形状为(1 imes 1)的卷积核,通常称这样的卷积运算为(1 imes 1)卷积,称包含这种卷积核的卷积层为(1 imes 1)卷积层。图5展示了使用输入通道数为3、输出通道数为2的(1 imes 1)卷积核的互相关计算。

技术图片

图5 1x1卷积核的互相关计算。输入和输出具有相同的高和宽

(1 imes 1)卷积核可在不改变高宽的情况下,调整通道数。(1 imes 1)卷积核不识别高和宽维度上相邻元素构成的模式,其主要计算发生在通道维上。假设将通道维当作特征维,将高和宽维度上的元素当成数据样本,那么(1 imes 1)卷积层的作用与全连接层等价。

4.卷积层与全连接层的对比

二维卷积层经常用于处理图像,与此前的全连接层相比,它主要有两个优势:

一是全连接层把图像展平成一个向量,在输入图像上相邻的元素可能因为展平操作不再相邻,网络难以捕捉局部信息。而卷积层的设计,天然地具有提取局部信息的能力。

二是卷积层的参数量更少。不考虑偏置的情况下,一个形状为((c_i, c_o, h, w))的卷积核的参数量是(c_i imes c_o imes h imes w),与输入图像的宽高无关。假如一个卷积层的输入和输出形状分别是((c_1, h_1, w_1))((c_2, h_2, w_2)),如果要用全连接层进行连接,参数数量就是(c_1 imes c_2 imes h_1 imes w_1 imes h_2 imes w_2)。使用卷积层可以以较少的参数数量来处理更大的图像。

5.池化

池化层主要用于缓解卷积层对位置的过度敏感性。同卷积层一样,池化层每次对输入数据的一个固定形状窗口(又称池化窗口)中的元素计算输出,池化层直接计算池化窗口内元素的最大值或者平均值,该运算也分别叫做最大池化或平均池化。图6展示了池化窗口形状为(2 imes 2)的最大池化。

技术图片

图6 池化窗口形状为 2 x 2 的最大池化

二维平均池化的工作原理与二维最大池化类似,但将最大运算符替换成平均运算符。池化窗口形状为(p imes q)的池化层称为(p imes q)池化层,其中的池化运算叫作(p imes q)池化。

池化层也可以在输入的高和宽两侧填充并调整窗口的移动步幅来改变输出形状。池化层填充和步幅与卷积层填充和步幅的工作机制一样。

在处理多通道输入数据时,池化层对每个输入通道分别池化,但不会像卷积层那样将各通道的结果按通道相加。这意味着池化层的输出通道数与输入通道数相等。

以上是关于动手学pytorch-卷积神经网络基础的主要内容,如果未能解决你的问题,请参考以下文章

经典卷积神经网络 LeNet 动手学深度学习v2 pytorch

CNN 卷积神经网络 池化层Pooling 动手学深度学习v2 pytorch

搞定《动手学深度学习》-(李牧)PyTorch版本的所有内容

动手学pytorch-循环神经网络基础

含并行连结的网络 GoogLeNet / Inception V3 动手学深度学习v2 pytorch

动手学pytorch-循环神经网络进阶