Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

TALLRec: An Effective and Efficient Tuning Framework to Align Large Language Model with Recommendation

About

Large Language Models (LLMs) have demonstrated remarkable performance across diverse domains, thereby prompting researchers to explore their potential for use in recommendation systems. Initial attempts have leveraged the exceptional capabilities of LLMs, such as rich knowledge and strong generalization through In-context Learning, which involves phrasing the recommendation task as prompts. Nevertheless, the performance of LLMs in recommendation tasks remains suboptimal due to a substantial disparity between the training tasks for LLMs and recommendation tasks, as well as inadequate recommendation data during pre-training. To bridge the gap, we consider building a Large Recommendation Language Model by tunning LLMs with recommendation data. To this end, we propose an efficient and effective Tuning framework for Aligning LLMs with Recommendation, namely TALLRec. We have demonstrated that the proposed TALLRec framework can significantly enhance the recommendation capabilities of LLMs in the movie and book domains, even with a limited dataset of fewer than 100 samples. Additionally, the proposed framework is highly efficient and can be executed on a single RTX 3090 with LLaMA-7B. Furthermore, the fine-tuned LLM exhibits robust cross-domain generalization. Our code and data are available at https://github.com/SAI990323/TALLRec.

Keqin Bao, Jizhi Zhang, Yang Zhang, Wenjie Wang, Fuli Feng, Xiangnan He• 2023

Related benchmarks

TaskDatasetResultRank
Sequential RecommendationML 1M
NDCG@100.0197
130
RecommendationMovieLens
Accuracy97.9
84
Sequential RecommendationAmazon Arts (test)
NDCG@1049.44
50
RecommendationMovieLens 1M (test)
NDCG@556.13
46
Sequential RecommendationMovieLens
ValidRatio0.9966
41
Sequential RecommendationAmazon Office (test)
NDCG@1033.18
31
Sequential RecommendationMicroLens (test)
Recall@565.97
31
RecommendationNetflix (test)--
30
RecommendationHRT
ND@50.8442
26
Sequential RecommendationAmazon Toys (test)
NDCG@1045.77
24
Showing 10 of 78 rows
...

Other info

Follow for update