junxia97 / SimGRACE

[WWW 2022] "SimGRACE: A Simple Framework for Graph Contrastive Learning without Data Augmentation"
77 stars 6 forks source link
graph-contrastive-learning graph-neural-networks graph-self-supervised-learning molecular-representation-learning pretraining

SimGRACE: A Simple Framework for Graph Contrastive Learning without Data Augmentation (WWW 2022)

PyTorch implementation for SimGRACE: A Simple Framework for Graph Contrastive Learning without Data Augmentation accepted by The Web Conference 2022 (WWW 2022).

Overview

In this repository, we provide the codes of SimGRACE to evaluate its performances in terms of generalizability (unsupervised & semi-supervised learning), transferability (transfer learning) and robustness (adversarial robustness).

Dataset download

Citation

@inproceedings{10.1145/3485447.3512156,
author = {Xia, Jun and Wu, Lirong and Chen, Jintao and Hu, Bozhen and Li, Stan Z.},
title = {SimGRACE: A Simple Framework for Graph Contrastive Learning without Data Augmentation},
year = {2022},
isbn = {9781450390965},
publisher = {Association for Computing Machinery},
address = {New York, NY, USA},
url = {https://doi.org/10.1145/3485447.3512156},
doi = {10.1145/3485447.3512156},
booktitle = {Proceedings of the ACM Web Conference 2022},
pages = {1070–1079},
numpages = {10},
keywords = {graph representation learning, contrastive learning, Graph neural networks, robustness, graph self-supervised learning},
location = {Virtual Event, Lyon, France},
series = {WWW '22}
}

Useful resources for Pretrained Graphs Models (PGMs)

Reference

  1. Graph Contrastive Learning Automated (ICML 2021)
  2. Graph Contrastive Learning with Augmentations (NeurIPS 2020)
  3. Strategies for Pre-training Graph Neural Networks (ICLR 2020)
  4. Adversarial Attack on Graph Structured Data (ICML 2018)