GPS: Graph Contrastive Learning via Multi-scale Augmented Views from Adversarial Pooling

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Ju, Wei, Gu, Yiyang, Mao, Zhengyang, Qiao, Ziyue, Qin, Yifang, Luo, Xiao, Xiong, Hui, Zhang, Ming
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914656464928768
author Ju, Wei
Gu, Yiyang
Mao, Zhengyang
Qiao, Ziyue
Qin, Yifang
Luo, Xiao
Xiong, Hui
Zhang, Ming
author_facet Ju, Wei
Gu, Yiyang
Mao, Zhengyang
Qiao, Ziyue
Qin, Yifang
Luo, Xiao
Xiong, Hui
Zhang, Ming
contents Self-supervised graph representation learning has recently shown considerable promise in a range of fields, including bioinformatics and social networks. A large number of graph contrastive learning approaches have shown promising performance for representation learning on graphs, which train models by maximizing agreement between original graphs and their augmented views (i.e., positive views). Unfortunately, these methods usually involve pre-defined augmentation strategies based on the knowledge of human experts. Moreover, these strategies may fail to generate challenging positive views to provide sufficient supervision signals. In this paper, we present a novel approach named Graph Pooling ContraSt (GPS) to address these issues. Motivated by the fact that graph pooling can adaptively coarsen the graph with the removal of redundancy, we rethink graph pooling and leverage it to automatically generate multi-scale positive views with varying emphasis on providing challenging positives and preserving semantics, i.e., strongly-augmented view and weakly-augmented view. Then, we incorporate both views into a joint contrastive learning framework with similarity learning and consistency learning, where our pooling module is adversarially trained with respect to the encoder for adversarial robustness. Experiments on twelve datasets on both graph classification and transfer learning tasks verify the superiority of the proposed method over its counterparts.
format Preprint
id arxiv_https___arxiv_org_abs_2401_16011
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle GPS: Graph Contrastive Learning via Multi-scale Augmented Views from Adversarial Pooling
Ju, Wei
Gu, Yiyang
Mao, Zhengyang
Qiao, Ziyue
Qin, Yifang
Luo, Xiao
Xiong, Hui
Zhang, Ming
Machine Learning
Artificial Intelligence
Social and Information Networks
Self-supervised graph representation learning has recently shown considerable promise in a range of fields, including bioinformatics and social networks. A large number of graph contrastive learning approaches have shown promising performance for representation learning on graphs, which train models by maximizing agreement between original graphs and their augmented views (i.e., positive views). Unfortunately, these methods usually involve pre-defined augmentation strategies based on the knowledge of human experts. Moreover, these strategies may fail to generate challenging positive views to provide sufficient supervision signals. In this paper, we present a novel approach named Graph Pooling ContraSt (GPS) to address these issues. Motivated by the fact that graph pooling can adaptively coarsen the graph with the removal of redundancy, we rethink graph pooling and leverage it to automatically generate multi-scale positive views with varying emphasis on providing challenging positives and preserving semantics, i.e., strongly-augmented view and weakly-augmented view. Then, we incorporate both views into a joint contrastive learning framework with similarity learning and consistency learning, where our pooling module is adversarially trained with respect to the encoder for adversarial robustness. Experiments on twelve datasets on both graph classification and transfer learning tasks verify the superiority of the proposed method over its counterparts.
title GPS: Graph Contrastive Learning via Multi-scale Augmented Views from Adversarial Pooling
topic Machine Learning
Artificial Intelligence
Social and Information Networks
url https://arxiv.org/abs/2401.16011