Convolutional Neural Network Incorporating Misclassification Information for Image Recognition

Junying Hu,Rongrong Fei,Fang Du,Peiju Chang,Jiangshe Zhang
DOI: https://doi.org/10.1007/s00500-023-09374-4
IF: 3.732
2023-01-01
Soft Computing
Abstract:In this paper, we introduce misclassification information for the improved training of convolutional neural network classifiers (CNNCs) for image recognition. We construct an additional autoencoder neural network, called tutor, that forces the CNNCs to learn the difference between the misclassified picture and the picture corresponding to the misclassified category. Making full use of the classification results to guide the CNNCs for purposeful learning is expected to improve learning efficiency and classification performance. We integrate the proposed tutor into several state-of-the-art CNNCs architectures and demonstrate improvement in their recognition performance on CIFAR-10/100 and MNIST datasets. Our results suggest that making the most of misclassification information to guide the training of the model can lead to significant performance improvement.
What problem does this paper attempt to address?