Share your thoughts, 1 month free Claude Pro on usSee more
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 RecommendationML 1M
NDCG@100.0989
130
Sequential RecommendationYelp
Recall@100.0392
120
Sequential RecommendationAmazon Beauty
NDCG@103.07
84
Sequential RecommendationYelp (Overall)
Hit Rate @100.0272
63
Sequential RecommendationBeauty
HR@106.83
58
Sequential RecommendationAmazon Toys
R@100.087
51
RecommendationMovieLens 1M (test)
NDCG@558.61
46
Sequential RecommendationSports
Recall@50.0227
43
Sequential RecommendationBeauty
Hit Rate @ 200.1068
43
Sequential RecommendationMovieLens 1M (test)
Hit@1023.46
42
Showing 10 of 61 rows

Other info

Follow for update