Co-augmentation of structure and feature for boosting graph contrastive learning

No Thumbnail Available
File version
Author(s)
Bao, Peng
Yan, Rong
Pan, Shirui
Griffith University Author(s)
Primary Supervisor
Other Supervisors
Editor(s)
Date
2024
Size
File type(s)
Location
License
Abstract

Graph Contrastive Learning (GCL) learns invariant representation by maximizing the consistency between different augmented graphs that share the same semantics. However, the performance of existing GCL methods is inseparable from varied manually designed augmentation techniques that randomly perturb edges/nodes/features, which unexpectedly change the semantic similarity and lead to biased structure and feature augmentation. In this paper, we propose a co-augmentation strategy for STRucture and FEature (STRFE) to eliminate augmentation bias. Specifically, we construct a relatively unbiased augmented graph by amplifying or suppressing graph frequency in the spectral domain, which promises structural consistency and feature diversity between the augmented and original graph. Moreover, we investigate external and internal contrastive loss to balance the consistency and diversity between the original and augmented graph, which facilitates preserving semantic similarity and encourages relatively unbiased structure and feature augmentation to enhance the performance of GCL. Theoretical analysis proves why our proposed structure and feature co-augmentation strategy can perform well. Extensive experiments show that STRFE achieves competitive results in three real-world datasets on different downstream tasks compared with more than ten benchmarks.

Journal Title

Information Sciences

Conference Title
Book Title
Edition
Volume

676

Issue
Thesis Type
Degree Program
School
Publisher link
Patent number
Funder(s)
Grant identifier(s)
Rights Statement
Rights Statement
Item Access Status
Note
Access the data
Related item(s)
Subject
Persistent link to this record
Citation

Bao, P; Yan, R; Pan, S, Co-augmentation of structure and feature for boosting graph contrastive learning, Information Sciences, 2024, 676, pp. 120792

Collections