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 条
[1]  
[Anonymous], 2017, 5 INT C LEARNING REP
[2]  
Ba J, 2014, ACS SYM SER
[3]   Two-Stream Graph Convolutional Network-Incorporated Latent Feature Analysis [J].
Bi, Fanghui ;
He, Tiantian ;
Xie, Yuetong ;
Luo, Xin .
IEEE TRANSACTIONS ON SERVICES COMPUTING, 2023, 16 (04) :3027-3042
[4]  
Bojchevski A, 2018, PR MACH LEARN RES, V80
[5]   Self-Supervised GANs via Auxiliary Rotation Loss [J].
Chen, Ting ;
Zhai, Xiaohua ;
Ritter, Marvin ;
Lucic, Mario ;
Houlsby, Neil .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :12146-12155
[6]  
De Cao N, 2018, Arxiv, DOI [arXiv:1805.11973, DOI 10.48550/ARXIV.1805.11973]
[7]  
Devon Hjelm R., 2019, INT C LEARN REPR
[8]   Unsupervised Visual Representation Learning by Context Prediction [J].
Doersch, Carl ;
Gupta, Abhinav ;
Efros, Alexei A. .
2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, :1422-1430
[9]  
Fout A, 2017, ADV NEUR IN, V30
[10]  
Gallicchio C, 2020, AAAI CONF ARTIF INTE, V34, P3898