arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2006.10233cs.IRcs.LG
收藏

一种基于属性级协同注意的知识增强推荐模型

A Knowledge-Enhanced Recommendation Model with Attribute-Level Co-Attention

Fudan University(复旦大学)

Deqing Yang, Zengcun Song, Lvxin Xue, Yanghua Xiao

摘要:

深度神经网络(DNN)已被广泛应用于推荐系统中,包括结合注意力机制以提高性能。然而,现有的基于注意力的推荐模型大多只对用户进行项目级的注意,限制了推荐性能的进一步提高。在本文中,我们提出了一个知识增强的推荐模型ACAM,它结合了从知识图谱(KG)提取的项目属性作为边信息,并建立了一个共同的注意力机制的属性级别,以实现性能增益。具体地说,ACAM中的每个用户和项目首先由一组属性嵌入来表示。然后,用户表示和项目表示的同时,通过捕获不同属性之间的相关性,由一个共同注意模块。我们在两个真实数据集上的广泛实验表明,通过属性级共同注意增强的用户表示和项目表示获得了ACAM优于最先进的深度模型的优势。

英文摘要:

Deep neural networks (DNNs) have been widely employed in recommender systems including incorporating attention mechanism for performance improvement. However, most of existing attention-based models only apply item-level attention on user side, restricting the further enhancement of recommendation performance. In this paper, we propose a knowledge-enhanced recommendation model ACAM, which incorporates item attributes distilled from knowledge graphs (KGs) as side information, and is built with a co-attention mechanism on attribute-level to achieve performance gains. Specifically, each user and item in ACAM are represented by a set of attribute embeddings at first. Then, user representations and item representations are augmented simultaneously through capturing the correlations between different attributes by a co-attention module. Our extensive experiments over two realistic datasets show that the user representations and item representations augmented by attribute-level co-attention gain ACAM's superiority over the state-of-the-art deep models.

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

↑