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

Learning Vector-Quantized Item Representation for Transferable Sequential Recommenders

About

Recently, the generality of natural language text has been leveraged to develop transferable recommender systems. The basic idea is to employ pre-trained language models~(PLM) to encode item text into item representations. Despite the promising transferability, the binding between item text and item representations might be too tight, leading to potential problems such as over-emphasizing the effect of text features and exaggerating the negative impact of domain gap. To address this issue, this paper proposes VQ-Rec, a novel approach to learning Vector-Quantized item representations for transferable sequential Recommenders. The main novelty of our approach lies in the new item representation scheme: it first maps item text into a vector of discrete indices (called item code), and then employs these indices to lookup the code embedding table for deriving item representations. Such a scheme can be denoted as "text $\Longrightarrow$ code $\Longrightarrow$ representation". Based on this representation scheme, we further propose an enhanced contrastive pre-training approach, using semi-synthetic and mixed-domain code representations as hard negatives. Furthermore, we design a new cross-domain fine-tuning method based on a differentiable permutation-based network. Extensive experiments conducted on six public benchmarks demonstrate the effectiveness of the proposed approach, in both cross-domain and cross-platform settings. Code and pre-trained model are available at: https://github.com/RUCAIBox/VQ-Rec.

Yupeng Hou, Zhankui He, Julian McAuley, Wayne Xin Zhao• 2022

Related benchmarks

TaskDatasetResultRank
Sequential RecommendationSports
Recall@50.0208
43
Sequential RecommendationToys
Recall@54.97
31
Sequential RecommendationCDs
Recall@53.52
21
Sequential RecommendationBooks Amazon (test)--
20
Sequential RecommendationBeauty
Recall@54.57
13
Sequential RecommendationAmazon Product Reviews Arts, Crafts and Sewing leave-one-out (test)
HR@14.08
12
Sequential RecommendationAmazon Product Reviews Musical Instruments leave-one-out (test)
HR@15.02
12
Sequential RecommendationAmazon Product Reviews Video Games leave-one-out (test)
HR@10.0075
12
Sequential RecommendationAmazon Electronics (test)
Recall@50.0331
11
Sequential RecommendationIndustry internal (test)
Recall@515.2
11
Showing 10 of 13 rows

Other info

Follow for update