首页
学习
活动
专区
工具
TVP
发布

我们建了个模型,搞定了 MNIST 数字识别任务

雷锋网(公众号:雷锋网)按:本文为雷锋字幕组编译的技术博客,原标题  A simple 2D CNN for MNIST digit recognition ,作者为Sambit Mahapatra。

翻译 |  王祎  霍雷刚   整理 |  MY

对于图像分类任务,当前最先进的架构是卷积神经网络 (CNNs) 。无论是面部识别、自动驾驶还是目标检测,CNN 得到广泛使用。在本文中,针对著名的 MNIST 数字识别任务,我们设计了一个以 tensorflow 为后台技术、基于 keras 的简单 2D 卷积神经网络 (CNN) 模型。整个工作流程如下:

1. 准备数据

2. 创建模型并编译

3. 训练模型并评估

4. 将模型存盘以便下次使用

1. 准备数据

数据集就使用上文所提到的 MNIST 数据集。MNIST 数据集 ( Modified

National Institute of Standards and Technoloy

数据集) 是一个大型的手写数字(0 到 9)数据集。该数据集包含 大小为 28x28 的图片 7 万张,其中 6 万张训练图片、1 万张测试图片。第一步,加载数据集,这一步可以很容易地通过 keras api 来实现。

其中,X_train 包含 6 万张 大小为 28x28 的训练图片,y_train 包含这些图片对应的标签。与之类似,X_test 包含了 1 万张大小为 28x28 的测试图片,y_test 为其对应的标签。我们将一部分训练数据可视化一下,来对深度学习模型的目标有一个认识吧。

如上所示,左上角图为「5」的图片数据被存在 X_train[0] 中,y_train[0] 中存储其对应的标签「5」。我们的深度学习模型应该能够仅仅通过手写图片预测实际写下的数字。 现在为了准备数据,我们需要对这些图片做一些诸如调整大小、像素值归一化之类的处理。

对图片数据做了必要的处理之后,需要将 y_train 和 y_test 标签数据进行转换,转换成分类的格式。例如,模型构建时,3 应该被转换成向量 [0, 0, 0, 1, 0, 0, 0, 0, 0, 0]。

创建模型并编译

数据加载进模型之后,我们需要定义模型结构,并优化函数、损失函数和性能指标。

接下来定义的架构为 2 个卷积层,分别在每个卷积层后接续一个池化层,一个全连接层和一个 softmax 层。在每一层卷积层上都会使用多个滤波器来提取不同类型的特征。直观的解释是,第一个滤波器有助于检测图片中的直线,第二个滤波器有助于检测图片中的圆形,等等。关于每一层技术实现的解释,将会在后续的帖子中进行讲解。如果想要更好的理解每一层的含义,可以参考 http://cs231n.github.io/convolutional-networks/  。

在最大池化和全连接层之后,在我们的模型中引入dropout来进行正则化,用以消除模型的过拟合问题。

确定模型架构之后需要对模型进行编译。这是项多类别的分类问题,因此我们需要使用 categorical_crossentropy 作为损失函数。由于所有的标签都带有相似的权重,我们更喜欢使用精确度作为性能指标。AdaDelta 是一个很常用的梯度下降方法,我们使用这个方法来优化模型参数。

训练模型并评估

在定义模型架构和编译模型之后,要使用训练集去训练模型,使得模型可以识别手写数字。这里,我们将使用 X_train 和 y_train 来拟合模型。

其中,一个 epoch 表示一次全量训练样例的前向和后向传播。batch_size 就是在一次前向/后向传播过程用到的训练样例的数量。训练输出结果如下:

现在,我们来评估训练得到模型的性能。

测试准确率达到了 99%+,这意味着这个预测模型训练的很成功。如果查看整个训练日志,就会发现随着 epoch 次数的增多,模型在训练数据、测试数据上的损失和准确率逐渐收敛,最终趋于稳定。

将模型存盘以便下次使用

现在需要将训练过的模型进行序列化。模型的架构或者结构保存在 json 文件,权重保存在 hdf 5 文件。

模型被保存后,可以被重用,也可以很方便地移植到其它环境中使用。在以后的帖子中,我们将会演示如何在生产环境中部署这个模型。

享受深度学习吧!

雷锋网雷锋网

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

扫码

添加站长 进交流群

领取专属 10元无门槛券

私享最新 技术干货

扫码加入开发者社群
领券