Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Contrastive Learning for Sequential Recommendation

About

Sequential recommendation methods play a crucial role in modern recommender systems because of their ability to capture a user's dynamic interest from her/his historical interactions. Despite their success, we argue that these approaches usually rely on the sequential prediction task to optimize the huge amounts of parameters. They usually suffer from the data sparsity problem, which makes it difficult for them to learn high-quality user representations. To tackle that, inspired by recent advances of contrastive learning techniques in the computer version, we propose a novel multi-task model called \textbf{C}ontrastive \textbf{L}earning for \textbf{S}equential \textbf{Rec}ommendation~(\textbf{CL4SRec}). CL4SRec not only takes advantage of the traditional next item prediction task but also utilizes the contrastive learning framework to derive self-supervision signals from the original user behavior sequences. Therefore, it can extract more meaningful user patterns and further encode the user representation effectively. In addition, we propose three data augmentation approaches to construct self-supervision signals. Extensive experiments on four public datasets demonstrate that CL4SRec achieves state-of-the-art performance over existing baselines by inferring better user representations.

Xu Xie, Fei Sun, Zhaoyang Liu, Shiwen Wu, Jinyang Gao, Bolin Ding, Bin Cui• 2020

Related benchmarks

TaskDatasetResultRank
Sequential RecommendationYelp
Recall@100.0392
80
Sequential RecommendationAmazon Beauty
Recall@105.79
48
Sequential RecommendationSports
Recall@50.0227
43
Sequential RecommendationYelp (Overall)
Hit Rate @100.0272
36
Sequential RecommendationAmazon Toys
R@100.087
30
Sequential RecommendationAmazon Sport
R@100.0491
30
Sequential RecommendationBeauty
HR@106.83
30
Sequential RecommendationDouyin
H@100.1305
24
Sequential RecommendationToys (Overall)
Hit Rate @106.52
24
Sequential RecommendationAmazon Sport
NDCG@200.0331
24
Showing 10 of 30 rows

Other info

Follow for update