论文标题
迈向以关系为中心的集合和卷积的异质图学习网络
Towards Relation-centered Pooling and Convolution for Heterogeneous Graph Learning Networks
论文作者
论文摘要
异质图神经网络在图表表示学习上发挥了巨大的潜力,并在下游任务(例如节点分类和聚类)上显示出卓越的性能。现有的异质图学习网络主要旨在依靠预定义的元路径,或者使用注意机制在不同的节点/边缘上使用特定类型的专注消息传播,从而产生许多自定义工作和计算成本。为此,我们设计了一个以关系为中心的集合和卷积,用于异质图学习网络,即PC-HGN,以实现特定于关系的采样和交叉融合卷积,可以通过自适应训练过程更好地将图的结构异质性编码到嵌入空间中。我们通过与三个不同现实世界数据集上的最先进的图形学习模型进行比较来评估所提出的模型的性能,结果表明,PC-HGN始终优于所有基线,并最大地提高了17.8%的性能。
Heterogeneous graph neural network has unleashed great potential on graph representation learning and shown superior performance on downstream tasks such as node classification and clustering. Existing heterogeneous graph learning networks are primarily designed to either rely on pre-defined meta-paths or use attention mechanisms for type-specific attentive message propagation on different nodes/edges, incurring many customization efforts and computational costs. To this end, we design a relation-centered Pooling and Convolution for Heterogeneous Graph learning Network, namely PC-HGN, to enable relation-specific sampling and cross-relation convolutions, from which the structural heterogeneity of the graph can be better encoded into the embedding space through the adaptive training process. We evaluate the performance of the proposed model by comparing with state-of-the-art graph learning models on three different real-world datasets, and the results show that PC-HGN consistently outperforms all the baseline and improves the performance maximumly up by 17.8%.