Hashing-Accelerated Graph Neural Networks for Link Prediction

05/29/2021
by   Wei Wu, et al.
0

Networks are ubiquitous in the real world. Link prediction, as one of the key problems for network-structured data, aims to predict whether there exists a link between two nodes. The traditional approaches are based on the explicit similarity computation between the compact node representation by embedding each node into a low-dimensional space. In order to efficiently handle the intensive similarity computation in link prediction, the hashing technique has been successfully used to produce the node representation in the Hamming space. However, the hashing-based link prediction algorithms face accuracy loss from the randomized hashing techniques or inefficiency from the learning to hash techniques in the embedding process. Currently, the Graph Neural Network (GNN) framework has been widely applied to the graph-related tasks in an end-to-end manner, but it commonly requires substantial computational resources and memory costs due to massive parameter learning, which makes the GNN-based algorithms impractical without the help of a powerful workhorse. In this paper, we propose a simple and effective model called #GNN, which balances the trade-off between accuracy and efficiency. #GNN is able to efficiently acquire node representation in the Hamming space for link prediction by exploiting the randomized hashing technique to implement message passing and capture high-order proximity in the GNN framework. Furthermore, we characterize the discriminative power of #GNN in probability. The extensive experimental results demonstrate that the proposed #GNN algorithm achieves accuracy comparable to the learning-based algorithms and outperforms the randomized algorithm, while running significantly faster than the learning-based algorithms. Also, the proposed algorithm shows excellent scalability on a large-scale network with the limited resources.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
02/27/2018

Link Prediction Based on Graph Neural Networks

Traditional methods for link prediction can be categorized into three ma...
research
03/01/2023

Asymmetric Learning for Graph Neural Network based Link Prediction

Link prediction is a fundamental problem in many graph based application...
research
09/11/2023

A parameterised model for link prediction using node centrality and similarity measure based on graph embedding

Link prediction is a key aspect of graph machine learning, with applicat...
research
01/07/2022

Neighbor2vec: an efficient and effective method for Graph Embedding

Graph embedding techniques have led to significant progress in recent ye...
research
09/30/2022

Graph Neural Networks for Link Prediction with Subgraph Sketching

Many Graph Neural Networks (GNNs) perform poorly compared to simple heur...
research
02/24/2023

PaGE-Link: Path-based Graph Neural Network Explanation for Heterogeneous Link Prediction

Transparency and accountability have become major concerns for black-box...
research
08/19/2020

GraphReach: Locality-Aware Graph Neural Networks using Reachability Estimations

Analyzing graphs by representing them in a low dimensional space using G...

Please sign up or login with your details

Forgot password? Click here to reset