HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Reconsidering the Performance of GAE in Link Prediction

Weishuo Ma Yanbo Wang Xiyuan Wang Muhan Zhang

Reconsidering the Performance of GAE in Link Prediction

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.

Code Repositories

graphpku/refined-gae
Official
pytorch
Mentioned in GitHub
GraphPKU/Refined-GAE
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
link-property-prediction-on-ogbl-collabRefined-GAE
Ext. data: No
Number of params: 126669825
Test Hits@50: 0.6816 ± 0.0041
Validation Hits@50: 1.0000 ± 0.0000
link-property-prediction-on-ogbl-ddiRefined-GAE
Ext. data: No
Number of params: 13816833
Test Hits@20: 0.9443 ± 0.0057
Validation Hits@20: 0.7979 ± 0.0159
link-property-prediction-on-ogbl-ppaRefined-GAE
Ext. data: No
Number of params: 295848449
Test Hits@100: 0.7334 ± 0.0092
Validation Hits@100: 0.7391 ± 0.0178

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Reconsidering the Performance of GAE in Link Prediction | Papers | HyperAI