Self-supervised Graph-level Representation Learning with Adversarial Contrastive Learning

被引:41
|
作者
Luo, Xiao [1 ]
Ju, Wei [2 ]
Gu, Yiyang [2 ]
Mao, Zhengyang [2 ]
Liu, Luchen [2 ]
Yuan, Yuhui [3 ]
Zhang, Ming [2 ]
机构
[1] Univ Calif Los Angeles, Dept Comp Sci, Los Angeles, CA 90095 USA
[2] Peking Univ, Sch Comp Sci, Beijing 100871, Peoples R China
[3] Microsoft Res Asia, Beijing, Peoples R China
基金
中国博士后科学基金; 中国国家自然科学基金;
关键词
Graph representation learning; graph neural networks; contrastive learning; self-supervised learning; PREDICTION;
D O I
10.1145/3624018
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
The recently developed unsupervised graph representation learning approaches apply contrastive learning into graph-structured data and achieve promising performance. However, these methods mainly focus on graph augmentation for positive samples, while the negative mining strategies for graph contrastive learning are less explored, leading to sub-optimal performance. To tackle this issue, we propose a Graph Adversarial Contrastive Learning (GraphACL) scheme that learns a bank of negative samples for effective self-supervised whole-graph representation learning. Our GraphACL consists of (i) a graph encoding branch that generates the representations of positive samples and (ii) an adversarial generation branch that produces a bank of negative samples. To generate more powerful hard negative samples, our method minimizes the contrastive loss during encoding updating while maximizing the contrastive loss adversarially over the negative samples for providing the challenging contrastive task. Moreover, the quality of representations produced by the adversarial generation branch is enhanced through the regularization of carefully designed bank divergence loss and bank orthogonality loss. We optimize the parameters of the graph encoding branch and adversarial generation branch alternately. Extensive experiments on 14 real-world benchmarks on both graph classification and transfer learning tasks demonstrate the effectiveness of the proposed approach over existing graph self-supervised representation learning methods.
引用
收藏
页数:23
相关论文
共 50 条
  • [1] Dual-channel graph contrastive learning for self-supervised graph-level representation learning
    Luo, Zhenfei
    Dong, Yixiang
    Zheng, Qinghua
    Liu, Huan
    Luo, Minnan
    PATTERN RECOGNITION, 2023, 139
  • [2] Self-supervised Graph-level Representation Learning with Local and Global Structure
    Xu, Minghao
    Wang, Hang
    Ni, Bingbing
    Guo, Hongyu
    Tang, Jian
    INTERNATIONAL CONFERENCE ON MACHINE LEARNING, VOL 139, 2021, 139
  • [3] HeGCL: Advance Self-Supervised Learning in Heterogeneous Graph-Level Representation
    Shi, Gen
    Zhu, Yifan
    Liu, Jian K.
    Li, Xuesong
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2024, 35 (10) : 13914 - 13925
  • [4] Adversarial Self-Supervised Contrastive Learning
    Kim, Minseon
    Tack, Jihoon
    Hwang, Sung Ju
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS (NEURIPS 2020), 2020, 33
  • [5] Graph Adversarial Self-Supervised Learning
    Yang, Longqi
    Zhang, Liangliang
    Yang, Wenjing
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 34 (NEURIPS 2021), 2021, 34
  • [6] TCGL: Temporal Contrastive Graph for Self-Supervised Video Representation Learning
    Liu, Yang
    Wang, Keze
    Liu, Lingbo
    Lan, Haoyuan
    Lin, Liang
    IEEE TRANSACTIONS ON IMAGE PROCESSING, 2022, 31 : 1978 - 1993
  • [7] Contrastive Self-supervised Learning for Graph Classification
    Zeng, Jiaqi
    Xie, Pengtao
    THIRTY-FIFTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, THIRTY-THIRD CONFERENCE ON INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE AND THE ELEVENTH SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2021, 35 : 10824 - 10832
  • [8] Decoupled Adversarial Contrastive Learning for Self-supervised Adversarial Robustness
    Zhang, Chaoning
    Zhang, Kang
    Zhang, Chenshuang
    Niu, Axi
    Feng, Jiu
    Yoo, Chang D.
    Kweon, In So
    COMPUTER VISION - ECCV 2022, PT XXX, 2022, 13690 : 725 - 742
  • [9] JGCL: Joint Self-Supervised and Supervised Graph Contrastive Learning
    Akkas, Selahattin
    Azad, Ariful
    COMPANION PROCEEDINGS OF THE WEB CONFERENCE 2022, WWW 2022 COMPANION, 2022, : 1099 - 1105
  • [10] Self-Supervised Contrastive Molecular Representation Learning with a Chemical Synthesis Knowledge Graph
    Xie, Jiancong
    Wang, Yi
    Rao, Jiahua
    Zheng, Shuangjia
    Yang, Yuedong
    JOURNAL OF CHEMICAL INFORMATION AND MODELING, 2024, 64 (06) : 1945 - 1954