基于对比学习的动态图序列推荐方法

摘要

为了缓解目前基于动态图表征的序列推荐研究中存在的用户交互数据高稀疏且包含大量噪音、模型训练需要大量有标记样本等问题，本文提出了基于对比学习的动态图序列推荐方法（CDGSR）。具体而言，CDGSR从粗粒度和细粒度不同层面设计了层间对比学习、两次传播对比学习和随机噪声扰动对比学习三种不同角度的对比学习方法。实验表明，CDGSR在Amazon-Beauty、Amazon-Games、Amazon-CDs三个现实数据集上的归一化折损累计增益NDCG@10分别达到了0.363 3、0.587 3、0.522 0,Hit@10分别达到了0.525 8、0.778 6、0.735 9。与基于矩阵分解的BPR-MF、FPMC，基于神经网络的GRU4Rec、Caser、SASRec和基于图神经网络的SR-GNN、HGN、HyperRec、DGSR等方法相比，CDGSR均取得了最好的结果。其中，与性能最好的DGSR相比，CDGSR在Amazon-CDs数据集上的NDCG@10提升了1.97%,Hit@10提升了1.60%。这些结果表明，本文提出的CDGSR能够有效利用对比学习方法提升动态图序列推荐方法的推荐性能。

关键词

序列推荐 / 图神经网络 / 动态图表征 / 对比学习

基金资助

浙江大学上海高等研究院繁星科学基金（SN-ZJU-SIAS-001）; 国家自然科学基金（62372399）