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

Code

• github.com/J-zin/SNUH
  OfficialIn paperpytorch★ 4
• github.com/MindSpore-paper-code-3/code9/tree/main/snuh
  mindspore★ 0
• github.com/MindCode-4/code-14/tree/main/SNUH
  mindspore★ 0
• github.com/pwc-1/Paper-9/tree/main/6/SNUH
  mindspore★ 0
• github.com/MS-Mind/MS-Code-01/tree/main/snuh
  mindspore★ 0
• github.com/MindCode-4/code-10/tree/main/SNUH
  mindspore★ 0

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.\

Tasks

Reproductions