Individuality-enhanced and multi-granularity consistency-preserving graph neural network for semi-supervised node classification

Abstract

Abstract Semi-supervised node classification is an important task that aims at classifying nodes based on the graph structure, node features, and class labels for a subset of nodes. While most graph convolutional networks (GCNs) perform well when an ample number of labeled nodes are available, they often degenerate when the amount of labeled data is limited. To address this problem, we propose a scheme, namely, Individuality-enhanced and Multi-granularity Consistency-preserving graph neural Network (IMCN), which can alleviate the problem of losing individual information within the encoder while providing a reliable supervised signal for learning purposes. First, one simple encoder based on node features only is integrated to enhance node individuality and amend node commonality learned by the GCN-based encoder. Then, three constraints are defined at different levels of granularity, encompassing node embedding agreement, semantic class alignment, and node-to-class distribution identity. They can maintain the consistency between the individuality and commonality of nodes and be leveraged as latent supervised signals for learning representative embeddings. Finally, the trade-off between the individuality and commonality of nodes captured by two encoders is taken into consideration for node classification. Extensive experiments on six real-world datasets have been conducted to validate the superiority of IMCN against state-of-the-art baselines in handling node classification tasks with scarce labeled data. Graphical abstract