首页 | 本学科首页   官方微博 | 高级检索  
     

基于Keras手写数字识别模型的改进
引用本文:高宇鹏,胡众义.基于Keras手写数字识别模型的改进[J].计算技术与自动化,2021,40(2):164-169.
作者姓名:高宇鹏  胡众义
作者单位:山西农业大学信息学院 信息工程学院,山西 晋中 030800;温州大学 计算机与人工智能学院,浙江 温州 325035
摘    要:针对结构设计不合理的卷积神经网络导致MNIST识别的准确率低、收敛速度慢和训练参数多等问题,提出卷积神经网络结构的改进模型.改进的模型采用2次卷积、2次池化和3次全连接、采用Relu激活函数和Softmax回顾函数相结合,加入Dropout层防止过拟合,加入Flatten层优化结构.为了缩减代码量,采用API功能强大的Keras模型替代Tensorflow.对MNIST的训练集和测试集数据的准确率进行仿真实验,实验结果表明:采用改进的结构在MNIST的训练中不仅收敛速度快、训练参数少、损失率低,而且在测试集上的准确率达到99.54%、高于改进前的99.25%,对后续手写数字的研究具有重要意义.

关 键 词:Keras  卷积神经网络  MNIST数据集

Improvement of Handwritten Numeral Recognition Model Based on Keras
GAO Yu-peng,HU Zhong-yi.Improvement of Handwritten Numeral Recognition Model Based on Keras[J].Computing Technology and Automation,2021,40(2):164-169.
Authors:GAO Yu-peng  HU Zhong-yi
Affiliation:(1.Deprtment of College of Information Engineering College of Information, Shanxi Agricultural University, Jinzhong,Shanxi 030800, China; 2 Deprtment of College Computer Science and Artificial Intelligence of Technology,Wenzhou University,Wenzhou,Zhejiang 325035,China)
Abstract:Aiming at the problems of low accuracy, slow convergence speed and many training parameters caused by the unreasonable structure design of convolutional neural network, an improved model of convolutional neural network structure is proposed. The improved model adopts 2-fold convolution, 2-pooling and 3-time full connection, combines relu activation function and softmax review function, adds dropout layer to prevent over fitting, and adds flatten layer to optimize the structure. In order to reduce the amount of code, tensorflow is replaced by keras model with powerful API. The experimental results show that the structure of this paper not only has fast convergence speed, less training parameters and low loss rate, but also has an accuracy rate of 99.54% on the test set, which is higher than 99.25% before the improvement. It is of great significance for the subsequent research of handwritten digits.
Keywords:Keras  convolutional neural network  MNIST data set
本文献已被 万方数据 等数据库收录!
点击此处可从《计算技术与自动化》浏览原始摘要信息
点击此处可从《计算技术与自动化》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号