ACMF:基于卷积注意力模型的评分预测研究

摘要

评分数据稀疏是影响评分预测的主要因素之一.为了解决数据稀疏问题,一些推荐模型利用辅助信息改善评分预测的准确率.然而大多数推荐模型缺乏对辅助信息的深入理解,因此还有很大的提升空间.鉴于卷积神经网络在特征提取方面和注意力机制在特征选择方面的突出表现,该文提出一种融合卷积注意力神经网络(Atten-tion Convolutional Neural Network,ACNN)的概率矩阵分解模型:基于卷积注意力的矩阵分解(Attention Convo-lutional Model based Matrix Factorization,ACMF),该模型首先使用词嵌入将高维、稀疏的词向量压缩成低维、稠密的特征向量;接着,通过局部注意力层和卷积层学习评论文档的特征;然后,利用用户和物品的潜在模型生成评分预测矩阵;最后计算评分矩阵的均方根误差.在ML-100k、ML-1m、ML-10m、Amazon数据集上的实验结果表明,与当前取得最好预测准确率的PHD模型相比,ACM F模型在预测准确率上分别提高了3.57％ 、1.25％ 、0.37％ 和0.16％.