Learning from Noisy Labeled Samples Using Prediction Norm for Image Classification

Learning of the convolutional neural network (CNN) from noisy labeled samples is a crucial problem, and many studies have been conducted. Although the state-of-the-art method called Joint training with Co-Regularization (JoCoR) has achieved high performance, it is a still challenging problem to accurately classify samples with asymmetric noise, i.e., wrong labels between similar classes (e.g., CAT ↔ DOG). In this paper, we newly found that there is a difference between the prediction norm for clean labeled samples and that for noisy labeled samples. In addition, we found that there is a positive correlation between the prediction norm difference and the classification accuracy for the CIFAR-10 dataset. Therefore, we hypothesize that the discriminative power would be improved if we increase the prediction norm difference. Based on this hypothesis, a novel method for learning CNN from noisy labeled samples is proposed. Specifically, we take JoCoR as the base architecture and weight the loss function of JoCoR to make the prediction norm difference large. Experimental results for the CIFAR-10 dataset suggest that our hypothesis is correct. The classification accuracy by the proposed method is higher than some state-of-the-art methods including JoCoR for samples with asymmetric noise as well as those with symmetric noise (i.e., wrong labels between other classes).