The implementation for Hierarchical Topology Isomorphism Expertise Embedded Graph Contrastive Learning at AAAI 2024.
-
Python 3 (tested on 3.7.12)
-
PyTorch (tested on 1.11.0+cu113)
-
CUDA (tested on 11.4)
-
torch-geometric (tested on 2.2.0)
For install Dependencies, run
pip install -r requirements.txt
-
Unsupervised representation learning in "unsupervised_TU"
-
Transfer learning in "transfer_learning"
The backbone implementation is reference to https://github.com/Shen-Lab/GraphCL.
@misc{li2023hierarchical,
title={Hierarchical Topology Isomorphism Expertise Embedded Graph Contrastive Learning},
author={Jiangmeng Li and Yifan Jin and Hang Gao and Wenwen Qiang and Changwen Zheng and Fuchun Sun},
year={2023},
eprint={2312.14222},
archivePrefix={arXiv},
primaryClass={cs.LG}
}