中国科学院机构知识库网格
Chinese Academy of Sciences Institutional Repositories Grid
Graph Contrastive Learning with Adaptive Augmentation

文献类型:会议论文

作者Zhu, Yanqiao3,4; Xu, Yichen1; Yu, Feng2; Liu, Qiang3,4; Wu, Shu3,4; Wang, Liang3,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收割

来源:自动化研究所

浏览0
下载0
收藏0
其他版本

除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。