Semi-Supervised Image Classification With Self-Paced Cross-Task Networks | IEEE Journals & Magazine | IEEE Xplore

Semi-Supervised Image Classification With Self-Paced Cross-Task Networks


Abstract:

In a semi-supervised setting, direct training of a deep discriminative model on partially labeled images often suffers from overfitting and poor performance, because only...Show More

Abstract:

In a semi-supervised setting, direct training of a deep discriminative model on partially labeled images often suffers from overfitting and poor performance, because only a small number of labeled images are available, and errors in label propagation are, in many cases, inevitable. In this paper, we introduce an auxiliary clustering task to explore the structure of the image data, and judiciously weigh unlabeled data to alleviate the influence of ambiguous data on model training. For this purpose, we propose a cross-task network composed of two streams to jointly learn two tasks: classification and clustering. Based on the model predictions, a large number of pairwise constraints can be generated from unlabeled images, and are fed to the clustering stream. Since pairwise constraints encode weak supervision information, the clustering is tolerant of errors in labeling. Unlabeled images are weighted according to the distances to the clusters discovered, and a better discriminative model is trained on the classification stream associated with a weighted softmax loss. Furthermore, a self-paced learning paradigm is adopted to gradually train our deep model from easy examples to difficult ones. Experimental results on widely used image classification datasets confirm the effectiveness and superiority of the proposed approach.
Published in: IEEE Transactions on Multimedia ( Volume: 20, Issue: 4, April 2018)
Page(s): 851 - 865
Date of Publication: 02 October 2017

ISSN Information:

Funding Agency:


Contact IEEE to Subscribe

References

References is not available for this document.