查看原文
其他

手把手教你如何用 TensorFlow 实现 CNN

2017-05-08 宋传标 AI研习社

AI 研习社按:本文作者宋传标,原文载于作者,AI 研习社已获授权。

  CNN 的引入

在人工的全连接神经网络中,每相邻两层之间的每个神经元之间都是有边相连的。当输入层的特征维度变得很高时,这时全连接网络需要训练的参数就会增大很多,计算速度就会变得很慢,例如一张黑白的 28×28 的手写数字图片,输入层的神经元就有 784 个,如下图所示: 

若在中间只使用一层隐藏层,参数 w 就有 784×15=11760 多个;若输入的是 28×28 带有颜色的 RGB 格式的手写数字图片,输入神经元就有 28×28×3=2352 个…… 。这很容易看出使用全连接神经网络处理图像中的需要训练参数过多的问题。

而在卷积神经网络(Convolutional Neural Network,CNN)中,卷积层的神经元只与前一层的部分神经元节点相连,即它的神经元间的连接是非全连接的,且同一层中某些神经元之间的连接的权重 w 和偏移 b 是共享的(即相同的),这样大量地减少了需要训练参数的数量。

卷积神经网络 CNN 的结构一般包含这几个层:

● 输入层:用于数据的输入

● 卷积层:使用卷积核进行特征提取和特征映射

● 激励层:由于卷积也是一种线性运算,因此需要增加非线性映射

● 池化层:进行下采样,对特征图稀疏处理,减少数据运算量。

● 全连接层:通常在 CNN 的尾部进行重新拟合,减少特征信息的损失

● 输出层:用于输出结果

当然中间还可以使用一些其他的功能层:

● 归一化层(Batch Normalization):在 CNN 中对特征的归一化

● 切分层:对某些(图片)数据的进行分区域的单独学习

● 融合层:对独立进行特征学习的分支进行融合

  CNN 的层次结构

输入层:

在 CNN 的输入层中,(图片)数据输入的格式 与 全连接神经网络的输入格式(一维向量)不太一样。CNN 的输入层的输入格式保留了图片本身的结构。

对于黑白的 28×28 的图片,CNN 的输入是一个 28×28 的的二维神经元,如下图所示:

而对于 RGB 格式的 28×28 图片,CNN 的输入则是一个 3×28×28 的三维神经元(RGB 中的每一个颜色通道都有一个 28×28 的矩阵),如下图所示:

卷积层:

在卷积层中有几个重要的概念:

local receptive fields(感受视野)

shared weights(共享权值)

假设输入的是一个 28×28 的的二维神经元,我们定义 5×5 的 一个 local receptive fields(感受视野),即 隐藏层的神经元与输入层的 5×5 个神经元相连,这个 5*5 的区域就称之为 Local Receptive Fields,如下图所示:

可类似看作:隐藏层中的神经元 具有一个固定大小的感受视野去感受上一层的部分特征。在全连接神经网络中,隐藏层中的神经元的感受视野足够大乃至可以看到上一层的所有特征。

而在卷积神经网络中,隐藏层中的神经元的感受视野比较小,只能看到上一次的部分特征,上一层的其他特征可以通过平移感受视野来得到同一层的其他神经元,由同一层其他神经元来看:

设移动的步长为 1:从左到右扫描,每次移动 1 格,扫描完之后,再向下移动一格,再次从左到右扫描。

具体过程如动图所示:

可看出 卷积层的神经元是只与前一层的部分神经元节点相连,每一条相连的线对应一个权重 w 。

一个感受视野带有一个卷积核,我们将 感受视野 中的权重 w 矩阵称为 卷积核 ;将感受视野对输入的扫描间隔称为步长(stride);当步长比较大时(stride>1),为了扫描到边缘的一些特征,感受视野可能会 “出界”,这时需要对边界扩充 (pad),边界扩充可以设为 0 或 其他值。步长 和 边界扩充值的大小由用户来定义。

