arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2405.02429cs.IRcs.AIcs.CLcs.LG
收藏

CALRec:用于序列推荐的生成式LLM的对比对齐

CALRec: Contrastive Alignment of Generative LLMs for Sequential Recommendation

University of Cambridge(剑桥大学);Google(谷歌公司)

Yaoyiran Li, Xiang Zhai, Moustafa Alzantot, Keyi Yu, Ivan Vulić, Anna Korhonen, Mohamed Hammad

AI总结:

提出CALRec两阶段LLM微调框架,结合对比损失与语言建模损失,显著提升序列推荐性能并超越基线。

摘要:

传统推荐系统(如矩阵分解方法)主要专注于学习共享的稠密嵌入空间来表示物品和用户偏好。随后,RNN、GRU以及最近的Transformers等序列模型相继出现,并在序列推荐任务中表现优异。该任务需要理解用户历史交互中的序列结构,以预测他们可能喜欢的下一个物品。基于大型语言模型(LLMs)在各种任务中的成功,研究人员最近探索使用在庞大文本语料库上预训练的LLMs进行序列推荐。为了将LLMs用于序列推荐,用户交互历史和模型对下一个物品的预测均以文本形式表达。我们提出CALRec,这是一个两阶段LLM微调框架,它使用两种对比损失和一种语言建模损失的混合,以双塔方式微调预训练的LLM:首先在来自多个领域的数据混合上微调LLM,随后在目标领域进行另一轮微调。我们的模型显著优于许多最先进的基线(Recall@1提升37%,NDCG@10提升24%),并且我们的系统性消融研究表明:两阶段的微调都至关重要,结合使用时我们实现了性能提升;对比对齐在我们实验探索的目标领域之间是有效的。

英文摘要:

Traditional recommender systems such as matrix factorization methods have primarily focused on learning a shared dense embedding space to represent both items and user preferences. Subsequently, sequence models such as RNN, GRUs, and, recently, Transformers have emerged and excelled in the task of sequential recommendation. This task requires understanding the sequential structure present in users' historical interactions to predict the next item they may like. Building upon the success of Large Language Models (LLMs) in a variety of tasks, researchers have recently explored using LLMs that are pretrained on vast corpora of text for sequential recommendation. To use LLMs for sequential recommendation, both the history of user interactions and the model's prediction of the next item are expressed in text form. We propose CALRec, a two-stage LLM finetuning framework that finetunes a pretrained LLM in a two-tower fashion using a mixture of two contrastive losses and a language modeling loss: the LLM is first finetuned on a data mixture from multiple domains followed by another round of target domain finetuning. Our model significantly outperforms many state-of-the-art baselines (+37% in Recall@1 and +24% in NDCG@10) and our systematic ablation studies reveal that (i) both stages of finetuning are crucial, and, when combined, we achieve improved performance, and (ii) contrastive alignment is effective among the target domains explored in our experiments.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