Simple Unsupervised Graph Representation Learning

Yujie Mo,Liang Peng,Jie Xu,Xiaoshuang Shi,Xiaofeng Zhu
DOI: https://doi.org/10.1609/aaai.v36i7.20748
2022-01-01
Proceedings of the AAAI Conference on Artificial Intelligence
Abstract:In this paper, we propose a simple unsupervised graph representation learning method to conduct effective and efficient contrastive learning. Specifically, the proposed multiplet loss explores the complementary information between the structural information and neighbor information to enlarge the inter-class variation, as well as adds an upper bound loss to achieve the finite distance between positive embeddings and anchor embeddings for reducing the intra-class variation. As a result, both enlarging inter-class variation and reducing intra-class variation result in small generalization error, thereby obtaining an effective model. Furthermore, our method removes widely used data augmentation and discriminator from previous graph contrastive learning methods, meanwhile available to output low-dimensional embeddings, leading to an efficient model. Experimental results on various real-world datasets demonstrate the effectiveness and efficiency of our method, compared to state-of-the-art methods. The source codes are released at https://github.com/YujieMo/SUGRL.
What problem does this paper attempt to address?