3 个月前

变分图归一化自编码器

变分图归一化自编码器

摘要

链接预测是图结构数据中的一项关键问题。随着图神经网络的发展,图自编码器(Graph Autoencoders, GAEs)和变分图自编码器(Variational Graph Autoencoders, VGAEs)被提出,用于以无监督方式学习图嵌入,已在链接预测任务中展现出良好的效果。然而,当涉及度为零的节点(即孤立节点)时,这些方法的性能显著下降。我们发现,GAEs/VGAEs会将孤立节点的嵌入表示压缩至接近零向量,无论其内容特征如何。针对这一问题,本文提出一种新型的变分图归一化自编码器(Variational Graph Normalized AutoEncoder, VGNAE),通过引入L2归一化机制,有效提升孤立节点的嵌入质量。实验结果表明,所提出的VGNAE在链接预测任务上优于现有最先进模型。代码已开源,获取地址为:https://github.com/SeongJinAhn/VGNAE。

代码仓库

SeongJinAhn/VGNAE
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
link-prediction-on-citeseerVGNAE
AP: 97.1
AUC: 97
link-prediction-on-citeseerGNAE
AP: 97
AUC: 96.5
link-prediction-on-coraVGNAE
AP: 95.8%
AUC: 95.4%
link-prediction-on-coraGNAE
AP: 95.7%
AUC: 95.6%
link-prediction-on-pubmedVGNAE
AP: 97.6%
AUC: 97.6%
link-prediction-on-pubmedGNAE
AP: 97.5%
AUC: 97.5%

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
变分图归一化自编码器 | 论文 | HyperAI超神经