How Can Graph Neural Networks Help Document Retrieval: A Case Study on CORD19 with Concept Map Generation

An overview of GNN-based document retrieval.
Publication
European Conference on Information Retrieval

Graph neural networks (GNNs), as a group of powerful tools for representation learning on irregular data, have manifested superiority in various downstream tasks. With unstructured texts represented as concept maps, GNNs can be exploited for tasks like document retrieval. Intrigued by how can GNNs help document retrieval, we conduct an empirical study on a large-scale multi-discipline dataset CORD-19. Results show that instead of the complex structure-oriented GNNs such as GINs and GATs, our proposed semantics-oriented graph functions achieve better and more stable performance based on the BM25 retrieved candidates. Our insights in this case study can serve as a guideline for future work to develop effective GNNs with appropriate semantics-oriented inductive biases for textual reasoning tasks like document retrieval and classification.

Hejie Cui
Hejie Cui
Doctoral Student in CS

My research interests include graph machine learning and knowledge graphs.