卷积核的大小由用户来定义,即定义的感受视野的大小;卷积核的权重矩阵的值,便是卷积神经网络的参数,为了有一个偏移项 ,卷积核可附带一个偏移项 b ,它们的初值可以随机来生成,可通过训练进行变化。

因此 感受视野 扫描时可以计算出下一层神经元的值为:

对下一层的所有神经元来说,它们从不同的位置去探测了上一层神经元的特征。

我们将通过 一个带有卷积核的感受视野 扫描生成的下一层神经元矩阵 称为 一个 feature map (特征映射图),如下图的右边便是一个 feature map:

因此在同一个 feature map 上的神经元使用的卷积核是相同的,因此这些神经元 shared weights,共享卷积核中的权值和附带的偏移。一个 feature map 对应 一个卷积核,若我们使用 3 个不同的卷积核,可以输出 3 个 feature map:(感受视野:5×5,布长 stride:1)

因此在 CNN 的卷积层,我们需要训练的参数大大地减少到了 (5×5+1)×3=78 个。

假设输入的是 28×28 的 RGB 图片,即输入的是一个 3×28×28 的的二维神经元,这时卷积核的大小不只用长和宽来表示,还有深度,感受视野也对应的有了深度,如下图所示:

由图可知:感受视野: 3×2×2 ; 卷积核: 3×2×2 ,深度为 3;下一层的神经元的值为:b+∑2d=0∑1i=0∑1j=0wdijxdij . 卷积核的深度和感受视野的深度相同,都由输入数据来决定,长宽可由自己来设定,数目也可以由自己来设定,一个卷积核依然对应一个 feature map 。

注:“stride=1” 表示在长和宽上的移动间隔都为 1,即 stridewidth=1 且 strideheight=1

激励层:

激励层主要对卷积层的输出进行一个非线性映射,因为卷积层的计算还是一种线性计算。使用的激励函数一般为 ReLu 函数:

f(x)=max(x,0)

卷积层和激励层通常合并在一起称为 “卷积层”。

池化层:

当输入经过卷积层时,若感受视野比较小,布长 stride 比较小,得到的 feature map (特征图)还是比较大,可以通过池化层来对每一个 feature map 进行降维操作,输出的深度还是不变的,依然为 feature map 的个数。

池化层也有一个 “池化视野(filter)” 来对 feature map 矩阵进行扫描,对 “池化视野” 中的矩阵值进行计算,一般有两种计算方式:

Max pooling:取 “池化视野” 矩阵中的最大值

Average pooling:取 “池化视野” 矩阵中的平均值

扫描的过程中同样地会涉及的扫描布长 stride,扫描方式同卷积层一样,先从左到右扫描,结束则向下移动布长大小,再从左到右。如下图示例所示:

其中 “池化视野”filter: 2×2;布长 stride:2。(注:“ 池化视野” 为个人叫法)

最后可将 3 个 24×24 的 feature map 下采样得到 3 个 24×24 的特征矩阵:

  归一化层

1. Batch Normalization

Batch Normalization(批量归一化)实现了在神经网络层的中间进行预处理的操作,即在上一层的输入归一化处理后再进入网络的下一层,这样可有效地防止 “梯度弥散”,加速网络训练。

Batch Normalization 具体的算法如下图所示: 

每次训练时,取 batch_size 大小的样本进行训练,在 BN 层中,将一个神经元看作一个特征,batch_size 个样本在某个特征维度会有 batch_size 个值,然后在每个神经元 xi 维度上的进行这些样本的均值和方差,通过公式得到 xi∧,再通过参数 γ 和 β 进行线性映射得到每个神经元对应的输出 yi 。在 BN 层中,可以看出每一个神经元维度上,都会有一个参数 γ 和 β ,它们同权重 w 一样可以通过训练进行优化。

在卷积神经网络中进行批量归一化时,一般对 未进行 ReLu 激活的 feature map 进行批量归一化,输出后再作为激励层的输入,可达到调整激励函数偏导的作用。

一种做法是将 feature map 中的神经元作为特征维度,参数 γ 和 β 的数量和则等于 2×fmapwidth×fmaplength×fmapnum,这样做的话参数的数量会变得很多;

