arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1908.11078cs.CLcs.IR

Document Hashing with Mixture-Prior Generative Models

School of Data and Computer Science, Sun Yat-sen University(中山大学数据与计算机学院);Guangdong Key Laboratory of Big Data Analysis and Processing(广东省大数据分析与处理重点实验室);ECE Department, Duke University(杜克大学电气与计算机工程系);CSE Department, SUNY at Buffalo(纽约州立大学布法罗分校计算机科学与工程系)

Wei Dong, Qinliang Su, Dinghan Shen, Changyou Chen

英文摘要:

Hashing is promising for large-scale information retrieval tasks thanks to the efficiency of distance evaluation between binary codes. Generative hashing is often used to generate hashing codes in an unsupervised way. However, existing generative hashing methods only considered the use of simple priors, like Gaussian and Bernoulli priors, which limits these methods to further improve their performance. In this paper, two mixture-prior generative models are proposed, under the objective to produce high-quality hashing codes for documents. Specifically, a Gaussian mixture prior is first imposed onto the variational auto-encoder (VAE), followed by a separate step to cast the continuous latent representation of VAE into binary code. To avoid the performance loss caused by the separate casting, a model using a Bernoulli mixture prior is further developed, in which an end-to-end training is admitted by resorting to the straight-through (ST) discrete gradient estimator. Experimental results on several benchmark datasets demonstrate that the proposed methods, especially the one using Bernoulli mixture priors, consistently outperform existing ones by a substantial margin.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