arXiv:2412.12464cs.IR2024-12中稿 · the 47th European …被引 12

让大模型当推理引擎,用知识图谱解决冷启动推荐难题

LLM is Knowledge Graph Reasoner: LLM's Intuition-aware Knowledge Graph Reasoning for Cold-start Sequential Recommendation

  • 用强化学习训练代理,让大模型生成图谱探索策略
  • 冷启动场景下优于现有方法,提升推荐准确率
  • 适合需要快速响应新用户/物品的推荐系统

知识图谱(KGs)以图结构表示实体间关系,是实现基于内容推荐的重要工具。然而,传统基于KG的推荐方法存在时间信息利用不足和冷启动表现差的问题。大型语言模型(LLMs)虽具备丰富的网页知识,可作为推荐系统,但其输入长度限制导致难以处理完整数据集,面临可扩展性挑战。为此,我们提出一种大模型直觉感知的知识图谱推理模型(LIKR)。核心思想是将大模型视为推理者,输出对知识图谱的直观探索策略。通过强化学习训练推荐代理,结合大模型直觉与图嵌入的奖励函数,实现知识融合。借助提示工程引入时间感知,并从有限交互中生成用户偏好文本表示,有效提升冷启动场景下的推荐性能。同时,通过使用知识图谱表示推荐数据集,仅让大模型输出图谱探索策略,避免了扩展性问题。在真实数据集上的实验表明,该模型在冷启动序列推荐任务中超越当前最优方法。

原文摘要 · Abstract (English)

Knowledge Graphs (KGs) represent relationships between entities in a graph structure and have been widely studied as promising tools for realizing recommendations that consider the accurate content information of items. However, traditional KG-based recommendation methods face fundamental challenges: insufficient consideration of temporal information and poor performance in cold-start scenarios. On the other hand, Large Language Models (LLMs) can be considered databases with a wealth of knowledge learned from the web data, and they have recently gained attention due to their potential application as recommendation systems. Although approaches that treat LLMs as recommendation systems can leverage LLMs' high recommendation literacy, their input token limitations make it impractical to consider the entire recommendation domain dataset and result in scalability issues. To address these challenges, we propose a LLM's Intuition-aware Knowledge graph Reasoning model (LIKR). Our main idea is to treat LLMs as reasoners that output intuitive exploration strategies for KGs. To integrate the knowledge of LLMs and KGs, we trained a recommendation agent through reinforcement learning using a reward function that integrates different recommendation strategies, including LLM's intuition and KG embeddings. By incorporating temporal awareness through prompt engineering and generating textual representations of user preferences from limited interactions, LIKR can improve recommendation performance in cold-start scenarios. Furthermore, LIKR can avoid scalability issues by using KGs to represent recommendation domain datasets and limiting the LLM's output to KG exploration strategies. Experiments on real-world datasets demonstrate that our model outperforms state-of-the-art recommendation methods in cold-start sequential recommendation scenarios.

知识图谱冷启动推荐大模型序列推荐

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