Classification of Universal Applied Datasets Based on Convolutional Neural Networks

Xingwen Zhao,Feng Ye,Lijun Hang
DOI: https://doi.org/10.23919/chicc.2018.8483260
2018-01-01
Abstract:This work focuses on the design of convolutional neural network to achieve precise classification of universal dataset in specific applications. The paper presented a high-resolution application-oriented image dataset with universal characteristics. The dataset includes a total of 20000 images of 12 categories, currently covering the most common categories of digital products, transportation, and animals. Aiming at the dataset, the methods of rotation, translation and flip, were adopted for data augmentation to enrich the diversity of dataset and to prevent the over-fitting problem. A total of eight-weight neural network was designed, including five convolutional neural networks and three full-connection layers. This paper uses the design of overlapping maximum pooling layer to better reflect the details of the information and to minimize the data loss of the pictures. The convolution kernel size of 3 3 was finally determined for feature extraction, which had been shown to be effective in alleviating the under-fitting problem in the early stage. It was proved that the Rectified Linear Units with a low initial learning rate of Stochastic Gradient Descent strategy would get better accuracy. And 12 regularization as well as dropout was used to optimize the model, which effectively prevented the over-fitting problem occurring in the latter part of the model training. The proposed optimal method and network configuration help to obtain an average accuracy of 94% on the dataset, and can precisely classify the transportation as well as digital products in specific applications, and also achieve the classification of the most common animal categories.
What problem does this paper attempt to address?