BinGAN: Learning Compact Binary Descriptors with a Regularized GAN

Performance comparison (mAP, %) of different unsupervised hashing algorithms on the CIFAR-10 dataset and top retrived matches given a query image

Abstract

In this paper, we propose a novel regularization method for Generative Adversarial Networks that allows the model to learn discriminative yet compact binary representations of image patches (image descriptors). We exploit the dimensionality reduction that takes place in the intermediate layers of the discriminator network and train the binarized penultimate layer’s low-dimensional representation to mimic the distribution of the higher-dimensional preceding layers. To achieve this, we introduce two loss terms that aim at: (i) reducing the correlation between the dimensions of the binarized penultimate layer’s low-dimensional representation (i.e. maximizing joint entropy) and (ii) propagating the relations between the dimensions in the high-dimensional space to the low-dimensional space. We evaluate the resulting binary image descriptors on two challenging applications, image matching and retrieval, where they achieve state-of-the-art results.

Publication
In 31st Advances in Neural Information Processing Systems 2018 Proceedings
Tomasz Trzciński
Tomasz Trzciński
Principal Investigator

Related