• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science ›› 2021, Vol. 43 ›› Issue (02): 280-287.

Previous Articles     Next Articles

Multi-label node classification based on generative adversarial network

CHEN Wen-qi1,WANG Ying1,3,WANG Xin2,3,WANG Hong-ji2   

  1. (1.School of Computer Science and Technology,Jilin University,Changchun 130012;

    2.School of Artificial Intelligence,Jilin University,Changchun 130012;

    3.Key Laboratory of Symbol Computation and Knowledge Engineering,
    Ministry of Education,Jilin University,Changchun 130012,China)
  • Received:2020-08-05 Revised:2020-11-12 Accepted:2021-02-25 Online:2021-02-25 Published:2021-02-23

Abstract: Node classification is widely used in social network and other network data. In order to study node classification, generative adversarial network (GAN) is used to obtain node representation, so as to obtain a good node classification effect. On this basis, a node classification-generative adversa- rial network (NC-GAN) model is proposed. This model uses GAN to conduct a binary game, considers the connectivity distribution in the network and the similarity between nodes to obtain the node representation that better fits the network, and then classifies the node representation to obtain a good classification effect. In order to verify the effect, the proposal is compared with DeepWalk, GraphGAN and other node representation model and graph convolutional network model  in terms of link prediction and node classification. The model is only weaker than the GraphGAN model in link prediction, but it is better than other models in node classification.




Key words: generative adversarial network, multi-label, node classification