Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Adversarially Regularized Graph Autoencoder for Graph Embedding

About

Graph embedding is an effective method to represent graph data in a low dimensional space for graph analytics. Most existing embedding algorithms typically focus on preserving the topological structure or minimizing the reconstruction errors of graph data, but they have mostly ignored the data distribution of the latent codes from the graphs, which often results in inferior embedding in real-world graph data. In this paper, we propose a novel adversarial graph embedding framework for graph data. The framework encodes the topological structure and node content in a graph to a compact representation, on which a decoder is trained to reconstruct the graph structure. Furthermore, the latent representation is enforced to match a prior distribution via an adversarial training scheme. To learn a robust embedding, two variants of adversarial approaches, adversarially regularized graph autoencoder (ARGA) and adversarially regularized variational graph autoencoder (ARVGA), are developed. Experimental studies on real-world graphs validate our design and demonstrate that our algorithms outperform baselines by a wide margin in link prediction, graph clustering, and graph visualization tasks.

Shirui Pan, Ruiqi Hu, Guodong Long, Jing Jiang, Lina Yao, Chengqi Zhang• 2018

Related benchmarks

TaskDatasetResultRank
Node ClassificationCora (test)--
951
Node ClusteringCora
NMI45.4
179
Link PredictionCiteseer
AUC92.4
174
Link PredictionPubmed
AUC97.3
173
Molecular property predictionMoleculeNet BBBP (scaffold)
ROC AUC70.5
158
Node ClusteringCiteseer
NMI35
151
Molecular property predictionMoleculeNet SIDER (scaffold)
ROC-AUC0.591
136
Node ClassificationPPI (test)
F1 (micro)0.6579
126
Molecular property predictionMoleculeNet BACE (scaffold)
ROC-AUC74.2
126
Link PredictionPubMed (test)
AUC96.8
120
Showing 10 of 65 rows

Other info

Follow for update