Papers › Neural Bellman-Ford Networks: A General Graph Neural Network Framework for Link Prediction

Neural Bellman-Ford Networks: A General Graph Neural Network Framework for Link Prediction

13 Jun 2021NeurIPS 2021 12arXiv:2106.06935archive 2025-07-28

Zhaocheng Zhu, Zuobai Zhang, Louis-Pascal Xhonneux, Jian Tang

Link prediction is a very fundamental task on graphs. Inspired by traditional path-based methods, in this paper we propose a general and flexible representation learning framework based on paths for link prediction. Specifically, we define the representation of a pair of nodes as the generalized sum of all path representations, with each path representation as the generalized product of the edge representations in the path. Motivated by the Bellman-Ford algorithm for solving the shortest path problem, we show that the proposed path formulation can be efficiently solved by the generalized Bellman-Ford algorithm. To further improve the capacity of the path formulation, we propose the Neural Bellman-Ford Network (NBFNet), a general graph neural network framework that solves the path formulation with learned operators in the generalized Bellman-Ford algorithm. The NBFNet parameterizes the generalized Bellman-Ford algorithm with 3 neural components, namely INDICATOR, MESSAGE and AGGREGATE functions, which corresponds to the boundary condition, multiplication operator, and summation operator respectively. The NBFNet is very general, covers many traditional path-based methods, and can be applied to both homogeneous graphs and multi-relational graphs (e.g., knowledge graphs) in both transductive and inductive settings. Experiments on both homogeneous graphs and knowledge graphs show that the proposed NBFNet outperforms existing methods by a large margin in both transductive and inductive settings, achieving new state-of-the-art results.

PaperPDFConference PDFCodeCode Syntology ran

In Syntology View this paper on Syntology: its repositories, every harvested function with whether it ran, its licence and the call to fetch it.

Open this paper in Syntology's Atlas, the map of the papers in Syntology's graph and their citations.

For agents, Syntology's MCP tool lists every function and class Syntology harvested from this paper and whether it ran (how to connect): get_harvested_code_for_paper(arxiv_id="2106.06935")

Code

Syntology Ran 10 of 15 code samples harvested from 3 repositories linked to this paper; 5 have no recorded run. Of those that ran: 2 ran · our draft was wrong; 8 ran with no contract checked.

By repository: official repository: 2 samples from 1 repository, 0 ran; community (archive-listed): 11 samples from 1 repository, 8 ran; community: 2 samples from 1 repository, 2 ran. The run record, sample by sample. “Ran” means executed on a synthesized input, not that the code is correct or reproduces the paper.

DeepGraphLearning/NBFNet officialmentioned in papermentioned on GitHubpytorchMIT report
fs302/EasyLink mentioned on GitHubpytorchApache-2.0 report
fs302/EasyLink mentioned on GitHubpytorch report

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

15 samples harvested; 10 ran; 0 honoured the contract we drafted; 5 have no recorded run. Read from Syntology's graph 2026-09-25; that is when this build read the record, not when the samples ran.

2ran · our draft was wrong
8ran
5unverified

Licence: 0 of the 15 samples are pointer only, meaning Syntology does not serve that copy's text. This page shows no code text for any sample; each one links to its file in the repository.

Harvested from 3 repositories linked to this paper, official or community; each sample names its own and says which. “Ran” means the sample executed on a synthesized input. It does not mean the output is correct, and nothing here reproduces the paper's results. “Honoured” and “violated” refer to a contract Syntology drafted from the code itself; “our draft was wrong” and “fixture could not drive it” are failures of Syntology's instrument, not of the code.

Each sample ends with its code_sha256, Syntology's identity for that exact code. An agent fetches the stored sample with Syntology's MCP tool get_code(code_sha256="…") (how to connect); click an identity to copy that call.

Repository labels, per sample. official repository: The archive marks this repository official for the paper. named in the paper: The archive records that the paper mentions this repository; it is not marked official. community (archive-listed): In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper. found in paper text by Syntology: Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted. community: Not in the archive's code links for this paper; a community repository Syntology harvested. Samples from a repository marked official are listed first. Licence labels name the repository's licence as recorded at harvest. “Pointer only” means Syntology does not serve that copy's text, for one of four reasons: no licence file was found; the licence was not identified; the licence is recorded as permissive but that copy's record is not marked cleared; or the licence is outside the permissive list Syntology serves text under (MIT, Apache-2.0, BSD and similar). Some licences outside that list permit redistribution, such as WTFPL, and GPL-3.0 under its conditions; they are simply not on the list. Hover a licence label for the reason. File links open the file on GitHub at the default branch, which may have changed since the harvest.

