SOTAVerified

Reconsidering the Performance of GAE in Link Prediction

2024-11-06Code Available1· sign in to hype

Weishuo Ma, Yanbo Wang, Xiyuan Wang, Muhan Zhang

Code Available — Be the first to reproduce this paper.

Reproduce

Code

Abstract

Various graph neural networks (GNNs) with advanced training techniques and model designs have been proposed for link prediction tasks. However, outdated baseline models may lead to an overestimation of the benefits provided by these novel approaches. To address this, we systematically investigate the potential of Graph Autoencoders (GAE) by meticulously tuning hyperparameters and utilizing the trick of orthogonal embedding and linear propagation. Our findings reveal that a well-optimized GAE can match the performance of more complex models while offering greater computational efficiency.

Tasks

Benchmark Results

DatasetModelMetricClaimedVerifiedStatus
ogbl-collabRefined-GAENumber of params126,669,825Unverified
ogbl-ddiRefined-GAENumber of params13,816,833Unverified
ogbl-ppaRefined-GAENumber of params295,848,449Unverified

Reproductions