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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Sequential Recommendation | Amazon Beauty (test) | NDCG@103.83 | 170 | |
| Sequential Recommendation | Amazon Sports and Outdoors (test) | Recall@50.0208 | 50 | |
| Sequential Recommendation | Sports | Recall@50.0208 | 43 | |
| Sequential Recommendation | Toys | Recall@54.97 | 42 | |
| Sequential Recommendation | Amazon Toys and Games (test) | Recall@54.97 | 40 | |
| Sequential Recommendation | Amazon CDs and Vinyl (test) | NDCG@102.92 | 35 | |
| Sequential Recommendation | CDs | Recall@53.52 | 21 | |
| Recommendation | Food | Recall@100.0646 | 20 | |
| Sequential Recommendation | Books Amazon (test) | -- | 20 | |
| Recommendation | Amazon Toys (test) | Recall@100.0834 | 17 |