Asymmetric Graph Contrastive Learning

被引:1
作者
Chang, Xinglong [1 ,2 ]
Wang, Jianrong [1 ,3 ]
Guo, Rui [3 ]
Wang, Yingkui [4 ]
Li, Weihao [5 ]
机构
[1] Tianjin Univ, Sch New Media & Commun, Tianjin 300350, Peoples R China
[2] Qijia Youdao Network Technol Beijing Co Ltd, Beijing 100012, Peoples R China
[3] Tianjin Univ, Coll Intelligence & Comp, Tianjin 300350, Peoples R China
[4] Tianjin Renai Coll, Dept Comp Sci & Technol, Tianjin 301636, Peoples R China
[5] Data61 CSIRO, Black Mt Labs, Canberra, ACT 2601, Australia
基金
中国国家自然科学基金;
关键词
contrastive learning; graph neural networks; graph representation learning; NEURAL-NETWORK;
D O I
10.3390/math11214505
中图分类号
O1 [数学];
学科分类号
0701 ; 070101 ;
摘要
Learning effective graph representations in an unsupervised manner is a popular research topic in graph data analysis. Recently, contrastive learning has shown its success in unsupervised graph representation learning. However, how to avoid collapsing solutions for contrastive learning methods remains a critical challenge. In this paper, a simple method is proposed to solve this problem for graph representation learning, which is different from existing commonly used techniques (such as negative samples or predictor network). The proposed model mainly relies on an asymmetric design that consists of two graph neural networks (GNNs) with unequal depth layers to learn node representations from two augmented views and defines contrastive loss only based on positive sample pairs. The simple method has lower computational and memory complexity than existing methods. Furthermore, a theoretical analysis proves that the asymmetric design avoids collapsing solutions when training together with a stop-gradient operation. Our method is compared to nine state-of-the-art methods on six real-world datasets to demonstrate its validity and superiority. The ablation experiments further validated the essential role of the asymmetric architecture.
引用
收藏
页数:13
相关论文
共 55 条
  • [1] [Anonymous], 2010, P 13 INT C ARTIFICIA
  • [2] [Anonymous], 2002, P 19 INT C MACH LEAR
  • [3] A Comprehensive Survey of Graph Embedding: Problems, Techniques, and Applications
    Cai, HongYun
    Zheng, Vincent W.
    Chang, Kevin Chen-Chuan
    [J]. IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2018, 30 (09) : 1616 - 1637
  • [4] Chen Ting, 2019, 25 AMERICAS C INFORM
  • [5] Chen XL, 2020, Arxiv, DOI arXiv:2011.10566
  • [6] A Survey on Network Embedding
    Cui, Peng
    Wang, Xiao
    Pei, Jian
    Zhu, Wenwu
    [J]. IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2019, 31 (05) : 833 - 852
  • [7] Devlin J, 2019, 2019 CONFERENCE OF THE NORTH AMERICAN CHAPTER OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS: HUMAN LANGUAGE TECHNOLOGIES (NAACL HLT 2019), VOL. 1, P4171
  • [8] Hjelm RD, 2019, Arxiv, DOI arXiv:1808.06670
  • [9] DeVries T, 2017, Arxiv, DOI [arXiv:1708.04552, DOI 10.48550/ARXIV.1708.04552]
  • [10] Graph Neural Networks for Social Recommendation
    Fan, Wenqi
    Ma, Yao
    Li, Qing
    He, Yuan
    Zhao, Eric
    Tang, Jiliang
    Yin, Dawei
    [J]. WEB CONFERENCE 2019: PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE (WWW 2019), 2019, : 417 - 426