Spatio-temporal Contrastive Learning-enhanced GNNs for Session-based Recommendation

被引:10
作者
Wan, Zhongwei [1 ]
Liu, Xin [2 ]
Wang, Benyou [3 ]
Qiu, Jiezhong [4 ]
Li, Boyu [5 ]
Guo, Ting [5 ]
Chen, Guangyong [6 ]
Wang, Yang [5 ]
机构
[1] Ohio State Univ, Columbus, OH 43210 USA
[2] Hong Kong Univ Sci & Technol, Hong Kong, Peoples R China
[3] Chinese Univ Hong Kong, Shenzhen, Peoples R China
[4] Tencent, Beijing, Peoples R China
[5] Univ Technol Sydney, Sydney, NSW, Australia
[6] Zhejiang Lab, Hangzhou, Peoples R China
关键词
Recommendation system; session-based recommendation; graph neural network; temporal information; contrastive learning; NETWORKS;
D O I
10.1145/3626091
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Session-based recommendation (SBR) systems aim to utilize the user's short-term behavior sequence to predict the next item without the detailed user profile. Most recent works try to model the user preference by treating the sessions as between-item transition graphs and utilize various graph neural networks (GNNs) to encode the representations of pair-wise relations among items and their neighbors. Some of the existing GNN-based models mainly focus on aggregating information from the view of spatial graph structure, which ignores the temporal relations within neighbors of an item during message passing and the information loss results in a sub-optimal problem. Other works embrace this challenge by incorporating additional temporal information but lack sufficient interaction between the spatial and temporal patterns. To address this issue, inspired by the uniformity and alignment properties of contrastive learning techniques, we propose a novel framework called Session-based Recommendation with Spatio-temporal Contrastive Learning-enhanced GNNs (RESTC). The idea is to supplement the GNN-based main supervised recommendation task with the temporal representation via an auxiliary cross-view contrastive learning mechanism. Furthermore, a novel global collaborative filtering graph embedding is leveraged to enhance the spatial view in the main task. Extensive experiments demonstrate the significant performance of RESTC compared with the state-of-the-art baselines. We release our source code at https://github.com/SUSTechBruce/RESTC-Source-code.
引用
收藏
页数:26
相关论文
共 70 条
[1]  
Arora S, 2019, PR MACH LEARN RES, V97
[2]  
Blum A., 1998, Proceedings of the Eleventh Annual Conference on Computational Learning Theory, P92, DOI 10.1145/279943.279962
[3]   Handling Information Loss of Graph Neural Networks for Session-based Recommendation [J].
Chen, Tianwen ;
Wong, Raymond Chi-Wing .
KDD '20: PROCEEDINGS OF THE 26TH ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING, 2020, :1172-1180
[4]  
Chen T, 2020, PR MACH LEARN RES, V119
[5]  
Chen XL, 2021, Arxiv, DOI arXiv:2104.02057
[6]   Hybrid-Order Gated Graph Neural Network for Session-Based Recommendation [J].
Chen, Yan-Hui ;
Huang, Ling ;
Wang, Chang-Dong ;
Lai, Jian-Huang .
IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS, 2022, 18 (03) :1458-1467
[7]   Intent Contrastive Learning for Sequential Recommendation [J].
Chen, Yongjun ;
Liu, Zhiwei ;
Li, Jia ;
McAuley, Julian ;
Xiong, Caiming .
PROCEEDINGS OF THE ACM WEB CONFERENCE 2022 (WWW'22), 2022, :2172-2182
[8]   G∧3SR: Global Graph Guided Session-Based Recommendation [J].
Deng, Zhi-Hong ;
Wang, Chang-Dong ;
Huang, Ling ;
Lai, Jian-Huang ;
Yu, Philip S. .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2023, 34 (12) :9671-9684
[9]  
Devlin J, 2019, 2019 CONFERENCE OF THE NORTH AMERICAN CHAPTER OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS: HUMAN LANGUAGE TECHNOLOGIES (NAACL HLT 2019), VOL. 1, P4171
[10]   Continuous-Time Sequential Recommendation with Temporal Graph Collaborative Transformer [J].
Fan, Ziwei ;
Liu, Zhiwei ;
Zhang, Jiawei ;
Xiong, Yun ;
Zheng, Lei ;
Yu, Philip S. .
PROCEEDINGS OF THE 30TH ACM INTERNATIONAL CONFERENCE ON INFORMATION & KNOWLEDGE MANAGEMENT, CIKM 2021, 2021, :433-442