文献类型： 会议论文

第一作者： Dingmin Wang

作者： Dingmin Wang ¹ ; Shengchao Liu ² ; Hanchen Wang ³ ; Bernardo Cuenca Grau ¹ ; Linfeng Song ⁴ ; Jian Tang ² ; Le Song ⁵ ; Qi Liu ⁶ ;

作者机构： 1.University of Oxford

2.Mila-HEC Montreal

3.University of Cambridge

4.Tencent AI Lab

5.BioMap and MBZUAI

6.The University of Hong Kong

会议名称： Conference on Prestigious Applications of Intelligent Systems

主办单位：

页码： 2443-2450

摘要： Graph Neural Networks (GNNs) are effective tools for graph representation learning. Most GNNs rely on a recursive neighborhood aggregation scheme, named message passing, thereby their theoretical expressive power is limited to the first-order Weisfeiler-Lehman test (1-WL). An effective approach to this challenge is to explicitly retrieve some annotated examples used to enhance GNN models. While retrieval-enhanced models have been proved to be effective in many language and vision domains, it remains an open question how effective retrieval-enhanced GNNs are when applied to graph datasets. Motivated by this, we want to explore how the retrieval idea can help augment the useful information learned in the graph neural networks, and we design a retrieval-enhanced scheme called GraphRetrieval, which is agnostic to the choice of graph neural network models. In GraphRetrieval, for each input graph, similar graphs together with their ground-true labels are retrieved from an existing database. Thus they can act as a potential enhancement to complete various graph property predictive tasks. We conduct comprehensive experiments over 13 datasets, and we observe that GRAPHRETRIEVAL is able to reach substantial improvements over existing GNNs. Moreover, our empirical study also illustrates that retrieval enhancement is a promising remedy for alleviating the long-tailed label distribution problem.

分类号： tp18-53