We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Pairwise Supervised Hashing with Bernoulli Variational Auto-Encoder and Self-Control Gradient Estimator

Abstract: Semantic hashing has become a crucial component of fast similarity search in many large-scale information retrieval systems, in particular, for text data. Variational auto-encoders (VAEs) with binary latent variables as hashing codes provide state-of-the-art performance in terms of precision for document retrieval. We propose a pairwise loss function with discrete latent VAE to reward within-class similarity and between-class dissimilarity for supervised hashing. Instead of solving the optimization relying on existing biased gradient estimators, an unbiased low-variance gradient estimator is adopted to optimize the hashing function by evaluating the non-differentiable loss function over two correlated sets of binary hashing codes to control the variance of gradient estimates. This new semantic hashing framework achieves superior performance compared to the state-of-the-arts, as demonstrated by our comprehensive experiments.
Comments: To appear in UAI 2020
Subjects: Machine Learning (cs.LG); Machine Learning (stat.ML)
Journal reference: Uncertainty in Artificial Intelligence Conference (UAI) 2020
Cite as: arXiv:2005.10477 [cs.LG]
  (or arXiv:2005.10477v1 [cs.LG] for this version)

Submission history

From: Shahin Boluki [view email]
[v1] Thu, 21 May 2020 06:11:33 GMT (651kb,D)

Link back to: arXiv, form interface, contact.