Papers › Distance-Enhanced Graph Neural Network for Link Prediction
Distance-Enhanced Graph Neural Network for Link Prediction
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.
Code
Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.
Code Syntology ran Syntology
Not run by Syntology. Nothing on this page verifies that the listed code works.
Tasks
Results from the paper archive 2025-07-28
| Task | Dataset | Model | Metric | Value | Rank at snapshot | Leaderboard | Report |
|---|---|---|---|---|---|---|---|
| Link Property Prediction | ogbl-ddi | GraphSAGE+anchor distance | Ext. data | No | #12 of 31 | Archive leaderboard | report |
| Link Property Prediction | ogbl-ddi | GraphSAGE+anchor distance | Number of params | 3760134 | #12 of 31 | Archive leaderboard | report |
| Link Property Prediction | ogbl-ddi | GraphSAGE+anchor distance | Test Hits@20 | 0.8239 ± 0.0437 | #12 of 31 | Archive leaderboard | report |
| Link Property Prediction | ogbl-ddi | GraphSAGE+anchor distance | Validation Hits@20 | 0.8206 ± 0.0298 | #12 of 31 | Archive leaderboard | report |
Ranks are positions in the archive's leaderboards as they stood at the 2025-07-28 snapshot. Results published since then are not among these rows, so a rank here is not a current standing.
Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections