4 个月前

基于会话的循环神经网络推荐系统

基于会话的循环神经网络推荐系统

摘要

我们在推荐系统这一新领域应用了循环神经网络(RNN)。现实中的推荐系统经常面临只能基于短会话数据(例如小型体育用品网站)而非长用户历史(如Netflix的情况)进行推荐的问题。在这种情况下,备受赞誉的矩阵分解方法并不准确。通常,实际中通过转向物品到物品的推荐来克服这一问题,即推荐相似的物品。我们认为,通过对整个会话进行建模,可以提供更准确的推荐。因此,我们提出了一种基于RNN的会话推荐方法。我们的方法还考虑了任务的实际方面,并对经典的RNN进行了若干改进,例如引入了排名损失函数,使其更适合解决这一特定问题。在两个数据集上的实验结果表明,该方法相比广泛使用的方法有显著改进。

代码仓库

yhs968/pyGRU4REC
pytorch
GitHub 中提及
hungthanhpham94/GRU4REC-pytorch
pytorch
GitHub 中提及
yeganegi-reza/torch-gru4rec
pytorch
GitHub 中提及
hu-dske/ILSTP
tf
GitHub 中提及
hidasib/GRU4Rec
tf
GitHub 中提及
ifuseok/TripRecommendation
tf
GitHub 中提及
yoavnavon/GRU4REC-spotify
tf
GitHub 中提及
jacklu2016/kerasGRU4Rec_c
tf
GitHub 中提及
hidasib/gru4rec_pytorch_official
pytorch
GitHub 中提及
maciejkula/spotlight
pytorch
GitHub 中提及
UlionTse/mlgb
pytorch
GitHub 中提及
paxcema/KerasGRU4Rec
tf
GitHub 中提及
miniii222/Graduate-Paper
tf
GitHub 中提及
nggianno/thesis
tf
GitHub 中提及
bekleyis95/RNN-RecSys
pytorch
GitHub 中提及
hungpthanh/gru4rec-pytorch
pytorch
GitHub 中提及
massquantity/LibRecommender
tf
GitHub 中提及

基准测试

基准方法指标
collaborative-filtering-on-movielens-1mGRU4Rec
HR@10 (full corpus): 0.2811
NDCG@10 (full corpus): 0.1648
collaborative-filtering-on-movielens-20mGRU4Rec
HR@10 (full corpus): 0.2813
nDCG@10 (full corpus): 0.1730
session-based-recommendations-on-digineticaGRU4REC
Hit@20: 29.45
MRR@20: 8
session-based-recommendations-on-yoochoose1-1GRU4REC
HR@20: 60.64
MRR@20: 22.89

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于会话的循环神经网络推荐系统 | 论文 | HyperAI超神经