Supervised Graph Contrastive Learning for Few-Shot Node Classification
Abstract
Graphs are present in many real-world applications, such as financial fraud detection, commercial recommendation, and social network analysis. But given the high cost of graph annotation or labeling, we face a severe graph label-scarcity problem, i.e., a graph might have a few labeled nodes. One example of such a problem is the so-called \textit{few-shot node classification}. A predominant approach to this problem resorts to \textit{episodic meta-learning}. In this work, we challenge the status quo by asking a fundamental question whether meta-learning is a must for few-shot node classification tasks. We propose a new and simple framework under the standard few-shot node classification setting as an alternative to meta-learning to learn an effective graph encoder. The framework consists of supervised graph contrastive learning with novel mechanisms for data augmentation, subgraph encoding, and multi-scale contrast on graphs. Extensive experiments on three benchmark datasets (CoraFull, Reddit, Ogbn) show that the new framework significantly outperforms state-of-the-art meta-learning based methods.
Cite
Text
Tan et al. "Supervised Graph Contrastive Learning for Few-Shot Node Classification." European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases, 2022. doi:10.1007/978-3-031-26390-3_24Markdown
[Tan et al. "Supervised Graph Contrastive Learning for Few-Shot Node Classification." European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases, 2022.](https://mlanthology.org/ecmlpkdd/2022/tan2022ecmlpkdd-supervised/) doi:10.1007/978-3-031-26390-3_24BibTeX
@inproceedings{tan2022ecmlpkdd-supervised,
title = {{Supervised Graph Contrastive Learning for Few-Shot Node Classification}},
author = {Tan, Zhen and Ding, Kaize and Guo, Ruocheng and Liu, Huan},
booktitle = {European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases},
year = {2022},
pages = {394-411},
doi = {10.1007/978-3-031-26390-3_24},
url = {https://mlanthology.org/ecmlpkdd/2022/tan2022ecmlpkdd-supervised/}
}