Joint Structure Similarity and Class Information for Image Classification

Wei XIONG,Hao LIU,Yuejingyan WANG,Juan WANG,Chunyan ZENG,Fan ZHANG
DOI: https://doi.org/10.3778/j.issn.1002-8331.1805-0289
2019-01-01
Abstract:A kind of weighted joint structure similarity and class information method is proposed for the problem of slow convergence of convolutional neural network training. Firstly, this paper designs a convolutional neural network which can effectively extract high-level information for small images. Secondly, this paper constructs a weighted joint structure simi-larity and class information loss function to train convolutional neural networks. Finally, the effectiveness of the designed network structure is verified by mnist handwritten numeral and cifar10 image classification experiments. The experimental results show that the error rate of image classification on mnist handwritten and cifar10 datasets are 0.33% and 11% by using the designed network, respectively. In the case of no data argumentation on the mnist dataset, the performance of the designed network is much better than all single networks. On the cifar10 dataset, the designed network can achieve higher image classification accuracy with less computation. At the same time, joint structure similarity and class information loss can speed up the network training process.
What problem does this paper attempt to address?