False Negative Sample Detection for Graph Contrastive Learning

被引:4
作者
Zhang, Binbin [1 ]
Wang, Li [1 ]
机构
[1] Taiyuan Univ Technol, Coll Data Sci, Jinzhong 030600, Peoples R China
来源
TSINGHUA SCIENCE AND TECHNOLOGY | 2024年 / 29卷 / 02期
基金
中国国家自然科学基金;
关键词
graph representation learning; contrastive learning; false negative sample detection;
D O I
10.26599/TST.2023.9010043
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Recently, self-supervised learning has shown great potential in Graph Neural Networks (GNNs) through contrastive learning, which aims to learn discriminative features for each node without label information. The key to graph contrastive learning is data augmentation. The anchor node regards its augmented samples as positive samples, and the rest of the samples are regarded as negative samples, some of which may be positive samples. We call these mislabeled samples as "false negative" samples, which will seriously affect the final learning effect. Since such semantically similar samples are ubiquitous in the graph, the problem of false negative samples is very significant. To address this issue, the paper proposes a novel model, False negative sample Detection for Graph Contrastive Learning (FD4GCL), which uses attribute and structure-aware to detect false negative samples. Experimental results on seven datasets show that FD4GCL outperforms the state-of-the-art baselines and even exceeds several supervised methods.
引用
收藏
页码:529 / 542
页数:14
相关论文
共 35 条
[1]  
[Anonymous], 2008, P 25 INT C MACH LEAR
[2]  
Bachman P, 2019, ADV NEUR IN, V32
[3]   Unsupervised Pre-Training of Image Features on Non-Curated Data [J].
Caron, Mathilde ;
Bojanowski, Piotr ;
Mairal, Julien ;
Joulin, Armand .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :2959-2968
[4]   Deep Clustering for Unsupervised Learning of Visual Features [J].
Caron, Mathilde ;
Bojanowski, Piotr ;
Joulin, Armand ;
Douze, Matthijs .
COMPUTER VISION - ECCV 2018, PT XIV, 2018, 11218 :139-156
[5]  
Chen Tianlong, 2022, ICLR
[6]  
Chuang C. Y., 2020, 34 C NEUR INF PROC S
[7]  
Errica F., 2020, 8 INT C LEARN REPR A
[8]  
Gilmer J, 2017, PR MACH LEARN RES, V70
[9]  
Gong XM, 2022, Arxiv, DOI arXiv:2212.07035
[10]  
Hassani K, 2020, PR MACH LEARN RES, V119