InfoGraph: Unsupervised and Semi-supervised Graph-Level Representation Learning via Mutual Information Maximization

Authors: Fan-yun Sun, Jordan Hoffman, Vikas Verma, Jian Tang

Link to Paper

Please install pytorch_geometric first

Code regarding of mutual information maximization is partially referenced from: https://github.com/rdevon/DIM