From Canonical Correlation Analysis to Self-supervised Graph Neural NetworksDownload PDF

May 21, 2021 (edited Jan 12, 2022)NeurIPS 2021 PosterReaders: Everyone
  • Keywords: Graph Neural Network, Self-supervised Learning, Canonical Correlation Analysis
  • TL;DR: Propose a Canonical Correlation Analysis-based method for self-supervised representation learning on graphs
  • Abstract: We introduce a conceptually simple yet effective model for self-supervised representation learning with graph data. It follows the previous methods that generate two views of an input graph through data augmentation. However, unlike contrastive methods that focus on instance-level discrimination, we optimize an innovative feature-level objective inspired by classical Canonical Correlation Analysis. Compared with other works, our approach requires none of the parameterized mutual information estimator, additional projector, asymmetric structures, and most importantly, negative samples which can be costly. We show that the new objective essentially 1) aims at discarding augmentation-variant information by learning invariant representations, and 2) can prevent degenerated solutions by decorrelating features in different dimensions. Our theoretical analysis further provides an understanding for the new objective which can be equivalently seen as an instantiation of the Information Bottleneck Principle under the self-supervised setting. Despite its simplicity, our method performs competitively on seven public graph datasets.
  • Supplementary Material: pdf
  • Code Of Conduct: I certify that all co-authors of this work have read and commit to adhering to the NeurIPS Statement on Ethics, Fairness, Inclusivity, and Code of Conduct.
  • Code: https://github.com/hengruizhang98/CCA-SSG
13 Replies

Loading