用简单的 2D CNN 进行 MNIST 数字识别

雷锋网 AI 研习社按：本文为雷锋网字幕组编译的技术博客，原标题  A simple 2D CNN for MNIST digit recognition，作者为 Sambit Mahapatra。

翻译 | 王祎   校对 | 霍雷刚     整理 |  孔令双

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

1. 准备数据

2. 创建模型并编译

3. 训练模型并评估

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

数据集就使用上文所提到的 MNIST 数据集。MNIST 数据集 (Modified National Institute of Standards and Technoloy 数据集) 是一个大型的手写数字（0 到 9）数据集。该数据集包含 大小为 28x28 的图片 7 万张，其中 6 万张训练图片、1 万张测试图片。第一步，加载数据集，这一步可以很容易地通过 keras api 来实现。

import keras

from keras.datasets import mnist

#load mnist dataset

(X_train, y_train), (X_test, y_test) = mnist.load_data() #everytime loading data won't be so easy :)

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

import matplotlib.pyplot as plt

fig = plt.figure()

for i in range(9):

plt.subplot(3,3,i+1)

plt.tight_layout()

plt.imshow(X_train[i], cmap='gray', interpolation='none')

plt.title("Digit: {}".format(y_train[i]))

plt.xticks([])

plt.yticks([])

fig

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

#reshaping

#this assumes our data format

#For 3D data, "channels_last" assumes (conv_dim1, conv_dim2, conv_dim3, channels) while

#"channels_first" assumes (channels, conv_dim1, conv_dim2, conv_dim3).

if k.image_data_format() == 'channels_first':

X_train = X_train.reshape(X_train.shape[0], 1, img_rows, img_cols)

X_test = X_test.reshape(X_test.shape[0], 1, img_rows, img_cols)

input_shape = (1, img_rows, img_cols)

else:

X_train = X_train.reshape(X_train.shape[0], img_rows, img_cols, 1)

X_test = X_test.reshape(X_test.shape[0], img_rows, img_cols, 1)

input_shape = (img_rows, img_cols, 1)

#more reshaping

X_train = X_train.astype('float32')

X_test = X_test.astype('float32')

X_train /= 255

X_test /= 255

print('X_train shape:', X_train.shape) #X_train shape: (60000, 28, 28, 1)

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

import keras

#set number of categories

num_category = 10

# convert class vectors to binary class matrices

创建模型并编译

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

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

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

##model building

model = Sequential()

#convolutional layer with rectified linear unit activation

model.add(Conv2D(32, kernel_size=(3, 3),

activation='relu',

input_shape=input_shape))

#32 convolution filters used each of size 3x3

#again

model.add(Conv2D(64, (3, 3), activation='relu'))

#64 convolution filters used each of size 3x3

#choose the best features via pooling

model.add(MaxPooling2D(pool_size=(2, 2)))

#randomly turn neurons on and off to improve convergence

model.add(Dropout(0.25))

#flatten since too many dimensions, we only want a classification output

model.add(Flatten())

#fully connected to get all relevant data

model.add(Dense(128, activation='relu'))

#one more dropout for convergence' sake :)

model.add(Dropout(0.5))

#output a softmax to squash the matrix into output probabilities

model.add(Dense(num_category, activation='softmax'))

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

#Adaptive learning rate (adaDelta) is a popular form of gradient descent rivaled only by adam and adagrad

#categorical ce since we have multiple classes (10)

optimizer=keras.optimizers.Adadelta(),

metrics=['accuracy'])

训练模型并评估

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

batch_size = 128

num_epoch = 10

#model training

model_log = model.fit(X_train, y_train,

batch_size=batch_size,

epochs=num_epoch,

verbose=1,

validation_data=(X_test, y_test))

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

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

score = model.evaluate(X_test, y_test, verbose=0)

print('Test accuracy:', score[1]) #Test accuracy: 0.9904

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

import os

# plotting the metrics

fig = plt.figure()

plt.subplot(2,1,1)

plt.plot(model_log.history['acc'])

plt.plot(model_log.history['val_acc'])

plt.title('model accuracy')

plt.ylabel('accuracy')

plt.xlabel('epoch')

plt.legend(['train', 'test'], loc='lower right')

plt.subplot(2,1,2)

plt.plot(model_log.history['loss'])

plt.plot(model_log.history['val_loss'])

plt.title('model loss')

plt.ylabel('loss')

plt.xlabel('epoch')

plt.legend(['train', 'test'], loc='upper right')

plt.tight_layout()

fig

将模型存盘以便下次使用

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

#Save the model

# serialize model to JSON

model_digit_json = model.to_json()

with open("model_digit.json", "w") as json_file:

json_file.write(model_digit_json)

# serialize weights to HDF5

model.save_weights("model_digit.h5")

print("Saved model to disk")

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

享受深度学习吧！

参考资料

Getting started with the Keras Sequential model

CS231n Convolutional Neural Networks for Visual Recognition

sambit9238/Deep-Learning

雷锋网字幕组编译

号外号外~

一个专注于

AI技术发展和AI工程师成长的求知求职社区

诞生啦！

欢迎大家访问以下链接或者扫码体验

https://club.leiphone.com/page/home