Self-supervised graph representations with generative adversarial learning

被引:4
作者
Sun, Xuecheng [1 ]
Wang, Zonghui [2 ]
Lu, Zheming [1 ]
Lu, Ziqian [1 ]
机构
[1] Zhejiang Univ, Sch Aeronaut & Astronaut, Hangzhou 310027, Peoples R China
[2] Zhejiang Univ, Coll Comp Sci & Technol, Hangzhou 310027, Peoples R China
关键词
Self-supervised learning; Graph representation learning; Graph convolutional network; Generative adversarial network; MODELS;
D O I
10.1016/j.neucom.2024.127786
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Graph Neural Networks (GNNs) are powerful neural models for representation learning on graphs. In this work we study a self -supervised learning GNN method GraphSSGAN (Graph Self -Supervised Generative Adversarial Network) that learns generalized node representations using only unlabeled data. The core idea lies in learning representations that capture both the local information of the links and the global information of the plausibility of the subgraph samples. Specifically, node features of the original graph are first encoded into latent representations by optimizing a link prediction objective. Then the generator converts noise vectors to node representations in the latent space and predicts the link probabilities from the generated node representations. The discriminator leverages the graph convolutional network (GCN) architecture to produce permutation -invariant graph -level embeddings, and the intermediate node representations are used by simple classifiers in the downstream tasks. In addition, we introduce several technical tricks including GumbelTop- k trick, Gumbel-Softmax trick and mini -batch training via subgraph sampling to improve the training process. Through extensive experiments on node classification and link prediction tasks, we demonstrate the effectiveness of the proposed model and the contribution of GAN framework on graph representation learning.
引用
收藏
页数:8
相关论文
共 53 条
[31]  
Pan EL, 2021, ADV NEUR IN, V34
[32]  
Pan SR, 2018, PROCEEDINGS OF THE TWENTY-SEVENTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P2609
[33]  
Pedregosa F, 2011, J MACH LEARN RES, V12, P2825
[34]  
Pei H., 2020, 8 INT C LEARN REPR I
[35]  
Radford A., 2016, UNSUPERVISED REPRESE
[36]   Simple and Effective Graph Autoencoders with One-Hop Linear Models [J].
Salha, Guillaume ;
Hennequin, Romain ;
Vazirgiannis, Michalis .
MACHINE LEARNING AND KNOWLEDGE DISCOVERY IN DATABASES, ECML PKDD 2020, PT I, 2021, 12457 :319-334
[37]  
Sermanet P, 2018, IEEE INT CONF ROBOT, P1134
[38]   GraphVAE: Towards Generation of Small Graphs Using Variational Autoencoders [J].
Simonovsky, Martin ;
Komodakis, Nikos .
ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING - ICANN 2018, PT I, 2018, 11139 :412-422
[39]  
Sun F.-Y., 2019, INT C LEARN REPR
[40]  
Velickovi.c, 2019, INT C LEARN REPR