面向图谱频繁关系模式挖掘的异质图神经网络

A HETEROGENEOUS GRAPH NEURAL NETWORK FOR MINING FREQUENT RELATION PATTERNS OF KNOWLEDGE GRAPH

  • 摘要: 鉴于目前挖掘算法难以对知识图谱建模等问题,提出一种描述和提取节点范围内结构的异质图神经网络模型,旨在挖掘其中的频繁关系模式以及各结构的分布。该模型将关系信息作为节点特征输入,利用自编码机制与多头注意力机制保留原始结构信息,同时引入特征结构平移层将相同结构映射到同一空间中,以获得频繁出现的结构。实验结果表明,该模型可以更快地挖掘图谱关系模式以及各结构在图中的分布;同时在验证特征表达能力的链接预测任务中有稳定表现,在关系类型较多的异质图中甚至优于部分联合学习模型。

     

    Abstract: Due to the difficulties in modeling knowledge graph with current mining algorithms, aimed at mining frequent relation patterns and distribution of each structure, a graph neural network model was designed to describe the heterogeneous structure within the scope of nodes. The model took relations as the input of node features, retained the original structure information by using the autoencoder and multi-head attention mechanism, and designed the translation layer of feature structure to map the same structure to the same space, so as to obtain the frequent heterogeneous structure. Experiments show that this model can mine the relation patterns and the distribution of each structure in the graph faster. In addition, it has a stable performance in the link prediction task, which verifies the feature expression ability, and is even better than some joint learning models in heterogeneous graphs with many relationship types.

     

/

返回文章
返回