Electronic Science and Technology ›› 2021, Vol. 34 ›› Issue (2): 52-56.doi: 10.16180/j.cnki.issn1007-7820.2021.02.009

Previous Articles     Next Articles

Image Classification Method Using Convolutional Neural Network Based on New Initial Module

ZHU Bin,LIU Zilong   

  1. School of Optical-Electrical and Computer Engineering,University of Shanghai for Science and Technology, Shanghai 200093,China
  • Received:2019-11-17 Online:2021-02-15 Published:2021-01-22
  • Supported by:
    National Natural Science Foundation of China(61603255)

Abstract:

Among the problems involving image classification, the classification method based on convolutional neural networks is preferred. In order to solve the problems of poor processing capacity and low classification accuracy, a new type of image classification model of convolutional neural network is proposed. The new Inception module is added on the basis of the traditional network model, which enhances the transmission of the feature information of the model and improves the ability of feature expression. Additionally, the performance of the model and the classification accuracy are improved by activation function, data enhancement, batch regularization, initial weight optimization and Adadelta optimization method. The new network model based on the new initial module is used to test the data on the CIFAR-10 data set, and compared with the traditional network model method, which proves that the modulated model effectives improves the network performance.

Key words: convolution neural network, image classification, new Inception module, data augmentation, weight initialization, Adadelta optimization method

CLC Number: 

  • TN391.41