arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.11147cs.LG
收藏

Hop-Hop Relation-aware Graph Neural Networks

University of Sheffield(谢菲尔德大学)

Li Zhang, Yan Ge, Haiping Lu

英文摘要:

Graph Neural Networks (GNNs) are widely used in graph representation learning. However, most GNN methods are designed for either homogeneous or heterogeneous graphs. In this paper, we propose a new model, Hop-Hop Relation-aware Graph Neural Network (HHR-GNN), to unify representation learning for these two types of graphs. HHR-GNN learns a personalized receptive field for each node by leveraging knowledge graph embedding to learn relation scores between the central node's representations at different hops. In neighborhood aggregation, our model simultaneously allows for hop-aware projection and aggregation. This mechanism enables the central node to learn a hop-wise neighborhood mixing that can be applied to both homogeneous and heterogeneous graphs. Experimental results on five benchmarks show the competitive performance of our model compared to state-of-the-art GNNs, e.g., up to 13K faster in terms of time cost per training epoch on large heterogeneous graphs.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