Graph Contrastive Learning with Adaptive Augmentation
文献类型:会议论文
作者 | Zhu, Yanqiao3,4![]() ![]() ![]() ![]() ![]() |
出版日期 | 2021-04 |
会议日期 | 2021-4 |
会议地点 | Online |
页码 | 2069-2080 |
英文摘要 | Recently, contrastive learning (CL) has emerged as a successful method for unsupervised graph representation learning. Most graph CL methods first perform stochastic augmentation on the input graph to obtain two graph views and maximize the agreement of representations in the two views. Despite the prosperous development of graph CL methods, the design of graph augmentation schemes---a crucial component in CL---remains rarely explored. We argue that the data augmentation schemes should preserve intrinsic structural and attribute information of graphs, which will force the model to learn representations that are insensitive to perturbation on unimportant nodes and edges. However, most existing methods adopt uniform data augmentation schemes, like uniformly dropping edges and uniformly shuffling features, leading to suboptimal performance. In this paper, we propose a novel graph contrastive representation learning method with adaptive augmentation that incorporates various priors for topological and semantic aspects of the graph. Specifically, on the topology level, we design augmentation schemes based on node centrality measures to highlight important connective structures. On the node attribute level, we corrupt node features by adding more noise to unimportant node features, to enforce the model to recognize underlying semantic information. We perform extensive experiments of node classification on a variety of real-world datasets. Experimental results demonstrate that our proposed method consistently outperforms existing state-of-the-art methods and even surpasses some supervised counterparts, which validates the effectiveness of the proposed contrastive framework with adaptive augmentation. |
会议录出版者 | ACM Press |
语种 | 英语 |
源URL | [http://ir.ia.ac.cn/handle/173211/48470] ![]() |
专题 | 自动化研究所_智能感知与计算研究中心 |
通讯作者 | Wu, Shu |
作者单位 | 1.School of Computer Science, Beijing University of Posts and Telecommunications 2.Alibaba Group 3.School of Artificial Intelligence, University of Chinese Academy of Sciences 4.Center for Research on Intelligent Perception and Computing, Institute of Automation, Chinese Academy of Sciences |
推荐引用方式 GB/T 7714 | Zhu, Yanqiao,Xu, Yichen,Yu, Feng,et al. Graph Contrastive Learning with Adaptive Augmentation[C]. 见:. Online. 2021-4. |
入库方式: OAI收割
来源:自动化研究所
其他版本
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。