Multi-granular representation learning framework for Convolutional Graph Neural Networks

Zhang Lei,Qian Feng,Zhao Shu,Chen Jie,Yang Xuejie,Zhang Yanping
DOI: https://doi.org/10.13232/j.cnki.jnju.2023.01.005
2023-01-01
Abstract:Due to its powerful representational capabilities,Convolutional Graph Neural Networks(ConvGNN) have been widely used for network representation learning in social networks,biological networks,and other domains. There is a lack of a framework to increase the performance of ConvGNN,even though multi-granular network representation learning has been demonstrated the improvement for current network embedding approaches. To address this problem,a ConvGNN-based multi-granular network representation learning framework,called M-NRL,is proposed,which is divided into four modules:granulation,training,inference and fusion modules. The task of the granulation module is to construct fine-to-coarse multigranular networks and retain the attribute and label information of the nodes at different granularity networks. The task of the training module is to train any kind of ConvGNN on the coarsest network in an end-to-end approach and optimize its model parameters. The task of the inference module is to use the optimized ConvGNN to reason about the node representations of the networks at different granularity networks. The task of the fusion module is to use attention weights to aggregate node representations of different granularities to produce the final node representation. The semi-supervised node classification task is carried out on four public citation network datasets to verify the effectiveness of M-NRL. Experimental results show that M-NRL accelerates the training of the existing ConvGNN models and enhances the quality of its final representation.
What problem does this paper attempt to address?