另一种做法是把 一个 feature map 看做一个特征维度,一个 feature map 上的神经元共享这个 feature map 的 参数 γ 和 β ,参数 γ 和 β 的数量和则等于 2×fmapnum,计算均值和方差则在 batch_size 个训练样本在每一个 feature map 维度上的均值和方差。

注:fmapnum 指的是一个样本的 feature map 数量,feature map 跟神经元一样也有一定的排列顺序。

Batch Normalization 算法的训练过程和测试过程的区别:

在训练过程中,我们每次都会将 batch_size 数目大小的训练样本 放入到 CNN 网络中进行训练,在 BN 层中自然可以得到计算输出所需要的 均值 和 方差 ;

而在测试过程中,我们往往只会向 CNN 网络中输入一个测试样本,这是在 BN 层计算的均值和方差会均为 0,因为只有一个样本输入,因此 BN 层的输入也会出现很大的问题,从而导致 CNN 网络输出的错误。所以在测试过程中,我们需要借助训练集中所有样本在 BN 层归一化时每个维度上的均值和方差,当然为了计算方便,我们可以在 batch_num 次训练过程中,将每一次在 BN 层归一化时每个维度上的均值和方差进行相加,最后再进行求一次均值即可。

2. Local Response Normalization

近邻归一化 (Local Response Normalization) 的归一化方法主要发生在不同的相邻的卷积核(经过 ReLu 之后)的输出之间,即输入是发生在不同的经过 ReLu 之后的 feature map 中。

LRN 的公式如下:

其中: 

a(i,x,y) 表示第 i 个卷积核的输出(经过 ReLu 层)的 feature map 上的 (x,y) 位置上的值。 

b(i,x,y) 表示 a(i,x,y) 经 LRN 后的输出。 

N 表示卷积核的数量,即输入的 feature map 的个数。 

n 表示近邻的卷积核(或 feature map)个数,由自己来决定。 

k,α,β是超参数,由用户自己调整或决定。

与 BN 的区别:BN 依据 mini batch 的数据, 近邻归一仅需要自己来决定,BN 训练中有学习参数; BN 归一化主要发生在不同的样本之间,LRN 归一化主要发生在不同的卷积核的输出之间。

切分层:

在一些应用中, 需要对图片进行切割,独立地对某一部分区域进行单独学习。这样可以对特定部分进行通过调整 感受视野 进行力度更大的学习。

融合层:

融合层可以对切分层进行融合,也可以对不同大小的卷积核学习到的特征进行融合。

例如在 GoogleLeNet 中,使用多种分辨率的卷积核对目标特征进行学习,通过 padding 使得每一个 feature map 的长宽都一致,之后再将多个 feature map 在深度上拼接在一起: 

融合的方法有几种,一种是特征矩阵之间的拼接级联,另一种是在特征矩阵进行运算 (+,-,x,max,conv)。

全连接层和输出层

全连接层主要对特征进行重新拟合,减少特征信息的丢失;输出层主要准备做好最后目标结果的输出。例如 VGG 的结构图,如下图所示: 

  典型的卷积神经网络

LeNet-5 模型

第一个成功应用于数字数字识别的卷积神经网络模型(卷积层自带激励函数,下同):

卷积层的卷积核边长都是 5,步长都为 1;池化层的窗口边长都为 2,步长都为 2。

AlexNet 模型

具体结构图:

从 AlexNet 的结构可发现:经典的卷积神经网络结构通常为:

输入层 → (卷积层 +→池化层?)+→全连接层 +→输出层

AlexNet 卷积层的卷积核边长为 5 或 3,池化层的窗口边长为 3。具体参数如图所示:

VGGNet 模型

VGGNet 模型 和 AlexNet 模型 在结构上没多大变化,在卷积层部位增加了多个卷积层。AlexNet(上) 和 VGGNet (下)的对比如下图所示:

具体参数如图所示:其中 CONV3-64:表示卷积核的长和宽为 3,个数有 64 个;POOL2:表示池化窗口的长和宽都为 2,其他类似。 

GoogleNet 模型

