Google Inc.

Kai Han

Xuhui Jia

Yukun Zhu

Bradley Ray Green

Systems and methods for providing deep learning models capable of performing joint representation learning and new category discovery on a mixture of labeled and unlabeled data, which may include single- and multi-modal data. In some examples, a flexible end-to-end framework uses unified contrastive learning on labeled and unlabeled data based on both instance discrimination and category discrimination, and further uses Winner-Take-All hashing to generate a pseudo-label based on the similarity between each pair of unlabeled data points that can be used to train the model to generate clustering assignments for each unlabeled data point. In some examples, the unified contrastive learning may be further based on cross-modal discrimination.

Flexible framework for joint representation learning and unknown category discovery

Methods, systems, and apparatus, including computer programs encoded on computer storage media, for performing novel category discovery. One of the methods includes generating first local feature tensors from a first training image; obtaining previous local feature tensors generated from a previous training image; generating a first similarity tensor representing a similarity between the first local feature tensors and the previous local feature tensors; obtaining a second similarity tensor for a second training image; processing, using a neural network, the first training image to generate a first training output representing a class prediction for the first training image; obtaining a second training output representing a class prediction for the second training image; and generating an update to the neural network from (i) a similarity between the first similarity tensor and the second similarity tensor and (ii) a similarity between the first training output and the second training output.

Growing community of inventors

Bristol, United Kingdom

Kai Han