A Convolutional Neural Network Based Image Classification Method

Abstract

The invention relates to a picture classification method based on a convolutional neural network, which improves the structure of the convolutional neural network, adds a feature filtering layer in the hidden layer, and removes some bands by filtering a large number of features extracted by the convolutional network. Features with more noise, improve the training efficiency of convolutional network, reduce training time, reduce memory usage requirements, and integrate a variety of training techniques to make convolutional neural network training converge to a better solution and avoid training parameters from falling into the local minimum area , by reducing the number of network parameters and reducing the possibility of over-fitting of the neural network, it can effectively improve the accuracy and efficiency of image classification.

Description

technical field [0001] The invention relates to a picture classification method based on a convolutional neural network, belonging to the technical field of image classification. Background technique [0002] In recent years, with the emergence of millions of labeled training sets and the emergence of GPU-based training algorithms, it is no longer a luxury to train complex convolutional network models. Convolutional neural networks have gradually developed and attracted widespread attention as an efficient recognition method. A large number of models based on convolutional neural networks have achieved good results in handwriting recognition and classification tests of the ImageNet library. Many papers have used convolutional neural networks to achieve good results in visual classification tasks. [0003] The basic structure of the convolutional neural network includes two layers, one is the feature extraction layer, the input of each neuron is connected to the local recept...

AI Technical Summary

Problems solved by technology

[0004] Due to the huge parameters of the convolutional neural network, in the field of face recognition, the number of network parameters ranges from several megabytes to hundreds of megabytes, which leads to long neural network training time, consumes a lot of memory space, and has relatively high requirements for the number of training samples. Prone to overfitting or convergence to a local minimum area

In addition, the characteristics obtained during the training and analysis of the network model are not easy to observe and difficult to analyze to adjust the network

Finally, the training method of the neural network is the gradient descent algorithm, which is not only susceptible to the influence of the initial value, but also cannot reflect the invariance ability of neurons

Images