We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: GCC: Graph Contrastive Coding for Graph Neural Network Pre-Training

Abstract: Graph representation learning has emerged as a powerful technique for addressing real-world problems. Various downstream graph learning tasks have benefited from its recent developments, such as node classification, similarity search, and graph classification. However, prior arts on graph representation learning focus on domain specific problems and train a dedicated model for each graph dataset, which is usually non-transferable to out-of-domain data. Inspired by the recent advances in pre-training from natural language processing and computer vision, we design Graph Contrastive Coding (GCC) -- a self-supervised graph neural network pre-training framework -- to capture the universal network topological properties across multiple networks. We design GCC's pre-training task as subgraph instance discrimination in and across networks and leverage contrastive learning to empower graph neural networks to learn the intrinsic and transferable structural representations. We conduct extensive experiments on three graph learning tasks and ten graph datasets. The results show that GCC pre-trained on a collection of diverse datasets can achieve competitive or better performance to its task-specific and trained-from-scratch counterparts. This suggests that the pre-training and fine-tuning paradigm presents great potential for graph representation learning.
Comments: 11 pages, 5 figures, to appear in KDD 2020 proceedings
Subjects: Machine Learning (cs.LG); Social and Information Networks (cs.SI); Machine Learning (stat.ML)
DOI: 10.1145/3394486.3403168
Cite as: arXiv:2006.09963 [cs.LG]
  (or arXiv:2006.09963v3 [cs.LG] for this version)

Submission history

From: Jiezhong Qiu [view email]
[v1] Wed, 17 Jun 2020 16:18:35 GMT (486kb,D)
[v2] Thu, 18 Jun 2020 08:44:52 GMT (486kb,D)
[v3] Thu, 2 Jul 2020 06:38:24 GMT (947kb,D)

Link back to: arXiv, form interface, contact.