从全连接层到卷积

学过MLP的小伙伴们都知道,假设我们对一个MLP模型进行图像分类任务时,我们对输入图像的第一步操作就是将一个二维的图像展平为一维作为输出。但这样的处理方式有一个显而易见的缺点——忽略了每张图像的空间结构信息。显然这是不准确的,它会导致我们对于图像的识别率降低。除此之外,使用MLP模型进行图像分类任务还有一个致命的缺点——所需参数过大。我们知道现代的图片一般都有个几百万的像素,这里我们做一个假设,假设我们手里有一张像素为1200万像素的图片,如果我们对其进行全连接的话,我们使用100个隐藏单元大小的隐藏层,则其所需参数量为1亿2千万,这甚至是不考虑3原色通道的情况下,已经相当于一个小语言模型的参数规模了,这显然是低效高耗的。那有没有什么办法能够既能读取图片的空间特征又可以减少参数量呢?答案当然是有,那就是我们接下来将要讨论的卷积神经网络(convolutional neural network,CNN)。


为了解决上述的问题,卷积神经网络利用了提取信息的两大原则:平移不变性和局部性

平移不变性

首先,多层感知机的输入是二维图像X,其隐藏表示H在数学上是一个矩阵,在代码中表示为二维张量。 其中X和H具有相同的形状。 为了方便理解,我们可以认为,无论是输入还是隐藏表示都拥有空间结构。

使用\([X]_{i, j}\)\([H]_{i, j}\)分别表示输入图像和隐藏表示中位置\((i, j)\)处的像素。 为了使每个隐藏神经元都能接收到每个输入像素的信息,我们将参数从权重矩阵(如同我们先前在多层感知机中所做的那样)替换为四阶权重张量\(W\)。假设\(U\)包含偏置参数,我们可以将全连接层形式化地表示为

\[\begin{split}\begin{aligned} \left[\mathbf{H}\right]_{i, j} &= [\mathbf{U}]_{i, j} + \sum_k \sum_l[\mathsf{W}]_{i, j, k, l} [\mathbf{X}]_{k, l}\\ &= [\mathbf{U}]_{i, j} + \sum_a \sum_b [\mathsf{V}]_{i, j, a, b} [\mathbf{X}]_{i+a, j+b}.\end{aligned}\end{split} \]

其中,从\(W\)\(V\)的转换只是形式上的转换,因为在这两个四阶张量的元素之间存在一一对应的关系。 我们只需重新索引下标\((k, l)\),使\(k=i+a, l=j+b\),由此可得\([V]_{i, j, a, b}=[W]_{i, j, i+a, j+b}\)。 索引a和b通过在正偏移和负偏移之间移动覆盖了整个图像。简单来说,就是不断地扫描图形中的像素点。

现在引用上述的第一个原则:平移不变性。这意味这检测对象再输入X上的平移并不会导致隐藏表示H中的平移,也就是不会影响权重,即\([V]_{i, j, a, b}=[V]_{a, b}\)。因此,我们可以简化\(H\)的定义为:

\[[\mathbf{H}]_{i, j} = u + \sum_a\sum_b [\mathbf{V}]_{a, b} [\mathbf{X}]_{i+a, j+b}. \]

这就是卷积(convolution),我们使用系数\([V]_{a, b}\)对位置\((i, j)\)附近的像素\((i+a, j+b)\)进行加权得到\([H]_{i, j}\)。注意,\([V]_{a, b}\)的系数\([V]_{i, j, a, b}\)少很多,因为前者不在依赖于图像中的位置,能大大减少参数量。

局部性

局部性简单来说,就是我们在识别图像时,并不关注整张图像,而是像侦探一样,通过一个放大镜来扫描图片的某些特征或者细节,只关注图形的一部分区域,因此,我们可以重写\(H\)的定义为:

\[[\mathbf{H}]_{i, j} = u + \sum_{a = -\Delta}^{\Delta} \sum_{b = -\Delta}^{\Delta} [\mathbf{V}]_{a, b} [\mathbf{X}]_{i+a, j+b}. \]

这就是一个卷积层,在深度学习社区\(V\)称为卷积核(convolution kernel)滤波器(filter),或者简单地称为该卷积核的权重,而这里的\(\Delta\)就是卷积核的范围。

总结

卷积的引入大大减少了模型的参数量,而且卷积层中的神经元仅与输入数据的局部区域连接,这反映了图像的空间局部性,使得网络能够有效提取局部特征,除此之外,较少的参数数量有助于减少过拟合的风险,使得模型在未见过的数据上具有更好的泛化能力。

posted @ 2024-08-01 23:02  codersgl  阅读(1)  评论(0编辑  收藏  举报