arXiv:2410.05939cs.IR2024-10中稿 · ICME 2025被引 8

用偏好优化让大模型更懂推荐,性能显著提升

Direct Preference Optimization for LLM-Enhanced Recommendation Systems

  • 通过直接偏好优化对齐大模型输出与推荐行为
  • 在多个数据集上重排效果超越强基线,提升明显
  • 适合想提升大模型推荐能力的研究者和工程师

大语言模型(LLMs)在多个领域表现出色,激发了其在推荐系统中的应用研究。早期工作通过上下文学习将推荐任务转化为提示,利用大模型的丰富知识和强泛化能力。然而,由于预训练目标与推荐任务不匹配,且预训练阶段缺乏推荐数据,大模型在推荐场景中表现受限。为此,我们提出 DPO4Rec 框架,将直接偏好优化(DPO)引入大模型增强的推荐系统。首先,提示大模型从历史交互中推断用户偏好,并用于增强基于ID的序列推荐模型;其次,构建基于知识增强的推荐架构作为奖励模型,评估大模型生成推理的质量;接着,从N个样本中选取最高和最低排名响应,构建用于大模型微调的数据集;最后,通过结构对齐策略使用DPO对齐大模型输出与理想推荐行为。大量实验表明,DPO4Rec显著优于强基线,在重排性能上实现显著提升,验证了大模型在推荐任务中指令遵循能力的增强。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have exhibited remarkable performance across a wide range of domains, motivating research into their potential for recommendation systems. Early efforts have leveraged LLMs' rich knowledge and strong generalization capabilities via in-context learning, where recommendation tasks are framed as prompts. However, LLM performance in recommendation scenarios remains limited due to the mismatch between their pretraining objectives and recommendation tasks, as well as the lack of recommendation-specific data during pretraining. To address these challenges, we propose DPO4Rec, a novel framework that integrates Direct Preference Optimization (DPO) into LLM-enhanced recommendation systems. First, we prompt the LLM to infer user preferences from historical interactions, which are then used to augment traditional ID-based sequential recommendation models. Next, we train a reward model based on knowledge-augmented recommendation architectures to assess the quality of LLM-generated reasoning. Using this, we select the highest- and lowest-ranked responses from N samples to construct a dataset for LLM fine-tuning. Finally, we apply a structure alignment strategy via DPO to align the LLM's outputs with desirable recommendation behavior. Extensive experiments show that DPO4Rec significantly improves re-ranking performance over strong baselines, demonstrating enhanced instruction-following capabilities of LLMs in recommendation tasks.

大模型推荐偏好优化指令对齐

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。