论文标题
与异质图的元路径自学辅助学习
Self-supervised Auxiliary Learning with Meta-paths for Heterogeneous Graphs
论文作者
论文摘要
图神经网络在广泛的应用程序中显示出卓越的性能,提供了图形结构化数据的强大表示。最近的作品表明,辅助任务可以进一步改善表示形式。但是,在文献中,具有各种类型的节点和边缘的辅助任务,其中包含具有各种节点和边缘的丰富语义信息。在本文中,为了学习异质图上的图形神经网络,我们提出了一种使用元路径的新型自我监督辅助学习方法,这是多种边缘类型的复合关系。我们提出的方法是通过预测元路径作为辅助任务来学习学习主要任务。这可以看作是一种元学习。提出的方法可以确定辅助任务的有效组合,并自动平衡它们以改善主要任务。我们的方法可以以插件方式应用于任何图形神经网络,而无需手动标记或其他数据。该实验表明,所提出的方法始终提高了链路预测的性能和异质图上的节点分类的性能。
Graph neural networks have shown superior performance in a wide range of applications providing a powerful representation of graph-structured data. Recent works show that the representation can be further improved by auxiliary tasks. However, the auxiliary tasks for heterogeneous graphs, which contain rich semantic information with various types of nodes and edges, have less explored in the literature. In this paper, to learn graph neural networks on heterogeneous graphs we propose a novel self-supervised auxiliary learning method using meta-paths, which are composite relations of multiple edge types. Our proposed method is learning to learn a primary task by predicting meta-paths as auxiliary tasks. This can be viewed as a type of meta-learning. The proposed method can identify an effective combination of auxiliary tasks and automatically balance them to improve the primary task. Our methods can be applied to any graph neural networks in a plug-in manner without manual labeling or additional data. The experiments demonstrate that the proposed method consistently improves the performance of link prediction and node classification on heterogeneous graphs.