融合时间上下文与特征级信息的推荐算法

作者：沈义峰; 金辰曦; 王瑶; 张家想; 卢先领^*

来源：计算机科学与探索, 2023, 17(02): 489-498.

• 推荐算法
• 序列推荐
• 自注意力机制
• 时间信息

针对基于自注意力机制的序列推荐模型忽略了各类辅助信息，导致模型不能利用它们捕捉多层次序列关系模式等问题，提出了一种融合时间上下文和特征级信息的推荐算法（ITFR）。首先将物品表示与其每一个属性表示连接起来输入到一个注意力网络，经过注意力加权后得到一种基于属性的物品表示。然后ITFR应用感知时间间隔的自注意力区块和基于物品-属性的自注意力区块分别捕捉物品与交互序列时间间隔之间的关系模式和物品与属性之间的隐式关系。最后将两个自注意力区块的输出表示连接起来，并将其作为联合输出表示输入到全连接层用于下一个物品的推荐。在两个公开数据集上进行实验，采用命中率（HR）和归一化折损累计增益（NDCG）两种性能指标进行评估。在Beauty数据集中，相较于最优的基线方法 ，HR@10和NDCG@10分别提升了4.6%和5.1%；在MovieLens-1M数据集中，HR@10和NDCG@10分别提升了1.7%和1.5%。实验结果表明，融入辅助信息增强序列表示的方法可以提升推荐性能。

• 单位
  江南大学