3 个月前

重新审视GAE在链接预测中的性能

重新审视GAE在链接预测中的性能

摘要

针对链接预测任务,已有多种具备先进训练技术与模型设计的图神经网络(GNN)被提出。然而,使用过时的基线模型可能导致对新型方法实际优势的过度评估。为解决这一问题,本文通过精细调优超参数,并引入正交嵌入(orthogonal embedding)与线性传播(linear propagation)等技巧,系统性地探究了图自编码器(Graph Autoencoders, GAE)的潜力。研究结果表明,经过充分优化的GAE模型在性能上可媲美更为复杂的模型,同时具备更高的计算效率。

代码仓库

graphpku/refined-gae
官方
pytorch
GitHub 中提及
GraphPKU/Refined-GAE
pytorch
GitHub 中提及

基准测试

基准方法指标
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

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
重新审视GAE在链接预测中的性能 | 论文 | HyperAI超神经