基于异构注意力网络的知识图嵌入元路径推荐算法

A knowledge graph embedded meta-path recommendation algorithm based on heterogeneous attention networks

  • 摘要: 针对现有模型在使用知识图嵌入方法时未进行任何约束,导致噪声随嵌入渗透到底层信息、基于元路径的推荐算法只采用统一的权重分配策略,未考虑元路径之间的细微差别、传统模型通常存在数据稀疏和冷启动等问题,提出了一种基于异构注意力网络的知识图嵌入元路径推荐算法(MRHAN),通过定义元路径来捕捉不同类型实体和关系之间的复杂语义信息,从而更好地利用丰富的异构信息来缓解数据稀疏和冷启动问题。在使用知识图嵌入方法时采用基于节点相关性的约束方法,将语义相关的高阶实体和关系建模到唯一的元路径中。使用层次化注意力网络来建模节点对不同邻居和不同元路径的偏好,学习节点对不同邻居的权重以及元路径对不同推荐任务的权重。实验结果表明,该模型能够有效学习知识图谱的表示以及节点邻居和元路径对节点的重要性,并进一步缓解数据稀疏和冷启动问题。

     

    Abstract: To address the problems that existing models use knowledge graph embedding methods without any constraint, resulting in noise penetrating into the underlying information with embedding, meta-path-based recommendation algorithms only use a uniform weight assignment strategy without considering the nuances between meta-paths, and traditional models usually suffer from data sparsity and cold start, a knowledge graph embedded meta-path recommendation algorithm based on heterogeneous attention networks (MRHAN) is proposed, which defines meta paths to capture the complex semantic information between different types of entities and relationships, so as to better utilize the rich heterogeneous information to alleviate the data sparsity and cold start problems. A constraint-based approach, utilizing node relevance, is employed to model semantically related higher-order entities and relationships into unique meta-paths using knowledge graph embedding methods. A hierarchical attention network is used to model node preferences for different neighbors and different meta-paths, learning the weights of nodes for different neighbors and the weights of meta-paths for different recommendation tasks. Experimental results show that the model can effectively learn the representation of knowledge graphs, as well as the importance of node neighbors and meta-paths, and further alleviate the data sparsity and cold-start problems.

     

/

返回文章
返回