Distance-Enhanced Graph Neural Network for Link Prediction

2021-05-20 · NA 2021 5 · Boling Li, Yingce Xia, Shufang Xie, Lijun Wu, Tao Qin

Link prediction, which is to predict the existence of a link/edge between two vertices in a graph, is a classical problem in machine learning. Intuitively, if it takes a long distance to walk from u to v along the existing edges, there should not be a link between them, and vice versa. This motivates us to explicitly combine the distance information with graph neural networks (GNNs) to improve link prediction. Calculating the distances between any two vertices (e.g., shortest path, expectation of random walk) in training is time consuming. To overcome this difficulty, we propose an anchorbased distance: First, we randomly select K anchor vertices from the graph and then calculate the shortest distances of all vertices in the graph to them. The distance between vertices u and v is estimated as the average of their distances to the K anchor vertices. After that, we feed the distance into the GNN module. Our method brings significant improvement for link prediction with few additional parameters. We achieved state-of-the-art result on the drug-drug-interaction (i.e., DDI) and protein-protein-association (i.e., PPA) tasks of OGB (Hu et al., 2020). Our code is available at https://github.com/lbn187/DLGNN.

📄 PDF Abstract BibTeX

Code (1)

lbn187/DLGNN pytorch

Tasks

Graph Neural Network Link Prediction Link Property Prediction Prediction

Promoting Fairness in Link Prediction with Graph Enhancement

2024-09-13 · Yezi Liu, Hanning Chen, Mohsen Imani

Link prediction is a crucial task in network analysis, but it has been shown to be prone to biased predictions, particularly when links are unfairly predicted between nodes from different sensitive groups. In this paper,…

Fairness Link Prediction Prediction

Joint embedding in Hierarchical distance and semantic representation learning for link prediction

2023-03-28 · Jin Liu, Jianye Chen, Chongfeng Fan, Fengyu Zhou

The link prediction task aims to predict missing entities or relations in the knowledge graph and is essential for the downstream application. Existing well-known models deal with this task by mainly focusing on represen…

Graph Embedding Knowledge Graph Embedding Link Prediction Prediction+2

Structure Enhanced Graph Neural Networks for Link Prediction

2022-01-14 · Baole Ai, Zhou Qin, Wenting Shen, Yong Li

Graph Neural Networks (GNNs) have shown promising results in various tasks, among which link prediction is an important one. GNN models usually follow a node-centric message passing procedure that aggregates the neighbor…

Graph Neural Network Link Prediction Prediction

Orthogonal Relation Transforms with Graph Context Modeling for Knowledge Graph Embedding

2019-11-09 · ACL 2020 6 · Yun Tang, Jing Huang, Guangtao Wang, Xiaodong He 외

Translational distance-based knowledge graph embedding has shown progressive improvements on the link prediction task, from TransE to the latest state-of-the-art RotatE. However, N-1, 1-N and N-N predictions still remain…

Graph Embedding Knowledge Graph Embedding Link Prediction Prediction+1

A Community-Enhanced Graph Representation Model for Link Prediction

2025-12-24 · Lei Wang, Darong Lai arxiv

Although Graph Neural Networks (GNNs) have become the dominant approach for graph representation learning, their performance on link prediction tasks does not always surpass that of traditional heuristic methods such as …

Graph Representation Learning Link Prediction