HashGAN: Deep Learning to Hash with Pair Conditional Wasserstein GAN

被引：81

作者：

Cao, Yue ^{[1
]}

Liu, Bin ^{[1
]}

Long, Mingsheng ^{[1
]}

Wang, Jianmin ^{[1
]}

机构：

[1] Tsinghua Univ, Sch Software, Beijing Key Lab Ind Big Data Syst & Applicat, Natl Engn Lab Big Data Software,MOE,KLiss, Beijing, Peoples R China

来源：

2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR) | 2018年

基金：

中国国家自然科学基金; 国家重点研发计划;

关键词：

QUANTIZATION;

D O I：

10.1109/CVPR.2018.00140

中图分类号：

TP18 [人工智能理论];

学科分类号：

081104 ; 0812 ; 0835 ; 1405 ;

摘要：

Deep learning to hash improves image retrieval performance by end-to-end representation learning and hash coding from training data with pairwise similarity information. Subject to the scarcity of similarity information that is often expensive to collect for many application domains, existing deep learning to hash methods may overfit the training data and result in substantial loss of retrieval quality. This paper presents HashGAN, a novel architecture for deep learning to hash, which learns compact binary hash codes from both real images and diverse images synthesized by generative models. The main idea is to augment the training data with nearly real images synthesized from a new Pair Conditional Wasserstein GAN (PC-WGAN) conditioned on the pairwise similarity information. Extensive experiments demonstrate that HashGAN can generate high-quality binary hash codes and yield state-of-the-art image retrieval performance on three benchmarks, NUS-WIDE, CIFAR-10, and MS-COCO.

引用

页码：1287 / 1296

页数：10