Coarse-to-Fine Contrastive Learning on Graphs

Peiyao Zhao, Yuangang Pan, Xin Li*, Xu Chen, Ivor W. Tsang, Lejian Liao

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

2 Citations (Scopus)

Abstract

Inspired by the impressive success of contrastive learning (CL), a variety of graph augmentation strategies have been employed to learn node representations in a self-supervised manner. Existing methods construct the contrastive samples by adding perturbations to the graph structure or node attributes. Although impressive results are achieved, it is rather blind to the wealth of prior information assumed: with the increase of the perturbation degree applied on the original graph: 1) the similarity between the original graph and the generated augmented graph gradually decreases and 2) the discrimination between all nodes within each augmented view gradually increases. In this article, we argue that both such prior information can be incorporated (differently) into the CL paradigm following our general ranking framework. In particular, we first interpret CL as a special case of learning to rank (L2R), which inspires us to leverage the ranking order among positive augmented views. Meanwhile, we introduce a self-ranking paradigm to ensure that the discriminative information among different nodes can be maintained and also be less altered to the perturbations of different degrees. Experiment results on various benchmark datasets verify the effectiveness of our algorithm compared with the supervised and unsupervised models.

Original languageEnglish
Pages (from-to)4622-4634
Number of pages13
JournalIEEE Transactions on Neural Networks and Learning Systems
Volume35
Issue number4
DOIs
Publication statusPublished - 1 Apr 2024

Keywords

  • Contrastive learning (CL)
  • graph representation learning
  • learning to rank (L2R)
  • node representation
  • self-supervised learning (SSL)

Fingerprint

Dive into the research topics of 'Coarse-to-Fine Contrastive Learning on Graphs'. Together they form a unique fingerprint.

Cite this