Skip to content
/ RGCL Public

Ratioanle-aware Graph Contrastive Learning codebase

Notifications You must be signed in to change notification settings

lsh0520/RGCL

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

26 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Let Invariant Discovery Inspire Graph Contrastive Learning

This is our PyTorch implementation for the paper:

Sihang Li, Xiang Wang*, An Zhang, Ying-Xin Wu, Xiangnan He and Tat-Seng Chua (2022). Let Invariant Rationale Discovery Inspire Graph Contrastive Learning, Paper in arXiv. In ICML'22, Baltimore, Maryland, USA, July 17-23, 2022.

Author: Sihang Li (sihang0520 at gmail.com)

Introduction

Without supervision signals, Rationale-aware Graph Contrastive Learning (RGCL) uses a rationale generator to reveal salient features about graph instance-discrimination as the rationale, and then creates rationale-aware views for contrastive learning. This rationale-aware pre-training scheme endows the backbone model with the powerful representation ability, further facilitating the fine-tuning on downstream tasks.

Citation

If you want to use our codes and datasets in your research, please cite:

@inproceedings{RGCL,
  author    = {Sihang Li and
               Xiang Wang and
               An Zhang and
               Xiangnan He and
               Tat-Seng Chua},
  title     = {Let Invariant Rationale Discovery Inspire Graph Contrastive Learning},
  booktitle = {{ICML}},
  year      = {2022}
}

Experiments

  • Transfer Learning on MoleculeNet datasets
  • Semi-supervised learning on Superpixel MNIST dataset
  • Unsupervised representation learning on TU datasets

Potential Issues

Some issues might occur due to the version mismatch.

Acknowledgements

The backbone implementation is reference to https://github.com/Shen-Lab/GraphCL.