使用了多个不同分辨率的卷积核,最后再对它们得到的 feature map 按深度融合在一起,结构如图:

其中,有一些主要的模块称为 Inception module,例如: 

在 Inception module 中使用到了很多 1×1 的卷积核,使用 1×1 的卷积核,步长为 1 时,输入的 feature map 和输出的 feature map 长宽不会发生改变,但可以通过改变 1×1 的卷积核的数目,来达到减小 feature map 的厚度的效果,从而做到一些训练参数的减少。

GoogleNet 还有一个特点就是它是全卷积结构(FCN)的,网络的最后没有使用全连接层,一方面这样可以减少参数的数目,不容易过拟合,一方面也带来了一些空间信息的丢失。代替全连接层的是全局平均池化(Global Average Pooling,GAP)的方法,思想是:为每一个类别输出一个 feature map ,再取每一个 feature map 上的平均值,作为最后的 softmax 层的输入。

ResNet 模型

在前面的 CNN 模型中,都是将输入一层一层地传递下去(图左),当层次比较深时,模型不是很好训练。在 ResNet 模型中,它将低层学习到的特征和高层的学习到的特征进行一个融合(加法运算, 图右),这样反向传递时,导数传递得更快,减少梯度弥散的现象。

注意:F(X)的 shape 需要等于 X 的 shape ,这样才可以进行相加。

  Tensorflow 代码

主要的函数说明:

卷积层: 

tf.nn.conv2d(input, filter, strides, padding, use_cudnn_on_gpu=None, data_format=None, name=None)

参数说明:

● data_format:表示输入的格式,有两种分别为:“NHWC” 和 “NCHW”,默认为 “NHWC”

● input:输入是一个 4 维格式的(图像)数据,数据的 shape 由 data_format 决定:当 data_format 为 “NHWC” 输入数据的 shape 表示为 [batch, in_height, in_width, in_channels],分别表示训练时一个 batch 的图片数量、图片高度、 图片宽度、 图像通道数。当 data_format 为 “NHWC” 输入数据的 shape 表示为 [batch, in_channels, in_height, in_width]

● filter:卷积核是一个 4 维格式的数据:shape 表示为:[height,width,in_channels, out_channels],分别表示卷积核的高、宽、深度(与输入的 in_channels 应相同)、输出 feature map 的个数(即卷积核的个数)。

● strides:表示步长:一个长度为 4 的一维列表,每个元素跟 data_format 互相对应,表示在 data_format 每一维上的移动步长。当输入的默认格式为:“NHWC”,则 strides = [batch , in_height , in_width, in_channels]。其中 batch 和 in_channels 要求一定为 1,即只能在一个样本的一个通道上的特征图上进行移动,in_height , in_width 表示卷积核在特征图的高度和宽度上移动的布长,即 strideheight 和 stridewidth 。

● padding:表示填充方式:“SAME” 表示采用填充的方式,简单地理解为以 0 填充边缘,但还有一个要求,左边(上边)补 0 的个数和右边(下边)补 0 的个数一样或少一个,“VALID” 表示采用不填充的方式,多余地进行丢弃。具体公式:

“SAME”: output_spatial_shape[i]=(input_spatial_shape[i] / strides[i])

