Label Contrastive Coding based Graph Neural Network for Graph Classification
arXiv:2101.05486
Abstract
Graph classification is a critical research problem in many applications from different domains. In order to learn a graph classification model, the most widely used supervision component is an output layer together with classification loss (e.g.,cross-entropy loss together with softmax or margin loss). In fact, the discriminative information among instances are more fine-grained, which can benefit graph classification tasks. In this paper, we propose the novel Label Contrastive Coding based Graph Neural Network (LCGNN) to utilize label information more effectively and comprehensively. LCGNN still uses the classification loss to ensure the discriminability of classes. Meanwhile, LCGNN leverages the proposed Label Contrastive Loss derived from self-supervised learning to encourage instance-level intra-class compactness and inter-class separability. To power the contrastive learning, LCGNN introduces a dynamic label memory bank and a momentum updated encoder. Our extensive evaluations with eight benchmark graph datasets demonstrate that LCGNN can outperform state-of-the-art graph classification models. Experimental results also verify that LCGNN can achieve competitive performance with less training data because LCGNN exploits label information comprehensively.
arXiv commentAccept by DASFAA'21
References in corpus (9)
- A Simple Framework for Contrastive Learning of Visual Representations
- graph2vec: Learning Distributed Representations of Graphs
- Deep Graph Contrastive Representation Learning
- Unsupervised Feature Learning via Non-Parametric Instance-level Discrimination
- InfoGraph: Unsupervised and Semi-supervised Graph-Level Representation Learning via Mutual Information Maximization
- Hierarchical Graph Pooling with Structure Learning
- The Multiscale Laplacian Graph Kernel
- Heterogeneous Deep Graph Infomax
- EnsemFDet: An Ensemble Approach to Fraud Detection based on Bipartite Graph