detect_variables DeepGraphLearning/NBFNet/nbfnet/util.py official repository unverified MIT (permissive) · eab8b6a6ced6d93e · report
get_root_logger DeepGraphLearning/NBFNet/nbfnet/util.py official repository unverified MIT (permissive) · 70a68fd646c6e4a7 · report
drnl_node_labeling fs302/EasyLink/easylink/common/seal_utils.py community (archive-listed) ran Apache-2.0 (permissive) · 2aad32b3b74abec9 · report
evaluate_auc fs302/EasyLink/easylink/common/eval_utils.py community (archive-listed) ran Apache-2.0 (permissive) · a27462150ae41053 · report
evaluate_hits fs302/EasyLink/easylink/common/eval_utils.py community (archive-listed) ran Apache-2.0 (permissive) · e18c89b0a6b12062 · report
extract_enclosing_subgraphs fs302/EasyLink/easylink/common/seal_utils.py community (archive-listed) ran Apache-2.0 (permissive) · 9df677a47f14ae09 · report
get_pos_neg_edges fs302/EasyLink/easylink/common/seal_utils.py community (archive-listed) ran Apache-2.0 (permissive) · fcaa1fff9342a53d · report
load_basic_network fs302/EasyLink/easylink/common/data_utils.py community (archive-listed) ran Apache-2.0 (permissive) · 00fab49fcf976780 · report
sparse_to_tuple fs302/EasyLink/easylink/common/data_utils.py community (archive-listed) ran Apache-2.0 (permissive) · 4ada13b9756aa281 · report
train_test_split fs302/EasyLink/easylink/common/data_utils.py community (archive-listed) ran Apache-2.0 (permissive) · f6e29676a8cc379b · report
adamic_adar fs302/EasyLink/easylink/model/heuristic_similarity.py community (archive-listed) unverified Apache-2.0 (permissive) · cb453fdedf2fff1d · report
common_neighbors fs302/EasyLink/easylink/model/heuristic_similarity.py community (archive-listed) unverified Apache-2.0 (permissive) · 5f30e06f74863f65 · report
resource_allocation fs302/EasyLink/easylink/model/heuristic_similarity.py community (archive-listed) unverified Apache-2.0 (permissive) · 213410a6123cfb65 · report
index_to_mask KiddoZhu/NBFNet-PyG/nbfnet/models.py community ran · our draft was wrong MIT (permissive) · 9664f4b5c60ab1bf · report
size_to_index KiddoZhu/NBFNet-PyG/nbfnet/models.py community ran · our draft was wrong MIT (permissive) · 4ddf5f547371a7f7 · report

Tasks

Graph Neural NetworkInductive Relation PredictionKnowledge Graph CompletionLink PredictionLink Property Prediction

Results from the paper archive 2025-07-28

TaskDatasetModelMetricValueRank at snapshotLeaderboardReport
Link Prediction Citeseer NBFNet AP 93.6% #9 of 13 Archive leaderboard report
Link Prediction Citeseer NBFNet AUC 92.3% #9 of 13 Archive leaderboard report
Link Prediction Cora NBFNet AP 96.2% #3 of 13 Archive leaderboard report
Link Prediction Cora NBFNet AUC 95.6% #3 of 13 Archive leaderboard report
Link Prediction FB15k-237 NBFNet Hits@1 0.321 #1 of 75 Archive leaderboard report
Link Prediction FB15k-237 NBFNet Hits@10 0.599 #1 of 75 Archive leaderboard report
Link Prediction FB15k-237 NBFNet Hits@3 0.454 #1 of 75 Archive leaderboard report
Link Prediction FB15k-237 NBFNet MR 114 #1 of 75 Archive leaderboard report
Link Prediction FB15k-237 NBFNet MRR 0.415 #1 of 75 Archive leaderboard report
Link Prediction Pubmed NBFNet AP 98.2% #3 of 13 Archive leaderboard report
Link Prediction Pubmed NBFNet AUC 98.3% #3 of 13 Archive leaderboard report
Link Prediction WN18RR NBFNet Hits@1 0.497 #12 of 75 Archive leaderboard report
Link Prediction WN18RR NBFNet Hits@10 0.666 #12 of 75 Archive leaderboard report
Link Prediction WN18RR NBFNet Hits@3 0.573 #12 of 75 Archive leaderboard report
Link Prediction WN18RR NBFNet MR 636 #12 of 75 Archive leaderboard report
Link Prediction WN18RR NBFNet MRR 0.551 #12 of 75 Archive leaderboard report
Link Prediction YAGO3-10 NBFNet Hits@1 0.480 #8 of 18 Archive leaderboard report
Link Prediction YAGO3-10 NBFNet Hits@10 0.708 #8 of 18 Archive leaderboard report
Link Prediction YAGO3-10 NBFNet Hits@3 0.612 #8 of 18 Archive leaderboard report
Link Prediction YAGO3-10 NBFNet MRR 0.563 #8 of 18 Archive leaderboard report
Link Property Prediction ogbl-biokg NBFNet Ext. data No #10 of 16 Archive leaderboard report
Link Property Prediction ogbl-biokg NBFNet Number of params 734,209 #10 of 16 Archive leaderboard report
Link Property Prediction ogbl-biokg NBFNet Test MRR 0.8317 #10 of 16 Archive leaderboard report
Link Property Prediction ogbl-biokg NBFNet Validation MRR 0.8318 #10 of 16 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.

Methods

Graph Neural Network

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