图神经网络在链接预测上的基准测试
机器学习
2021-02-26 v1
摘要
在本文中，我们在不同数据集上针对链接预测任务对若干现有图神经网络（GNN）模型进行基准测试。具体而言，我们实现了图卷积网络（GCN）、GraphSAGE、图注意力网络（GAT）以及变分图自编码器（VGAE）并专用于链接预测任务，进行了深入分析，复现了若干不同论文的结果，并提供了更为公平和系统的比较。我们的实验表明，这些GNN架构在各种链接预测任务的基准上表现相近。
引用
@article{arxiv.2102.12557,
title = {Benchmarking Graph Neural Networks on Link Prediction},
author = {Xing Wang and Alexander Vinel},
journal= {arXiv preprint arXiv:2102.12557},
year = {2021}
}