
摘要
我们介绍了变分图自编码器(Variational Graph Auto-Encoder, VGAE),这是一种基于变分自编码器(Variational Auto-Encoder, VAE)的框架,用于图结构数据的无监督学习。该模型利用潜在变量,能够为无向图学习可解释的潜在表示。我们通过使用图卷积网络(Graph Convolutional Network, GCN)编码器和简单的内积解码器来展示这一模型。我们的模型在引文网络的链接预测任务中取得了具有竞争力的结果。与大多数现有的图结构数据无监督学习和链接预测模型不同,我们的模型可以自然地融入节点特征,这在多个基准数据集上显著提高了预测性能。
代码仓库
Monti03/LinkPredictionOverClusters
tf
GitHub 中提及
flawless1202/vgae_pyg
pytorch
GitHub 中提及
xiyou3368/DGVAE
tf
GitHub 中提及
Monti03/VGAE
tf
GitHub 中提及
jjzgeeks/vgae-based_model_poisoning_attack_fl
pytorch
GitHub 中提及
Omairss/RepresentationLearning
tf
GitHub 中提及
DaehanKim/vgae_pytorch
pytorch
GitHub 中提及
leffff/vgae-pytorch
pytorch
GitHub 中提及
zfjsail/gae-pytorch
pytorch
GitHub 中提及
qkrdmsghk/GOODHSE
pytorch
GitHub 中提及
fleverest/gravity_graph_autoencoders
tf
GitHub 中提及
deezer/gravity_graph_autoencoders
tf
GitHub 中提及
MysteryVaibhav/DW-GAE
pytorch
GitHub 中提及
deezer/linear_graph_autoencoders
tf
GitHub 中提及
TeamLab/bert-gcn-for-paper-citation
tf
GitHub 中提及
JuliaSun623/VGAE_dgl
pytorch
GitHub 中提及
tkipf/gae
官方
tf
GitHub 中提及
pygod-team/pygod
pytorch
lfhase/ciga
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| graph-clustering-on-citeseer | GAE | ACC: 40.8 |
| graph-clustering-on-cora | GAE | ACC: 59.6 |
| graph-clustering-on-pubmed | VGAE | ACC: 65.48 |
| link-prediction-on-citeseer | Variational graph auto-encoders | ACC: 91.4 |
| link-prediction-on-cora | Variational graph auto-encoders | ACC: 92.0 |
| link-prediction-on-pubmed | Variational graph auto-encoders | ACC: 97.1 |