Integrating Semantics and Neighborhood Information with Graph-Driven Generative Models for Document Retrieval

Zijing Ou, Qinliang Su, Jianxing Yu, Bang Liu, Jingwen Wang +3 more
5/27/2021
cs.IRcs.AI

Official code: J-zin/SNUH (4 stars, MIT, last commit 2021-09-09).

Abstract

With the need of fast retrieval speed and small memory footprint, document hashing has been playing a crucial role in large-scale information retrieval. To generate high-quality hashing code, both semantics and neighborhood information are crucial. However, most existing methods leverage only one of them or simply combine them via some intuitive criteria, lacking a theoretical principle to guide the integration process. In this paper, we encode the neighborhood information with a graph-induced Gaussian distribution, and propose to integrate the two types of information with a graph-driven generative model. To deal with the complicated correlations among documents, we further propose a tree-structured approximation method for learning. Under the approximation, we prove that the training objective can be decomposed into terms involving only singleton or pairwise documents, enabling the model to be trained as efficiently as uncorrelated ones. Extensive experimental results on three benchmark datasets show that our method achieves superior performance over state-of-the-art methods, demonstrating the effectiveness of the proposed model for simultaneously preserving semantic and neighborhood information.\

View on arXivView PDF

Code Implementations(1)

J-zin/SNUHOfficial100%
MIT

source code for paper "Integrating Semantics and Neighborhood Information with Graph-DrivenGenerative Models for Document Retrieval"

40Shell, PythonMay 18, 20215 years agoMIT

Similar Papers

Cite this paper

@article{ou2021integrating,
  title  = {Integrating Semantics and Neighborhood Information with Graph-Driven Generative Models for Document Retrieval},
  author = {Zijing Ou and Qinliang Su and Jianxing Yu and Bang Liu and Jingwen Wang and Ruihui Zhao and Changyou Chen and Yefeng Zheng},
  year   = {2021},
  eprint = {2105.13066},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/2105.13066}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2105.13066.svg)](https://codeofpaper.com/paper/2105.13066)

Discussion