Timezone: »
Graph Neural Networks (GNNs) have recently been applied to graph learning tasks and achieved state-of-the-art (SOTA) results. However, many competitive methods run GNNs multiple times with subgraph extraction and customized labeling to capture information that is hard for normal GNNs to learn. Such operations are time-consuming and do not scale to large graphs. In this paper, we propose an efficient GNN framework called Geodesic GNN (GDGNN) that requires only one GNN run and injects conditional relationships between nodes into the model without labeling. This strategy effectively reduces the runtime of subgraph methods. Specifically, we view the shortest paths between two nodes as the spatial graph context of the neighborhood around them. The GNN embeddings of nodes on the shortest paths are used to generate geodesic representations. Conditioned on the geodesic representations, GDGNN can generate node, link, and graph representations that carry much richer structural information than plain GNNs. We theoretically prove that GDGNN is more powerful than plain GNNs. We present experimental results to show that GDGNN achieves highly competitive performance with SOTA GNN models on various graph learning tasks while taking significantly less time.
Author Information
Lecheng Kong (Washington University, Saint Louis)
Yixin Chen (Washington University in St. Louis)
Muhan Zhang (Peking University)
More from the Same Authors
-
2022 Poster: Rethinking Knowledge Graph Evaluation Under the Open-World Assumption »
Haotong Yang · Zhouchen Lin · Muhan Zhang -
2022 Poster: How Powerful are K-hop Message Passing Graph Neural Networks »
Jiarui Feng · Yixin Chen · Fuhai Li · Anindya Sarkar · Muhan Zhang -
2019 Poster: D-VAE: A Variational Autoencoder for Directed Acyclic Graphs »
Muhan Zhang · Shali Jiang · Zhicheng Cui · Roman Garnett · Yixin Chen -
2018 Poster: Link Prediction Based on Graph Neural Networks »
Muhan Zhang · Yixin Chen -
2018 Spotlight: Link Prediction Based on Graph Neural Networks »
Muhan Zhang · Yixin Chen