Benchmarking Graph Neural Networks on Link Prediction
arXiv:2102.12557
Abstract
In this paper, we benchmark several existing graph neural network (GNN) models on different datasets for link predictions. In particular, the graph convolutional network (GCN), GraphSAGE, graph attention network (GAT) as well as variational graph auto-encoder (VGAE) are implemented dedicated to link prediction tasks, in-depth analysis are performed, and results from several different papers are replicated, also a more fair and systematic comparison are provided. Our experiments show these GNN architectures perform similarly on various benchmarks for link prediction tasks.
References in corpus (7)
- Auto-Encoding Variational Bayes
- Semi-Supervised Classification with Graph Convolutional Networks
- Link Prediction in Complex Networks: A Survey
- Hierarchical structure and the prediction of missing links in networks
- Predicting Missing Links via Local Information
- Fast Graph Representation Learning with PyTorch Geometric
- Wiki-CS: A Wikipedia-Based Benchmark for Graph Neural Networks