Graph Contrastive Learning Meets Graph Meta Learning: A Unified Method for Few-shot Node Tasks

Hao Liu, Jiarui Feng, Lecheng Kong, Dacheng Tao, Yixin Chen +1 more
9/19/2023
cs.LGcs.AI

Official code: Haoliu-cola/COLA (12 stars, last commit 2024-08-22).

Abstract

Graph Neural Networks (GNNs) have become popular in Graph Representation Learning (GRL). One fundamental application is few-shot node classification. Most existing methods follow the meta learning paradigm, showing the ability of fast generalization to few-shot tasks. However, recent works indicate that graph contrastive learning combined with fine-tuning can significantly outperform meta learning methods. Despite the empirical success, there is limited understanding of the reasons behind it. In our study, we first identify two crucial advantages of contrastive learning compared to meta learning, including (1) the comprehensive utilization of graph nodes and (2) the power of graph augmentations. To integrate the strength of both contrastive learning and meta learning on the few-shot node classification tasks, we introduce a new paradigm: Contrastive Few-Shot Node Classification (COLA). Specifically, COLA employs graph augmentations to identify semantically similar nodes, which enables the construction of meta-tasks without the need for label information. Therefore, COLA can utilize all nodes to construct meta-tasks, further reducing the risk of overfitting. Through extensive experiments, we validate the essentiality of each component in our design and demonstrate that COLA achieves new state-of-the-art on all tasks.

View on arXivView PDF

Code Implementations(1)

Haoliu-cola/COLAOfficial67%

This repository is the implementation of the model COLA from paper: Graph Contrastive Learning Meets Graph Meta Learning: A Unified Method for Few-shot Node Tasks.

120PythonMay 24, 20232 years ago

Similar Papers

Cite this paper

@article{liu2023graph,
  title  = {Graph Contrastive Learning Meets Graph Meta Learning: A Unified Method for Few-shot Node Tasks},
  author = {Hao Liu and Jiarui Feng and Lecheng Kong and Dacheng Tao and Yixin Chen and Muhan Zhang},
  year   = {2023},
  eprint = {2309.10376},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2309.10376v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2309.10376.svg)](https://codeofpaper.com/paper/2309.10376)

Discussion