A Unique Framework of Heterogeneous Augmentation Graph Contrastive Learning for Both Node and Graph Classification

被引:0
作者
Shao, Qi [1 ]
Chen, Duxin [1 ]
Yu, Wenwu [2 ,3 ]
机构
[1] Southeast Univ, Sch Math, Nanjing 210096, Peoples R China
[2] Southeast Univ, Frontiers Sci Ctr Mobile Informat Commun & Secur, Sch Math, Nanjing 210096, Peoples R China
[3] Purple Mt Labs, Nanjing 211102, Peoples R China
来源
IEEE TRANSACTIONS ON NETWORK SCIENCE AND ENGINEERING | 2024年 / 11卷 / 06期
基金
中国国家自然科学基金; 国家重点研发计划;
关键词
heterogeneous graph networks; transfer learning; Graph contrastive learning; unsupervised learning;
D O I
10.1109/TNSE.2024.3454993
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
Graph contrastive learning has gained significant attention for its effectiveness in leveraging unlabeled data and achieving superior performance. However, prevalent graph contrastive learning methods often resort to graph augmentation, typically involving the removal of anchor graph structures. This strategy may compromise the essential graph information, constraining the adaptability of contrastive learning approaches across diverse tasks. To overcome this limitation, we introduce a novel augmentation technique for graph contrastive learning: heterogeneous augmentation. Through the application of heterogeneous augmentation to homogeneous anchor graphs, our method obviates the need for modifying edges and nodes, preserving the structural integrity of the anchor graph to the fullest extent. The proposed method could become a significant technique in graph augmentation, potentially influencing further research and development in this area. Our work provides a valuable contribution to the advancement of graph contrastive learning methodologies.
引用
收藏
页码:5818 / 5828
页数:11
相关论文
共 54 条
[1]   Sub2Vec: Feature Learning for Subgraphs [J].
Adhikari, Bijaya ;
Zhang, Yao ;
Ramakrishnan, Naren ;
Prakash, B. Aditya .
ADVANCES IN KNOWLEDGE DISCOVERY AND DATA MINING, PAKDD 2018, PT II, 2018, 10938 :170-182
[2]   Graph Barlow Twins: A self-supervised representation learning framework for graphs [J].
Bielak, Piotr ;
Kajdanowicz, Tomasz ;
Chawla, Nitesh V. .
KNOWLEDGE-BASED SYSTEMS, 2022, 256
[3]   Comparison of Random Forest and Pipeline Pilot Naive Bayes in Prospective QSAR Predictions [J].
Chen, Bin ;
Sheridan, Robert P. ;
Hornak, Viktor ;
Voigt, Johannes H. .
JOURNAL OF CHEMICAL INFORMATION AND MODELING, 2012, 52 (03) :792-803
[4]  
Chen H, 2023, PROCEEDINGS OF THE THIRTY-SECOND INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, IJCAI 2023, P2059
[5]  
Chen T, 2020, PR MACH LEARN RES, V119
[6]  
CUTURI M., 2013, Advances in neural information processing systems, V2, P2292
[7]  
Fey M., 2019, P ICLR WORKSH REPR L
[8]   node2vec: Scalable Feature Learning for Networks [J].
Grover, Aditya ;
Leskovec, Jure .
KDD'16: PROCEEDINGS OF THE 22ND ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2016, :855-864
[9]  
Guo TY, 2022, AAAI CONF ARTIF INTE, P762
[10]  
Hassani K, 2020, PR MACH LEARN RES, V119