arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2212.10046cs.IR
收藏

基于知识图谱推荐的因果推理

Causal Inference for Knowledge Graph based Recommendation

Yinwei Wei, Xiang Wang, Liqiang Nie, Shaoyu Li, Dingxian Wang, Tat-Seng Chua

摘要:

知识图谱作为一种辅助信息,被用来补充基于协同过滤的推荐模型。以往的研究大多是通过将项目与知识库中的实体进行映射,从知识库中提取知识信息,并将其注入到用户和项目的表示中。尽管他们的显着的性能,他们未能在KG中的属性建模的用户偏好,因为他们忽略了(1)KG的结构信息可能会阻碍用户偏好学习,(2)用户的交互属性将导致在相似性得分的偏差问题。在因果关系工具的帮助下,我们构建了基于知识生成的推荐中变量之间的因果关系,并找出了导致上述挑战的原因。因此,我们开发了一个新的框架,称为知识图谱的因果推荐(KGCR),它实现了去基础的用户偏好学习,并采用反事实推理,以消除偏见的相似性评分。最终,我们在三个数据集上评估了我们提出的模型,包括Amazon-book,LastFM和Yelp 2018数据集。通过对数据集进行广泛的实验,我们证明了KGCR优于几个最先进的基线,如KGNN-LS,KGAT和KGIN。

英文摘要:

Knowledge Graph (KG), as a side-information, tends to be utilized to supplement the collaborative filtering (CF) based recommendation model. By mapping items with the entities in KGs, prior studies mostly extract the knowledge information from the KGs and inject it into the representations of users and items. Despite their remarkable performance, they fail to model the user preference on attributes in the KG, since they ignore that (1) the structure information of KG may hinder the user preference learning, and (2) the user's interacted attributes will result in the bias issue on the similarity scores. With the help of causality tools, we construct the causal-effect relation between the variables in KG-based recommendation and identify the reasons causing the mentioned challenges. Accordingly, we develop a new framework, termed Knowledge Graph-based Causal Recommendation (KGCR), which implements the deconfounded user preference learning and adopts counterfactual inference to eliminate bias in the similarity scoring. Ultimately, we evaluate our proposed model on three datasets, including Amazon-book, LastFM, and Yelp2018 datasets. By conducting extensive experiments on the datasets, we demonstrate that KGCR outperforms several state-of-the-art baselines, such as KGNN-LS, KGAT and KGIN.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