RecBench
May 16, 2025 ยท View on GitHub
Can LLMs Outshine Conventional Recommenders? A Comparative Evaluation
Installation
gh repo clone Jyonn/RecBench
cd RecBench
pip install -r requirements.txt
๐ Supported Datasets
RecBench supports 15 datasets across domains like news, books, movies, music, fashion, and e-commerce:
- ๐ฐ MIND: Large-scale Microsoft news data for CTR prediction.
- ๐ฐ PENS: Personalized news recommendation dataset.
- ๐ Goodreads: Book reviews and metadata.
- ๐ Amazon Books: Subset of Amazon product reviews.
- ๐ฅ MovieLens: Classic movie rating dataset.
- ๐บ MicroLens: MovieLens dataset with user-item interactions.
- ๐บ Netflix Prize: Large-scale movie rating competition dataset.
- ๐ต Amazon CDs: Music CD reviews and metadata.
- ๐ต Last.fm: Music playback logs and tagging data.
- ๐ H&M: Apparel and fashion product data.
- ๐ POG: Fashion product reviews and metadata.
- ๐ฑ Amazon Electronics: Electronics product reviews and metadata.
- ๐ฎ Steam: Video game reviews and metadata.
- ๐จ HotelRec: Hotel recommendation dataset.
- ๏ธ๏ธ๐ฝ๏ธ Yelp: Restaurant reviews and metadata.
You can download our preprocessed data from Kaggle (Recommended), Google Drive, and Github Release.
Usage
Example 1: Zero-shot, Pair-wise
python worker.py --model llama1 --data mind
Example 2: Fine-tune, Pair-wise
python tuner.py --model llama1 --train mind --valid mind
Example 3: Fine-tune, List-wise, Unique-ID-based
python seq_processor.py --data mind # preprocess SeqRec data
python id_coder.py --data mind --seq true # use unique identifier to represent items
python seq_tuner.py --model llama1seq --data mind --code_path ./code/mind.id.seq.code
Example 4: Fine-tune, List-wise, Semantic-ID-based
python embedder.py --data mind --model llama1 # extract item embeddings
python code_generator.py --data mind --model llama1 # use RQ-VAE for discrete tokenization
python seq_tuner.py --model llama1seq --data mind --code_path ./code/mind.llama1.seq.code
More documentations will be available soon.
Updates
- 2025-03-07: Our first benchmark paper is posted on arXiv: Benchmarking LLMs in Recommendation Tasks: A Comparative Evaluation with Conventional Recommenders.
- 2024-12-15: RecBench v1 library is released.
- 2024-06-04: RecBench project is initiated.
Citations
If you find RecBench useful in your research, please consider citing our project:
@article{liu2025benchmarking,
title={Benchmarking LLMs in Recommendation Tasks: A Comparative Evaluation with Conventional Recommenders},
author={Liu, Qijiong and Zhu, Jieming and Fan, Lu and Wang, Kun and Hu, Hengchang and Guo, Wei and Liu, Yong and Wu, Xiao-Ming},
journal={arXiv preprint arXiv:2503.05493},
year={2025}
}