Natural image classification based on multi-parameter optimization in deep convolutional neural network

Lei Wang, Yanning Zhang, Runping Xi, Lu Ling

Research output: Contribution to journalArticlepeer-review

Abstract

Traditional machine learning algorithms cannot adequately train the parameters of networks using massive data. A deep convolutional neural network based on multi-parameter optimization by the TensorFlow deep learning framework is designed in this paper. In order to improve the training speed and prevent over-fitting, we improve and optimize the multi-parameters, including the batch value, dropout, and momentum in the network structure. The experiment involves training and testing on the standard natural image data sets in cifar-10 and cifar-100. The experimental results show that the method achieve better classification accuracy in less time compared with other algorithms, such as Conv-KN, ImageNet-2010, SVM, LR, and Boosting.

Original languageEnglish
Pages (from-to)2515-2521
Number of pages7
JournalInternational Journal of Performability Engineering
Volume15
Issue number9
DOIs
StatePublished - 2019

Keywords

  • Batch normalization
  • Deep convolutional neural network
  • Dropout
  • Momentum
  • Multi-parameter optimization

Fingerprint

Dive into the research topics of 'Natural image classification based on multi-parameter optimization in deep convolutional neural network'. Together they form a unique fingerprint.

Cite this