4 个月前

对抗正则化图自动编码器用于图嵌入

对抗正则化图自动编码器用于图嵌入

摘要

图嵌入是一种有效的方法,用于在低维空间中表示图数据以进行图分析。现有的大多数嵌入算法通常专注于保留图数据的拓扑结构或最小化重构误差,但它们大多忽略了来自图的潜在代码的数据分布,这往往导致在真实世界图数据中的嵌入效果较差。本文提出了一种新颖的对抗图嵌入框架,用于处理图数据。该框架将图中的拓扑结构和节点内容编码为紧凑表示,并在此基础上训练解码器以重构图结构。此外,通过对抗训练方案强制潜在表示匹配先验分布。为了学习鲁棒的嵌入,我们开发了两种对抗方法的变体:对抗正则化图自编码器(Adversarially Regularized Graph Autoencoder, ARGA)和对抗正则化变分图自编码器(Adversarially Regularized Variational Graph Autoencoder, ARVGA)。在真实世界图上的实验研究验证了我们的设计,并表明我们的算法在链接预测、图聚类和图可视化任务中显著优于基线方法。

代码仓库

basiralab/LG-DADA
tf
GitHub 中提及
basiralab/HCAE
tf
GitHub 中提及
basiralab/HADA
tf
GitHub 中提及
Ruiqi-Hu/ARGA
tf
GitHub 中提及

基准测试

基准方法指标
graph-clustering-on-citeseerARVGE
ACC: 54.4
ARI: 24.5
F1: 52.9
NMI: 26.1
Precision: 54.9
graph-clustering-on-citeseerARGE
ACC: 57.3
ARI: 34.1
F1: 54.6
NMI: 0.35
Precision: 57.3
graph-clustering-on-coraARGE
ACC: 64
ARI: 35.2
F1: 61.9
NMI: 0.449
Precision: 64.6
graph-clustering-on-coraARVGE
ACC: 63.8
ARI: 37.4
F1: 62.7
NMI: 45
Precision: 62.4
link-prediction-on-citeseerARGE
AP: 93
AUC: 91.9
link-prediction-on-coraARGE
AP: 93.2%
AUC: 92.4%
link-prediction-on-pubmedARGE
AP: 97.1%
AUC: 96.8%

用 AI 构建 AI

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

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

Hyper Newsletters

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