专栏名称: Python开发者
人生苦短,我用 Python。伯乐在线旗下账号「Python开发者」分享 Python 相关的技术文章、工具资源、精选课程、热点资讯等。
目录
相关文章推荐
Python爱好者社区  ·  YYDS!《机器学习基础》PDF下载 ·  2 天前  
Python开发者  ·  三种 Transformer ... ·  2 天前  
Python开发者  ·  美国教授痛心:UC伯克利GPA ... ·  2 天前  
Python爱好者社区  ·  一旦过了​35岁,是不是很难再找到工作? ·  5 天前  
51好读  ›  专栏  ›  Python开发者

构建多层感知器神经网络对数字图片进行文本识别

Python开发者  · 公众号  · Python  · 2017-09-30 16:30

正文

(点击上方蓝字,快速关注我们)


来源:JackieFang

segmentfault.com/a/1190000010800119

如有好文章投稿,请点击 → 这里了解详情


在Keras环境下构建多层感知器模型,对数字图像进行精确识别。


模型不消耗大量计算资源,使用了cpu版本的keras,以Tensorflow 作为backended,在ipython交互环境jupyter notebook中进行编写。


1.数据来源


在Yann LeCun的博客页面上下载开源的mnist数据库:

http://yann.lecun.com/exdb/mnist/


此数据库包含四部分:训练数据集、训练数据集标签、测试数据集、测试数据集标签。由于训练模型为有监督类型的判别模型,因此标签必不可少。若使用该数据集做k-means聚类,则不需要使用标签。将数据整合之后放入user\.keras\datasets文件夹以供调用。


也可以直接从keras建议的url直接下载:https://s3.amazonaws.com/img-datasets/mnist.npz


其中训练数据集包含了60000张手写数字的图片和这些图片分别对应的标签;测试数据集包含了10000张手写数字的图片和这些图片分别对应的标签.



2.数据格式和前期处理(在此不涉及)


训练数据集包含60000张图片,测试数据集包含10000张,所有图片都被当量化为28pixel*28pixel的大小。为减少向量长度,将图片灰度处理,每个像素用一个RGB值表示(0~255),这是因为灰度处理后的RGB值加了归一约束,向量长度相是灰度处理前的1/3。至此,每个图片都可以用28*28的向量表示。


3.导入依赖库


打开jupyter notebook后导入依赖库numpy,此处的seed为随机量的标签,可随意设置:


from __future__ import print_function

import numpy as np

np.random.seed(9999)


继续从keras中导入使用到的模块:


from keras.datasets import mnist

from keras.models import Sequential

from keras.layers.core import Dense, Dropout, Activation

from keras.optimizers import SGD, Adam, RMSprop

from keras.utils import np_utils


mnist为之前准备的数据集,Dense为全连接神经元层,Dropout为神经元输入的断接率,Activation为神经元层的激励函数设置。


导入绘图工具,以便之后绘制模型简化图:


from keras.utils.vis_utils import plot_model as plot


4.处理导入的数据集


处理数据集


  1. 为了符合神经网络对输入数据的要求,原本为60000*28*28shape的三维ndarray,改变成了尺寸为60000*784的2维数组,每行为一个example,每一列为一个feature。

  2. 神经网络用到大量线性与求导运算,将输入的feature的数值类型改变为32位float。

  3. 将feature值归一化,原本0~255的feature归一为0~1。

  4. 测试数据集同理。


(X_train, y_train), (X_test, y_test) = mnist.load_data()

X_train = X_train.reshape(60000, 28*28)

X_test = X_test.reshape(10000, 28*28)

X_train = X_train.astype('float32')

X_test = X_test.astype('float32')

X_train /= 255

X_test /= 255


处理标签


文本识别问题本质是一个多元分类问题。将类向量转换为二进制数表示的类矩阵,其中每一行都是每一个example对应一个label。label为10维向量,每一位代表了此label对应的example属于特定类(0~10)的概率。此时Y_train为60000*10的向量,Y_test为10000*10的向量


Y_train = np_utils.to_categorical(y_train, nb_classes)

Y_test = np_utils.to_categorical(y_test, nb_classes)


5.用keras建立神经网络模型


batch_size = 128

nb_classes = 10

nb_epoch = 20

model = Sequential()

model.add(Dense(500, input_shape=(28*28,)))

model.add(Activation('relu'))

model.add(Dropout(0.2))

model.add(Dense(500))

model.add(Activation('relu'))

model.add(Dropout(0.2))

model.add(Dense(500))

model.add(Activation('relu'))

model.add(Dropout(0.2))

model.add(Dense(10))

model.add(Activation('softmax'))


  • 每次iter时,每一批次梯度下降运算所包含的example数量为128;

  • softmax输出值为10维向量;

  • 一共迭代20次iteration。


三层的神经网络,其中输入层为28*28=784维的全连接层。


Hidden Layer有3层,每一层有500个神经元,input layer->hidden layer->output layer都是全连接方式(DENSE)。


hidden layer的激活函数采用ReLu函数,表达式:


如下图所示:



相比与传统的sigmoid函数,ReLU更容易学习优化。因为其分段线性性质,导致其前传、后传、求导都是分段线性。而传统的sigmoid函数,由于两端饱和,在传播过程中容易丢弃信息。且Relu在x


文本识别本质是多元分类(此处为10元分类),因此输出层采用softmax函数进行feature处理,如下图所示:



其中第j个输出层神经元输出值与当层输入feature的关系为:


该神经网络示意图如图所示:



调用summary方法做一个总览:


model.summary()


结果如下:



该神经网络一共有898510个参数,即在后向反馈过程中,每一次用梯度下降都要求898510次导数。


用plot函数打印model:


plot(model, to_file='mlp_model.png',show_shapes=True)


如下图所示:


编译模型,使用cross_entropy交叉熵函数作为loss function,公式如下图所示:



用交叉熵可量化输出向量与标签向量的差异,p与q分别为输出向量与标签向量。对于每一个example,其交叉熵值就是要通过迭代尽量往小优化的值。优过程使用梯度算法,计算过程中使用反向传播算法求导。


交叉熵的作用如下图所示:



在此分类神经网络中,使用判别结果的accuracy作为参数值好坏的度量标准。


6.用数据训练和测试网络


history = model.fit(X_train, Y_train,

                    batch_size=batch_size, nb_epoch=nb_epoch,

                    verbose=1, validation_data=(X_test, Y_test))


在这个地方运行碰到warning,原因是最新版的keras使用的iteration参数名改成了epoch,而非之前沿用的nb_epoch。将上面的代码作修改即可。


训练结果如下所示。第一次迭代,通过对60000/128个的batch训练,已经达到了比较好的结果,accuracy已经高达0.957。之后Loss值继续下降,精确度继续上升。从第9个itearation开始,loss函数值(交叉熵cross_entropy)开始震荡在0.05附近,accuracy保持在0.98以上。说明前9次迭代就已经训练了足够好的θ值和bias,不需要后11次训练。



7.评估模型


用score函数打印模型评估结果:


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

print('Test score:', score[0])

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


输出结果如下图所示:



训练的multi-layer_perceptron神经网络在对数字文本识别时具有98.12%的准确率。


手写数字图片数据库和Iris_Flower_dataset一样,算是dl界的基本素材,可以拿来做很多事情,比如k-means聚类,LSTM(长短记忆网络)。


看完本文有收获?请转发分享给更多人

关注「大数据与机器学习文摘」,成为Top 1%