In this paper, a semi-supervised classification algorithm that is based on weighted pseudo labeled data and mutual learning is proposed. The purpose of our method is to improve the classification performance of semi-supervised learning models and rectify the incorrect pseudo labels in the process of training. Specifically, the algorithm is built with a deep convolutional neural network and an ensemble learning model. First, output smearing is employed to construct different training sets and perform model initialization. The pseudo labels of unlabeled data are inferred by network predictions. Second, based on selection and weighting strategies for pseudo labeled data, pseudo labeled data with high confidence are selected and added to the real labeled training set. Accordingly, the model is retrained on the weighted pseudo labeled data. Last, a mutual learning strategy is applied to enhance the prediction consistency among classifiers. Furthermore, diversity fine tuning and mutual learning are performed alternately to determine the optimal balance between diversity and consistency, which consequently improves the accuracy of the pseudo label predictions. The experimental results on three benchmark datasets, namely, MNIST, CIFAR10 and SVHN, demonstrate that the proposed method effectively rectifies the incorrect pseudo labels. Notably, the method achieves the best performance compared with state-of-the-art semisupervised classification methods.INDEX TERMS Image classification, semi-supervised learning, weighted pseudo labeled data, mutual learning.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.
hi@scite.ai
10624 S. Eastern Ave., Ste. A-614
Henderson, NV 89052, USA
Copyright © 2024 scite LLC. All rights reserved.
Made with 💙 for researchers
Part of the Research Solutions Family.