用LLM生成推荐轨迹,训练更高效可靠的推荐模型
Beyond Static LLM Policies: Imitation-Enhanced Reinforcement Learning for Recommendation
- 通过逆强化学习从LLM推荐轨迹中提取奖励模型
- 在两个数据集上超越主流RL和上下文学习方法
- 无需微调LLM,降低延迟与计算开销,适合工业部署
推荐系统已成为提升用户参与度的关键工具。大型语言模型(LLMs)凭借出色的泛化能力和上下文理解能力,展现出提升推荐系统的潜力。然而,直接将LLM作为推荐策略存在显著问题:频繁调用API导致高延迟,且存在幻觉和偏见等固有缺陷。为此,本文提出一种新型离线强化学习框架,利用LLM生成的推荐轨迹进行模仿学习。具体采用逆强化学习从LLM示范轨迹中提取稳健的奖励模型,避免了对LLM的微调,大幅降低计算开销。同时,强化学习策略由这些示范轨迹的累积奖励引导,有效迁移了LLM捕捉到的语义信息。在两个基准数据集上的全面实验验证了该方法的有效性,性能优于当前最先进的基于RL和上下文学习的方法。代码已开源:https://github.com/ArronDZhang/IL-Rec。
原文摘要 · Abstract (English)
Recommender systems (RecSys) have become critical tools for enhancing user engagement by delivering personalized content across diverse digital platforms. Recent advancements in large language models (LLMs) demonstrate significant potential for improving RecSys, primarily due to their exceptional generalization capabilities and sophisticated contextual understanding, which facilitate the generation of flexible and interpretable recommendations. However, the direct deployment of LLMs as primary recommendation policies presents notable challenges, including persistent latency issues stemming from frequent API calls and inherent model limitations such as hallucinations and biases. To address these issues, this paper proposes a novel offline reinforcement learning (RL) framework that leverages imitation learning from LLM-generated trajectories. Specifically, inverse reinforcement learning is employed to extract robust reward models from LLM demonstrations. This approach negates the need for LLM fine-tuning, thereby substantially reducing computational overhead. Simultaneously, the RL policy is guided by the cumulative rewards derived from these demonstrations, effectively transferring the semantic insights captured by the LLM. Comprehensive experiments conducted on two benchmark datasets validate the effectiveness of the proposed method, demonstrating superior performance when compared against state-of-the-art RL-based and in-context learning baselines. The code can be found at https://github.com/ArronDZhang/IL-Rec.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。