Knowledge enhanced edge-driven graph neural ranking for biomedical information retrieval

Abstract

Neural networks used for information retrieval tend to capture textual matching signals between a query and a document. However, neural ranking models for biomedical information retrieval often struggle to semantically well match the query to the documents. The main reasons are that biomedical terms have many different representations and the fact description related to the query is non-consecutive and non-local in the documents. In this paper, we present an edge-driven graph neural ranking method for biomedical information retrieval by incorporating knowledge from medical databases. First, we propose to form an edge-driven graph by connecting some biomedical terms in the query and the document through different types of edges. Then, we design a novel way of knowledge integration to introduce knowledge related to biomedical terms into the graph and construct a knowledge-query-doc graph. Finally, a graph neural ranking model is applied to capture non-local and non-contiguous match signals between the query and the document. Experimental results show on the biomedical datasets that our method outperforms the advanced neural models. And further analysis shows that the knowledge integration method can well reduce the semantic gap between the query and the document, and our graph model can provide interpretation for matching between the query and the document.