McH-HGCN: multi-curvature hyperbolic heterogeneous graph convolutional network with type triplets

Yanxi Liu,Bo Lang
DOI: https://doi.org/10.1007/s00521-023-08473-5
2023-04-06
Neural Computing and Applications
Abstract:Most existing representation learning models for heterogeneous graphs depend on meta-paths, which requires domain-specific prior knowledge and reduces model practicality. In addition, real-world graphs usually conform to power-law distributions, and conventional graph models defined in Euclidean space lead to high distortion for such data. In this paper, we propose a Multi-curvature Hyperbolic Heterogeneous Graph Convolutional Network (McH-HGCN) based on the graph's inherent type triplets. By selecting triplets as data units for message passing and defining the model in hyperbolic space, our model caters to the power-law properties of heterogeneous graphs while avoiding meta-paths dependence. To model the heterogeneity of the graph, we set distinct hyperbolic curvatures as learnable parameters for different types of nodes to obtain the optimal parameterized space mapping after training. Additionally, we introduce a dynamic heterogeneous attention mechanism to compute the attention weights for heterogeneous neighbor aggregation. Node classification and recommendation experiments with several heterogeneous graph datasets show that our model outperforms state-of-the-art methods on multiple datasets, achieving excellent performance without relying on meta-paths.
computer science, artificial intelligence
What problem does this paper attempt to address?