Our new X account is live! Follow @wizwand_team for updates
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
RecommendationMovieLens
Accuracy97.9
84
Sequential RecommendationMovieLens
ValidRatio0.9966
41
Sequential RecommendationAmazon Toys (test)
NDCG@1045.77
24
RecommendationMovies & TV
NDCG@1035.4
18
RecommendationMIND
Accuracy81.96
16
RecommendationAMAZON
Accuracy94.91
16
Response SelectionPersonaMem
Accuracy64.36
16
List-wise RecommendationLastFM
Hit Ratio@139.71
16
Response SelectionAlignX
Accuracy66.3
16
Pair-wise RecommendationLastFM
Hit Ratio@176.97
16
Showing 10 of 32 rows

Other info

Follow for update