Self-supervised reconstructed graph learning for link prediction in bipartite graphs

被引:0
|
作者
Jin, Xu [1 ]
Kong, Desheng [1 ]
Xie, Maoqiang [1 ,2 ]
Huang, Yalou [3 ]
Liu, Mingming [1 ]
Yang, Weiwei [4 ]
Shi, Hao [1 ]
Liu, Yue [1 ]
机构
[1] Nankai Univ, Coll Software, TianJin, Peoples R China
[2] Tianjins Key Lab Software Experience & Human Comp, TianJin, Peoples R China
[3] TianJin Int Joint Acad Biomed, TianJin, Peoples R China
[4] China Elect Divital Innovat, TianJin, Peoples R China
关键词
Graph learning; Contrastive learning; Link prediction; Bipartite graphs;
D O I
10.1016/j.neucom.2024.128250
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Graph Neural Network(GNN) has achieved remarkable performance in classification tasks due to its strong distinctive power of different graph topologies. However, traditional GNNs face great limitations in link prediction tasks since they learn vertex embeddings from fixed input graphs thus the learned embeddings cannot reflect unobserved graph structures. Graph-learning based GNNs have shown better performance by collaborative learning of graph structures and vertex embeddings, but most of them rely on available initial features to refine graphs and almost perform graph learning at once. Recently, some methods utilizes contrastive learning to facilitate link prediction, but their graph augmentation strategies are predefined only on original graphs, and do not introduce unobserved edges into augmented graphs. To this end, a self-supervised reconstructed graph learning (SRGL) method is proposed. The key points of SRGL lie in two folds: Firstly, it generates augmented graphs for contrasting by learning reconstructed graphs and vertex embeddings from each other, which brings unobserved edges into augmented graphs. Secondly, it maximizes the mutual information between edge-level embeddings of reconstructed graphs and the graph-level embedding of an original graph, which guarantees learned reconstructed graphs to be relevant to the original graph.
引用
收藏
页数:11
相关论文
共 50 条
  • [1] Learning Representations for Bipartite Graphs Using Multi-task Self-supervised Learning
    Sethi, Akshay
    Gupta, Sonia
    Malhotra, Aakarsh
    Asthana, Siddhartha
    [J]. MACHINE LEARNING AND KNOWLEDGE DISCOVERY IN DATABASES: RESEARCH TRACK, ECML PKDD 2023, PT III, 2023, 14171 : 19 - 35
  • [2] Self-Supervised Representation Learning via Latent Graph Prediction
    Xie, Yaochen
    Xu, Zhao
    Ji, Shuiwang
    [J]. INTERNATIONAL CONFERENCE ON MACHINE LEARNING, VOL 162, 2022,
  • [3] Hierarchical Molecular Graph Self-Supervised Learning for property prediction
    Zang, Xuan
    Zhao, Xianbing
    Tang, Buzhou
    [J]. COMMUNICATIONS CHEMISTRY, 2023, 6 (01)
  • [4] Hierarchical Molecular Graph Self-Supervised Learning for property prediction
    Xuan Zang
    Xianbing Zhao
    Buzhou Tang
    [J]. Communications Chemistry, 6
  • [5] Graph Barlow Twins: A self-supervised representation learning framework for graphs
    Bielak, Piotr
    Kajdanowicz, Tomasz
    Chawla, Nitesh V.
    [J]. KNOWLEDGE-BASED SYSTEMS, 2022, 256
  • [6] Relational Self-Supervised Learning on Graphs
    Lee, Namkyeong
    Hyun, Dongmin
    Lee, Junseok
    Park, Chanyoung
    [J]. PROCEEDINGS OF THE 31ST ACM INTERNATIONAL CONFERENCE ON INFORMATION AND KNOWLEDGE MANAGEMENT, CIKM 2022, 2022, : 1054 - 1063
  • [7] Self-supervised Graph Learning for Recommendation
    Wu, Jiancan
    Wang, Xiang
    Feng, Fuli
    He, Xiangnan
    Chen, Liang
    Lian, Jianxun
    Xie, Xing
    [J]. SIGIR '21 - PROCEEDINGS OF THE 44TH INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL, 2021, : 726 - 735
  • [8] Graph Adversarial Self-Supervised Learning
    Yang, Longqi
    Zhang, Liangliang
    Yang, Wenjing
    [J]. ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 34 (NEURIPS 2021), 2021, 34
  • [9] Graph Self-Supervised Learning: A Survey
    Liu, Yixin
    Jin, Ming
    Pan, Shirui
    Zhou, Chuan
    Zheng, Yu
    Xia, Feng
    Yu, Philip S.
    [J]. IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2023, 35 (06) : 5879 - 5900
  • [10] Self-Supervised Learning of Contextual Embeddings for Link Prediction in Heterogeneous Networks
    Wang, Ping
    Agarwal, Khushbu
    Ham, Colby
    Choudhury, Sutanay
    Reddy, Chandan K.
    [J]. PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE 2021 (WWW 2021), 2021, : 2946 - 2957