International Business Machines Corporation
Uncertainty guided semi-supervised neural network training for image classification

Abstract:

Aspects of the invention include systems and methods that train a teacher neural network using labeled images to obtain a trained teacher neural network, each pixel of each of the labeled images being assigned a label that indicates one of a set of classifications. A method includes providing a set of unlabeled images to the trained teacher neural network to generate a set of soft-labeled images, each pixel of each of the soft-labeled images being assigned a soft label that indicates one of the set of classifications and an uncertainty value associated with the soft label, and training a student neural network with a subset of the labeled images and the set of soft-labeled images to obtain a trained student neural network. Student-labeled images are obtained from unlabeled images using the trained student neural network.