Few-Shot Node Classification on Attributed Networks Based on Prototypical Network

Yaliang Zhao, Guangming Zhang, Jinke Wang

Abstract: In today’s society, the interactions between various entities form complicated attributed networks. The node classification task as the core task of analyzing attribute networks has gained a considerable amount of attention from researchers. The existing graph neural network-based node classification approaches all demand lots of labeled data, but a high percentage of node classes in practical scenarios contain only restricted labeled data. So their classification effect is not good. In response to this problem, we put forward an improved graph prototypical network (IGPN) to enhance the few-shot node classification’s accuracy. In particular, we start by obtaining the embedding representations of nodes on the attributed network through a graph attention network. Then combined with the node’s degree information and neighbor information, the importance coefficient of all nodes in the support set is learned. After that, we can obtain the prototype representation of every class. Finally, the node classification can be completed by the distance of the query nodes’ representation from all class prototypes. Our method is preferable to the baseline method on three datasets according to a large number of experiments.