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

被引:11
作者
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 条
[51]   A Survey on Session-based Recommender Systems [J].
Wang, Shoujin ;
Cao, Longbing ;
Wang, Yan ;
Sheng, Quan Z. ;
Orgun, Mehmet A. ;
Lian, Defu .
ACM COMPUTING SURVEYS, 2021, 54 (07)
[52]  
Wang T., 2020, INT C MACH LEARN, P9
[53]   Global Context Enhanced Graph Neural Networks for Session-based Recommendation [J].
Wang, Ziyang ;
Wei, Wei ;
Cong, Gao ;
Li, Xiao-Li ;
Mao, Xian-Ling ;
Qiu, Minghui .
PROCEEDINGS OF THE 43RD INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL (SIGIR '20), 2020, :169-178
[54]   Contrastive Learning for Cold-Start Recommendation [J].
Wei, Yinwei ;
Wang, Xiang ;
Li, Qi ;
Nie, Liqiang ;
Li, Yan ;
Li, Xuanping ;
Chua, Tat-Seng .
PROCEEDINGS OF THE 29TH ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA, MM 2021, 2021, :5382-5390
[55]   Self-supervised Graph Learning for Recommendation [J].
Wu, Jiancan ;
Wang, Xiang ;
Feng, Fuli ;
He, Xiangnan ;
Chen, Liang ;
Lian, Jianxun ;
Xie, Xing .
SIGIR '21 - PROCEEDINGS OF THE 44TH INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL, 2021, :726-735
[56]  
Wu S, 2019, AAAI CONF ARTIF INTE, P346
[57]   Self-Supervised Graph Co-Training for Session-based Recommendation [J].
Xia, Xin ;
Yin, Hongzhi ;
Yu, Junliang ;
Shao, Yingxia ;
Cui, Lizhen .
PROCEEDINGS OF THE 30TH ACM INTERNATIONAL CONFERENCE ON INFORMATION & KNOWLEDGE MANAGEMENT, CIKM 2021, 2021, :2180-2190
[58]  
Xia X, 2021, AAAI CONF ARTIF INTE, V35, P4503
[59]   Contrastive Learning for Sequential Recommendation [J].
Xie, Xu ;
Sun, Fei ;
Liu, Zhaoyang ;
Wu, Shiwen ;
Gao, Jinyang ;
Zhang, Jiandong ;
Ding, Bolin ;
Cui, Bin .
2022 IEEE 38TH INTERNATIONAL CONFERENCE ON DATA ENGINEERING (ICDE 2022), 2022, :1259-1273
[60]  
Xu CF, 2019, PROCEEDINGS OF THE TWENTY-EIGHTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P3940