Data Science and Engineering (Jun 2024)

Multi-view Heterogeneous Graph Neural Networks for Node Classification

  • Xi Zeng,
  • Fang-Yuan Lei,
  • Chang-Dong Wang,
  • Qing-Yun Dai

DOI
https://doi.org/10.1007/s41019-024-00253-y
Journal volume & issue
Vol. 9, no. 3
pp. 294 – 308

Abstract

Read online

Abstract Recently, with graph neural networks (GNNs) becoming a powerful technique for graph representation, many excellent GNN-based models have been proposed for processing heterogeneous graphs, which are termed Heterogeneous graph neural networks (HGNNs). However, existing HGNNs tend to aggregate information from either direct neighbors or those connected by short metapaths, thereby neglecting the higher-order information and global feature similarity information in heterogeneous graphs. In this paper, we propose a Multi-View Heterogeneous graph neural network (MV-HGNN) to aggregate these information. Firstly, two auxiliary views, specifically a global feature similarity view and a graph diffusion view, are generated from the original heterogeneous graph. Secondly, MV-HGNN performs two message-passing strategies to get the representation of different views. Subsequently, a transformer-based aggregator is used to get the semantic information. Subsequently, the representations of the three views are fused into a final composite representation. We evaluate our method on the node classification task over three commonly used heterogeneous graph datasets, and the results demonstrate that our proposed MV-HGNN significantly outperforms state-of-the-art baselines.

Keywords