基于Bi-GRU和外部注意网络的会话推荐模型

作者:张海桃; 祁正华*; 谭小辉; 何菲菲
来源:南京邮电大学学报(自然科学版), 2023, 43(05): 92-101.
DOI:10.14132/j.cnki.1673-5439.2023.05.012

摘要

基于会话的推荐系统(SRS)可以根据匿名用户的历史行为序列预测用户的下一个动作。现有推荐模型大都会忽略当前会话与其他会话的潜在相关性,不能捕捉用户的主要意图。为解决该问题,提出一种基于Bi-GRU和外部注意网络的会话推荐模型(SR-BGEAN)。该模型首次将外部注意机制引入会话推荐任务中,并结合Bi-GRU进行建模,同时捕获会话内物品双向顺序信息和会话样本间全局信息,形成最终的兴趣表示。在Diginetica和Yoochoose数据集上的实验结果表明,相对最优的基线模型,SR-BGEAN模型在推荐的前20个项目中的命中率更高,可达到71.50%。

全文