Command Palette
Search for a command to run...
Weishuo Ma Yanbo Wang Xiyuan Wang Muhan Zhang

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
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| link-property-prediction-on-ogbl-collab | Refined-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-ddi | Refined-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-ppa | Refined-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.