InfoGraph: Unsupervised and Semi-supervised Graph-Level Representation Learning via Mutual Information Maximization
Authors: Fan-yun Sun, Jordan Hoffman, Vikas Verma, Jian Tang
Tested on pytorch 1.6.0 and pytorch_geometric 1.6.1
Code regarding mutual information maximization is partially referenced from: https://github.com/rdevon/DIM
Please cite our paper if you use this code in your own work:
@inproceedings{sun2019infograph,
title={InfoGraph: Unsupervised and Semi-supervised Graph-Level Representation Learning via Mutual Information Maximization},
author={Sun, Fan-Yun and Hoffman, Jordan and Verma, Vikas and Tang, Jian},
booktitle={International Conference on Learning Representations},
year={2019}
}