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

What's Behind the Mask: Understanding Masked Graph Modeling for Graph Autoencoders

About

The last years have witnessed the emergence of a promising self-supervised learning strategy, referred to as masked autoencoding. However, there is a lack of theoretical understanding of how masking matters on graph autoencoders (GAEs). In this work, we present masked graph autoencoder (MaskGAE), a self-supervised learning framework for graph-structured data. Different from standard GAEs, MaskGAE adopts masked graph modeling (MGM) as a principled pretext task - masking a portion of edges and attempting to reconstruct the missing part with partially visible, unmasked graph structure. To understand whether MGM can help GAEs learn better representations, we provide both theoretical and empirical evidence to comprehensively justify the benefits of this pretext task. Theoretically, we establish close connections between GAEs and contrastive learning, showing that MGM significantly improves the self-supervised learning scheme of GAEs. Empirically, we conduct extensive experiments on a variety of graph benchmarks, demonstrating the superiority of MaskGAE over several state-of-the-arts on both link prediction and node classification tasks.

Jintang Li, Ruofan Wu, Wangbin Sun, Liang Chen, Sheng Tian, Liang Zhu, Changhua Meng, Zibin Zheng, Weiqiang Wang• 2022

Related benchmarks

TaskDatasetResultRank
Graph ClassificationPROTEINS
Accuracy77.3
1252
Graph ClassificationMUTAG
Accuracy88.6
1103
Node ClassificationCiteseer
Accuracy71.4
1037
Node ClassificationChameleon
Accuracy74.5
867
Node ClassificationCornell
Accuracy46.8
851
Node ClassificationTexas
Accuracy0.705
801
Node ClassificationSquirrel
Accuracy68.53
786
Graph ClassificationNCI1
Accuracy82.2
658
Node ClassificationActor
Accuracy33.44
556
Node Classificationogbn-arxiv (test)
Accuracy71.2
497
Showing 10 of 55 rows

Other info

Follow for update