“VALID”: output_spatial_shape[i]=((input_spatial_shape[i]-(spatial_filter_shape[i]-1)/strides[i])

池化层: 

tf.nn.max_pool(value, ksize,strides,padding,data_format=’NHWC’,name=None) 

或者 

tf.nn.avg_pool(…)

参数说明:

● value:表示池化的输入:一个 4 维格式的数据,数据的 shape 由 data_format 决定,默认情况下 shape 为 [batch, height, width, channels]

● 其他参数与 tf.nn.cov2d 类型

● ksize:表示池化窗口的大小:一个长度为 4 的一维列表,一般为 [1, height, width, 1],因不想在 batch 和 channels 上做池化,则将其值设为 1。

Batch Nomalization 层: 

batch_normalization(x,mean,variance,offset,scale, variance_epsilon,name=None)

● mean 和 variance 通过 tf.nn.moments 来进行计算: 

batch_mean, batch_var = tf.nn.moments(x, axes = [0, 1, 2], keep_dims=True),注意 axes 的输入。对于以 feature map 为维度的全局归一化,若 feature map 的 shape 为 [batch, height, width, depth],则将 axes 赋值为 [0, 1, 2]

● x 为输入的 feature map 四维数据,offset、scale 为一维 Tensor 数据,shape 等于 feature map 的深度 depth。

代码示例:

通过搭建卷积神经网络来实现 sklearn 库中的手写数字识别,搭建的卷积神经网络结构如下图所示: 

import tensorflow as tf

from sklearn.datasets import load_digits

import numpy as np

digits = load_digits()

X_data = digits.data.astype(np.float32)

Y_data = digits.target.astype(np.float32).reshape(-1,1)

print X_data.shape

print Y_data.shape

(1797, 64)

(1797, 1)

from sklearn.preprocessing import MinMaxScaler

scaler = MinMaxScaler()

X_data = scaler.fit_transform(X_data)

from sklearn.preprocessing import OneHotEncoder

Y = OneHotEncoder().fit_transform(Y_data).todense() #one-hot 编码

Y

matrix([[1.,  0.,  0., ...,  0.,  0.,  0.],

        [0.,  1.,  0., ...,  0.,  0.,  0.],

        [0.,  0.,  1., ...,  0.,  0.,  0.],

        ..., 

        [0.,  0.,  0., ...,  0.,  1.,  0.],

        [0.,  0.,  0., ...,  0.,  0.,  1.],

        [0.,  0.,  0., ...,  0.,  1.,  0.]])

# 转换为图片的格式 (batch,height,width,channels)

X = X_data.reshape(-1,8,8,1)

batch_size = 8 # 使用 MBGD 算法,设定 batch_size 为 8

def generatebatch(X,Y,n_examples, batch_size):

    for batch_i in range(n_examples // batch_size):

        start = batch_i*batch_size

        end = start + batch_size

        batch_xs = X[start:end]

        batch_ys = Y[start:end]

        yield batch_xs, batch_ys # 生成每一个 batch

tf.reset_default_graph()

# 输入层

tf_X = tf.placeholder(tf.float32,[None,8,8,1])

tf_Y = tf.placeholder(tf.float32,[None,10])

# 卷积层 + 激活层

conv_filter_w1 = tf.Variable(tf.random_normal([3, 3, 1, 10]))

conv_filter_b1 =  tf.Variable(tf.random_normal([10]))

relu_feature_maps1 = tf.nn.relu(\

                tf.nn.conv2d(tf_X, conv_filter_w1,strides=[1, 1, 1, 1], padding='SAME') + conv_filter_b1)

# 池化层

max_pool1 = tf.nn.max_pool(relu_feature_maps1,ksize=[1,3,3,1],strides=[1,2,2,1],padding='SAME')

print max_pool1

Tensor("MaxPool:0", shape=(?, 4, 4, 10), dtype=float32)

# 卷积层

conv_filter_w2 = tf.Variable(tf.random_normal([3, 3, 10, 5]))

conv_filter_b2 =  tf.Variable(tf.random_normal([5]))

conv_out2 = tf.nn.conv2d(relu_feature_maps1, conv_filter_w2,strides=[1, 2, 2, 1], padding='SAME') + conv_filter_b2

print conv_out2

Tensor("add_4:0", shape=(?, 4, 4, 5), dtype=float32)

# BN 归一化层 + 激活层 

batch_mean, batch_var = tf.nn.moments(conv_out2, [0, 1, 2], keep_dims=True)

shift = tf.Variable(tf.zeros([5]))

scale = tf.Variable(tf.ones([5]))

epsilon = 1e-3

BN_out = tf.nn.batch_normalization(conv_out2, batch_mean, batch_var, shift, scale, epsilon)

print BN_out

relu_BN_maps2 = tf.nn.relu(BN_out)

Tensor("batchnorm/add_1:0", shape=(?, 4, 4, 5), dtype=float32)

# 池化层

max_pool2 = tf.nn.max_pool(relu_BN_maps2,ksize=[1,3,3,1],strides=[1,2,2,1],padding='SAME')

print max_pool2

Tensor("MaxPool_1:0", shape=(?, 2, 2, 5), dtype=float32)

# 将特征图进行展开

max_pool2_flat = tf.reshape(max_pool2, [-1, 2*2*5])

# 全连接层

fc_w1 = tf.Variable(tf.random_normal([2*2*5,50]))

fc_b1 =  tf.Variable(tf.random_normal([50]))

fc_out1 = tf.nn.relu(tf.matmul(max_pool2_flat, fc_w1) + fc_b1)

# 输出层

out_w1 = tf.Variable(tf.random_normal([50,10]))

out_b1 = tf.Variable(tf.random_normal([10]))

pred = tf.nn.softmax(tf.matmul(fc_out1,out_w1)+out_b1)

loss = -tf.reduce_mean(tf_Y*tf.log(tf.clip_by_value(pred,1e-11,1.0)))

train_step = tf.train.AdamOptimizer(1e-3).minimize(loss)

y_pred = tf.arg_max(pred,1)

bool_pred = tf.equal(tf.arg_max(tf_Y,1),y_pred)

accuracy = tf.reduce_mean(tf.cast(bool_pred,tf.float32)) # 准确率

with tf.Session() as sess:

    sess.run(tf.global_variables_initializer())

    for epoch in range(1000): # 迭代 1000 个周期

        for batch_xs,batch_ys in generatebatch(X,Y,Y.shape[0],batch_size): # 每个周期进行 MBGD 算法

            sess.run(train_step,feed_dict={tf_X:batch_xs,tf_Y:batch_ys})

        if(epoch%100==0):

            res = sess.run(accuracy,feed_dict={tf_X:X,tf_Y:Y})

            print (epoch,res)

    res_ypred = y_pred.eval(feed_dict={tf_X:X,tf_Y:Y}).flatten() # 只能预测一批样本,不能预测一个样本

    print res_ypred

(0, 0.36338341)

(100, 0.96828049)

(200, 0.99666113)

(300, 0.99554813)

(400, 0.99888706)

(500, 0.99777406)

(600, 0.9961046)

(700, 0.99666113)

(800, 0.99499166)

(900, 0.99888706)

[0 1 2 ..., 8 9 8]

在第 100 次个 batch size 迭代时,准确率就快速接近收敛了,这得归功于 Batch Normalization 的作用!需要注意的是,这个模型还不能用来预测单个样本,因为在进行 BN 层计算时,单个样本的均值和方差都为 0,会得到相反的预测效果,解决方法详见归一化层。

from sklearn.metrics import  accuracy_score

print accuracy_score(Y_data,res_ypred.reshape(-1,1))

0.998887033945

延伸阅读:从原理到实战 英伟达教你用PyTorch搭建RNN(下)


 研习社特供福利  ID:OKweiwu

关注 AI 研习社后,回复【1】获取

【千G神经网络/ AI /大数据、教程、论文!】
百度云盘地址!



   “TensorFlow & 神经网络算法高级应用班” 要开课啦!

从初级到高级,理论 + 实战,一站式深度了解 TensorFlow!

本课程面向深度学习开发者,讲授如何利用 TensorFlow 解决图像识别、文本分析等具体问题。课程跨度为 10 周,将从 TensorFlow 的原理与基础实战技巧开始,一步步教授学员如何在 TensorFlow 上搭建 CNN、自编码、RNN、GAN 等模型,并最终掌握一整套基于 TensorFlow 做深度学习开发的专业技能。

两名授课老师佟达、白发川身为 ThoughtWorks 的资深技术专家,具有丰富的大数据平台搭建、深度学习系统开发项目经验。

时间:每周二、四晚 20:00-21:00

开课时长:总学时 20 小时,分 10 周完成,每周 2 次,每次 1 小时

线上授课地址:http://www.mooc.ai/(点击文末阅读原文

您可能也对以下帖子感兴趣

文章有问题?点此查看未经处理的缓存