Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Reconsidering the Performance of GAE in Link Prediction

About

Recent advancements in graph neural networks (GNNs) for link prediction have introduced sophisticated training techniques and model architectures. However, reliance on outdated baselines may exaggerate the benefits of these new approaches. To tackle this issue, we systematically explore Graph Autoencoders (GAEs) by applying model-agnostic tricks in recent methods and tuning hyperparameters. We find that a well-tuned GAE can match the performance of recent sophisticated models while offering superior computational efficiency on widely-used link prediction benchmarks. Our approach delivers substantial performance gains on datasets where structural information dominates and feature data is limited. Specifically, our GAE achieves a state-of-the-art Hits@100 score of 78.41\% on the ogbl-ppa dataset. Furthermore, we examine the impact of various tricks to uncover the reasons behind our success and to guide the design of future methods. Our study emphasizes the critical need to update baselines for a more accurate assessment of progress in GNNs for link prediction. Our code is available at https://github.com/GraphPKU/Refined-GAE.

Weishuo Ma, Yanbo Wang, Xiyuan Wang, Muhan Zhang• 2024

Related benchmarks

TaskDatasetResultRank
Link Predictionogbl-ppa (test)
Hits@1000.7374
77
Link Predictionogbl-citation2 (test)
MRR0.8455
57
Link Predictionogbl-citation2 (official)
MRR88.74
29
Link Predictionogbl-ddi (official)
Hits@2094.43
22
Link PredictionPubmed Planetoid (70%-10%-20% train-val-test)
Hits@1000.8009
11
Link Predictionogbl-ppa (official)
Hits@10078.41
11
Link PredictionCora Planetoid (70%-10%-20% train-val-test)
Hits@10088.17
11
Link PredictionCiteseer Planetoid (70%-10%-20% train-val-test)
Hits@10092.4
11
Link Predictionogbl-collab (official)
Hits@500.6611
11
Showing 9 of 9 rows

Other info

Code

Follow for update