arXiv · 2106.10124
Graph Context Encoder: Graph Feature Inpainting for Graph Generation and Self-supervised Pretraining
Abstract
We propose the Graph Context Encoder (GCE), a simple but efficient approach for graph representation learning based on graph feature masking and reconstruction. GCE models are trained to efficiently reconstruct input graphs similarly to a graph autoencoder where node and edge labels are masked. In particular, our model is also allowed to change graph structures by masking and reconstructing graphs augmented by random pseudo-edges. We show that GCE can be used for novel graph generation, with applications for molecule generation. Used as a pretraining method, we also show that GCE improves baseline performances in supervised classification tasks tested on multiple standard benchmark graph datasets.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Oriel Frigo, Rémy Brossard, David Dehaene. 2021-06-18. Graph Context Encoder: Graph Feature Inpainting for Graph Generation and Self-supervised Pretraining. https://arxiv.org/abs/2106.10124
Cite the original work for its findings. Save a collection to share your selection of sources.