基于注意力机制的项目相似性推荐模型

作者：沈冬东; 汪海涛^*; 姜瑛; 陈星

来源：电子测量技术, 2019, 42(15): 150-154.

DOI：10.19651/j.cnki.emt.1902734

• 协同过滤
• 神经推荐模型
• 项目嵌入
• 注意力网络

传统基于项目的协同过滤算法具有优秀的可解释性以及实时性,但是存在推荐精确度不高、难以充分挖掘数据间隐含信息等问题,该模型引入注意力机制来改善这些问题。该模型分为两个主要的部分:神经项目嵌入、改进的注意力神经网络学习。首先采用item2vec项目嵌入技术生成项目的统一向量表示;其次使用注意力机制学习用户历史交互项目对当前偏好预测的重要性,得到用户的偏好进行推荐。通过在公共数据集MovieLens和Pinterest上的广泛实验,与传统的推荐算法以及同类的神经推荐模型进行对比。结果表明改进后的模型在推荐准确性上有明显的提升,最高可提升6.04%。该模型在保证推荐算法的可解释性与实时性的同时提高了推荐算法的精确性。

• 单位
  自动化学院; 昆明理工大学