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

SeeGera: Self-supervised Semi-implicit Graph Variational Auto-encoders with Masking

About

Generative graph self-supervised learning (SSL) aims to learn node representations by reconstructing the input graph data. However, most existing methods focus on unsupervised learning tasks only and very few work has shown its superiority over the state-of-the-art graph contrastive learning (GCL) models, especially on the classification task. While a very recent model has been proposed to bridge the gap, its performance on unsupervised learning tasks is still unknown. In this paper, to comprehensively enhance the performance of generative graph SSL against other GCL models on both unsupervised and supervised learning tasks, we propose the SeeGera model, which is based on the family of self-supervised variational graph auto-encoder (VGAE). Specifically, SeeGera adopts the semi-implicit variational inference framework, a hierarchical variational framework, and mainly focuses on feature reconstruction and structure/feature masking. On the one hand, SeeGera co-embeds both nodes and features in the encoder and reconstructs both links and features in the decoder. Since feature embeddings contain rich semantic information on features, they can be combined with node embeddings to provide fine-grained knowledge for feature reconstruction. On the other hand, SeeGera adds an additional layer for structure/feature masking to the hierarchical variational framework, which boosts the model generalizability. We conduct extensive experiments comparing SeeGera with 9 other state-of-the-art competitors. Our results show that SeeGera can compare favorably against other state-of-the-art GCL methods in a variety of unsupervised and supervised learning tasks.

Xiang Li, Tiandi Ye, Caihua Shan, Dongsheng Li, Ming Gao• 2023

Related benchmarks

TaskDatasetResultRank
Graph ClassificationPROTEINS
Accuracy74.6
1252
Graph ClassificationMUTAG
Accuracy88.9
1103
Node ClassificationCiteseer
Accuracy73
1037
Node ClassificationPubmed
Accuracy80.4
865
Graph ClassificationNCI1
Accuracy77.6
658
Graph ClassificationCOLLAB
Accuracy80.8
469
Graph ClassificationIMDB-M
Accuracy51.4
425
Node ClassificationPhoto
Accuracy92.8
254
Node ClassificationPhysics
Accuracy95.2
205
Node ClusteringCora
NMI56.1
168
Showing 10 of 25 rows

Other info

Follow for update