一步步提高手写数字的识别率(3)

在《

一步步构建卷积模型

》这篇文章中我们介绍了如何从头构建卷积模型,如果我们使用TensorFlow框架,我们就不需要这么麻烦,只需调用其中的卷积函数即可。不过在开始编程之前,还是简单介绍一下卷积神经网络。

卷积神经网络简介

卷积神经网络(Convolutional Neural Network, CNN)最初是为解决图像识别等问题设计的,到现在不仅限于图像和视频,也可用于时间序列信号,比如音频信号、文本数据等。CNN作为一个深度学习框架被提出的最初诉求,是降低对图像数据预处理的要求,以及避免复杂的特征工程。CNN可以直接使用图像的原始像素作为输入,减轻了使用传统算法必需做的大量重复、繁琐的数据预处理工作。CNN的最大特点在于卷积的权值共享结构,可以大幅减少神经网络的参数量,防止过拟合的同时又降低了神经网络的复杂度。

在卷积神经网络中,第一个卷积层会直接接受图像像素级的输入,每个卷积操作只处理一小块图像,进行卷积变化后再传到后面的网络,每一层卷积(也称作滤波器,filter)都会提取数据中最有效的特征。

一般卷积神经网络由多个卷积层构成,每个卷积层中通常会进行如下几个操作:

图像通过多个不同的卷积核的滤波,并加偏置(bias),提取局部特征,每个卷积核会映射出一个新的2D图像。

将卷积核的输出结果,进行非线性的激活函数(ReLU最常用)处理。

对激活函数的结果再进行池化操作,即降采样,一般采用最大池化方法,保留最显著的特征,并提升模型的畸变容错能力。

这几个步骤就构成了最常见的卷积层,我们还可以将多个卷积层拼在一起,构成卷积神经网络,比如下图就表示包含两个卷积层的卷积神经网络:

TensorFlow实现简单的卷积神经网络

卷积神经网络可以有很多层,比如大名鼎鼎的LeNet5,有一百多个卷积层,具有相当的复杂度。对于手写数字识别问题,我们当然不需要像LeNet5这样复杂的卷积神经网络,这里我们使用两个卷积层加一个全连接层构建一个简单但非常有代表性的卷积神经网络。

和前两个的TensorFlow程序一样,我们先加载MNIST数据集:

接下来定义初始化函数初始化权重和偏置,我们需要给权重制造一些随机的噪声来打破对称,而给偏置增加一些小的正值(0.1)来避免死亡节点(dead neurons)。

关于strides、padding等卷积运算的概念,请参考《一步步构建卷积模型》。

在正式设计卷积神经网络的结构之前,先定义输入的placeholder,x是特征,y_是真实的标签。因为卷积神经网络会利用到空间结构信息,因此需要将一维的输入向量x转化为2D的图片结构,即从1x784转化为原始的28x28的结构。

接下来定义两个卷积层和全连接层:

为了减少过拟合,这里也需要使用到Dropout,正如上一篇文章中所讲到的,最后将Dropout层的输出连接一个Softmax层,得到最后的输出。

最后的步骤就是定义损失函数、选择优化器、迭代训练、评估结果,和前面两篇文章的步骤一致,这里就不再详细讨论了。

注意这里没有使用全部的测试数据集,因为在我的GTX 960显卡下,一次性喂入所有的测试数据集,会出现内存不足的问题,所以这里只随机选择了2000个数据集。

总结

这个简单的卷积神经网络模型的准确率大约为99.2%,基本可以满足对手写数字识别准确率的要求。相比之前的深度神经网络2%的错误率,CNN的错误率下降了60%。这其中主要的性能提升都来自更优秀的网络模型设计,充分说明卷积网络对图像特征的提取和抽象能力。依靠卷积核的权值共享,CNN的参数数量并没有爆炸,降低计算量的同时也减轻了过拟合,整个模型的性能有着较大的提升。

至此,一个比较实用的手写数字识别程序就完成了,你也可以尝试增加几个卷积层,检验一下效果。这三篇文章中,我们先设计出了一个简单的机器学习模型,然后逐步优化模型。在实际工作中,我们也通常遵循这样一个流程。通过这一系列的文章,想必对TensorFlow的编程流程也有一定的掌握。

参考

TensorFlow实战,黄文坚、唐源著,电子工业出版社。

  • 发表于:
  • 原文链接https://kuaibao.qq.com/s/20180716G1SS0J00?refer=cp_1026
  • 腾讯「腾讯云开发者社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 cloudcommunity@tencent.com 删除。

扫码关注腾讯云开发者

领取腾讯云代金券