arXiv:2410.12229cs.IRcs.AI2024-10中稿 · as a full paper by…被引 32

用大模型补全知识图谱,提升推荐系统准确性

Comprehending Knowledge Graphs with Large Language Models for Recommender Systems

  • 利用大模型理解图谱中物品的局部与全局语义
  • 在四个真实数据集上显著优于现有方法
  • 适合需要精准推荐的电商、内容平台场景

近年来,知识图谱(KG)的引入显著提升了推荐系统的能力,有助于发现物品间的潜在关联。然而,现有方法仍面临多重挑战:多数知识图谱存在事实缺失或范围有限;传统方法将文本信息转化为编号,导致语义联系丢失;且难以捕捉全局高阶连接。为此,我们提出一种新方法 CoLaKG,借助大语言模型(LLMs)改进基于知识图谱的推荐。凭借其丰富知识和强大推理能力,该方法可补充知识图谱中的缺失事实,并更充分地利用语义信息。CoLaKG 在局部与全局层面提取有用信息:通过以物品为中心的子图抽取与提示工程准确理解局部信息;通过基于语义的检索模块,从整个知识图谱中为每个物品补充相关项,有效利用全局信息。此外,局部与全局信息分别通过表示融合模块与检索增强表示学习模块整合进推荐模型。在四个真实世界数据集上的广泛实验表明,本方法具有显著优势。

原文摘要 · Abstract (English)

In recent years, the introduction of knowledge graphs (KGs) has significantly advanced recommender systems by facilitating the discovery of potential associations between items. However, existing methods still face several limitations. First, most KGs suffer from missing facts or limited scopes. Second, existing methods convert textual information in KGs into IDs, resulting in the loss of natural semantic connections between different items. Third, existing methods struggle to capture high-order connections in the global KG. To address these limitations, we propose a novel method called CoLaKG, which leverages large language models (LLMs) to improve KG-based recommendations. The extensive knowledge and remarkable reasoning capabilities of LLMs enable our method to supplement missing facts in KGs, and their powerful text understanding abilities allow for better utilization of semantic information. Specifically, CoLaKG extracts useful information from KGs at both local and global levels. By employing the item-centered subgraph extraction and prompt engineering, it can accurately understand the local information. In addition, through the semantic-based retrieval module, each item is enriched by related items from the entire knowledge graph, effectively harnessing global information. Furthermore, the local and global information are effectively integrated into the recommendation model through a representation fusion module and a retrieval-augmented representation learning module, respectively. Extensive experiments on four real-world datasets demonstrate the superiority of our method.

知识图谱推荐系统大模型应用

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