首页 | 官方网站   微博 | 高级检索  
     

融合注意力LSTM的神经张量分解推荐模型
引用本文:李晶晶,夏鸿斌,刘渊. 融合注意力LSTM的神经张量分解推荐模型[J]. 中文信息学报, 2021, 35(5): 91-100
作者姓名:李晶晶  夏鸿斌  刘渊
作者单位:1.江南大学 人工智能与计算机学院,江苏 无锡 214122;
2.江苏省媒体设计与软件技术重点实验室,江苏 无锡 214122
基金项目:国家科学支撑计划(2015BAH54F01);国家自然科学基金(61672264)
摘    要:针对结合深度学习模型的协同过滤算法未考虑关联数据的多维交互随时间动态变化的问题,该文提出一种融合时间交互学习和注意力长短期记忆网络的张量分解推荐模型(LA-NTF).通过采用基于注意力机制的长短期记忆网络从项目文本信息中提取项目的潜在向量,然后使用融合注意力机制的长短期记忆网络来表征用户—项目关系数据在时间上的多维交互...

关 键 词:注意力机制  长短期记忆网络  时间交互学习  推荐系统  张量分解
收稿时间:2019-12-20

Neural Tensor Factorization Recommendation Model Based on Attention LSTM
LI Jingjing,XIA Hongbin,LIU Yuan. Neural Tensor Factorization Recommendation Model Based on Attention LSTM[J]. Journal of Chinese Information Processing, 2021, 35(5): 91-100
Authors:LI Jingjing  XIA Hongbin  LIU Yuan
Affiliation:1.School of Artificial Intelligence and Computer Science, Jiangnan University, Wuxi, Jiangsu 214122, China;2.Jiangsu Key Laboratory of Media Design and Software Technology, Wuxi, Jiangsu 214122, China
Abstract:Current collaborative filtering algorithms combined with deep learning models fail to consider the problem of dynamical change over time of multi-dimension interaction of linked data. This paper proposes a tensor factorization recommendation model that combines time interaction learning and long short-term memory networks with attention (LA-NTF). Firstly, the long short-term memory network with attention mechanism is applied to extract the latent vector of the item from the item text information. Secondly, the multi-dimension interaction of user-item relational data in time is characterized by the long short-term memory networks with attention mechanism. Finally, the user-item-time 3D tensor is embedded in the multi-layer perceptron to learn the non-linear structural features between different latent factors, to predict the user's rating of the item. Experiments on two real-world datasets show that RMSE and MAE indicators significantly outperform neural network based factorization models and other traditional methods, indicating that the significant improvement in rating prediction task on various dynamic relational data by our LA-NTF model.
Keywords:attention mechanism    long short-term memory network    temporal interaction learning    recommendation system    tensor factorization  
点击此处可从《中文信息学报》浏览原始摘要信息
点击此处可从《中文信息学报》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司    京ICP备09084417号-23

京公网安备 11010802026262号