arXiv:2409.15315cs.LGcs.AI2024-09被引 10

用知识图谱和注意力机制提升推荐系统性能

An Efficient Recommendation Model Based on Knowledge Graph Attention-Assisted Network (KGATAX)

  • 引入知识图谱与注意力机制,显式捕捉高阶连接关系
  • 通过多层交互传播聚合信息,增强模型泛化能力
  • 用全息嵌入融合辅助信息,更好利用实体关联数据

推荐系统在帮助用户从海量信息中筛选内容方面起着关键作用。然而,传统推荐算法往往忽视多源信息的整合与利用,限制了系统性能。为此,本文提出一种新型推荐模型——知识图谱注意力辅助网络(KGAT-AX)。首先将知识图谱引入推荐模型,通过注意力机制更显式地探索高阶连接关系。利用多层交互信息传播,模型聚合信息以增强泛化能力。此外,通过全息嵌入将辅助信息融入实体,学习邻近实体间的推断关系,实现对实体关联辅助信息的有效利用。我们在真实数据集上进行了实验,验证了KGAT-AX模型的合理性与有效性。实验分析表明,相较于其他基线模型,KGAT-AX在公开数据集上展现出更优的知识信息捕获与关系学习能力。

原文摘要 · Abstract (English)

Recommendation systems play a crucial role in helping users filter through vast amounts of information. However, traditional recommendation algorithms often overlook the integration and utilization of multi-source information, limiting system performance. Therefore, this study proposes a novel recommendation model, Knowledge Graph Attention-assisted Network (KGAT-AX). We first incorporate the knowledge graph into the recommendation model, introducing an attention mechanism to explore higher order connectivity more explicitly. By using multilayer interactive information propagation, the model aggregates information to enhance its generalization ability. Furthermore, we integrate auxiliary information into entities through holographic embeddings, aggregating the information of adjacent entities for each entity by learning their inferential relationships. This allows for better utilization of auxiliary information associated with entities. We conducted experiments on real datasets to demonstrate the rationality and effectiveness of the KGAT-AX model. Through experimental analysis, we observed the effectiveness and potential of KGAT-AX compared to other baseline models on public datasets. KGAT-AX demonstrates better knowledge information capture and relationship learning capabilities.

推荐系统知识图谱注意力机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。