[🌐 Website] · [🤗 Huggingface Datasets] · [📑 Paper] · [🔬 McAuley Lab]
This repository contains:
- Scripts for processing Amazon Reviews 2023 dataset into recommendation benchmarks;
- Checkpoints & implementations for BLaIR: "Bridging Language and Items for Retrieval and Recommendation";
- Scripts for constructing Amazon-C4, a new dataset for evaluating product search performance under complex contexts.
Based on the released Amazon Reviews 2023 dataset, we provide scripts to preprocess raw data into standard train/validation/test splits to encourage benchmarking recommendation models.
More details here -> [datasets & processing scripts]
BLaIR, which is short for "Bridging Language and Items for Retrieval and Recommendation", is a series of language models pre-trained on Amazon Reviews 2023 dataset.
BLaIR is grounded on pairs of (item metadata, language context), enabling the models to:
- derive strong item text representations, for both recommendation and retrieval;
- predict the most relevant item given simple / complex language context.
More details here -> [checkpoints & code]
Amazon-C4, which is short for "Complex Contexts Created by ChatGPT", is a new dataset for the complex product search task.
Amazon-C4 is designed to assess a model's ability to comprehend complex language contexts and retrieve relevant items.
More details here -> [datasets & code]
- Please refer to seq_rec_results for scripts that can reproduce our results on sequential recommendation.
- Please refer to product_search_results for scripts that can reproduce our results on product search.
Please let us know if you encounter a bug or have any suggestions/questions by filling an issue or emailing Yupeng Hou (@hyp1231) at yphou@ucsd.edu.
If you find Amazon Reviews 2023 dataset, BLaIR checkpoints, Amazon-C4 dataset, or our scripts/code helpful, please cite the following paper.
@article{hou2024bridging,
title={Bridging Language and Items for Retrieval and Recommendation},
author={Hou, Yupeng and Li, Jiacheng and He, Zhankui and Yan, An and Chen, Xiusi and McAuley, Julian},
journal={arXiv preprint arXiv:2403.03952},
year={2024}
}
The recommendation experiments in the BLaIR paper are implemented using the open-source recommendation library RecBole.
The pre-training scripts refer a lot to huggingface language-modeling examples and SimCSE.