计算机应用研究2021,Vol.38Issue(10) :3055-3059.DOI:10.19734/j.issn.1001-3695.2021.03.0088

基于多头注意力机制和位置信息的xDeepFM推荐模型

xDeepFM recommendation model based on multi-head attention mechanism and location information

牛路帅 彭龑
计算机应用研究2021,Vol.38Issue(10) :3055-3059.DOI:10.19734/j.issn.1001-3695.2021.03.0088

基于多头注意力机制和位置信息的xDeepFM推荐模型

xDeepFM recommendation model based on multi-head attention mechanism and location information

牛路帅 1彭龑1
扫码查看

作者信息

  • 1. 四川轻化工大学 自动化与信息工程学院,四川 宜宾644000
  • 折叠

摘要

为了解决推荐模型中无法挖掘用户兴趣多样性和捕捉用户行为序列之间的顺序信息,以及交互发生在元素级并非特征向量之间等问题,提出一种基于多头注意力机制和位置信息的xDeepFM推荐模型(extreme deep multiple attention and location information factorization machine,xDMALFM).首先通过多头注意力机制进行不同子空间的特征深度提取,然后利用位置信息去捕捉用户行为序列之间的顺序关系.最后,利用三个公开数据集进行对比实验,以AUC指标进行评估.实验结果表明所提算法相比xDeepFM模型具有更好的推荐性能,验证了其有效性与可行性.

关键词

推荐算法/深度学习/位置信息/多头注意力机制/xDeepFM

引用本文复制引用

基金项目

出版年

2021
计算机应用研究
四川省电子计算机应用研究中心

计算机应用研究

CSTPCDCSCD北大核心
影响因子:0.93
ISSN:1001-3695
被引量2
参考文献量3
段落导航相关论文