arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2307.07697cs.CL
收藏

Think-on-Graph:大语言模型在知识图谱上的深度与负责任推理

Think-on-Graph: Deep and Responsible Reasoning of Large Language Model on Knowledge Graph

IDEA Research(IDEA研究院);International Digital Economy Academy(国际数字经济学院);Xiamen University(厦门大学);The Hong Kong University of Science and Technology (Guangzhou)(香港科技大学(广州));Microsoft Research Asia(微软亚洲研究院);University of Southern California(南加州大学);The Hong Kong University of Science and Technology(香港科技大学)

Jiashuo Sun, Chengjin Xu, Lumingyuan Tang, Saizhuo Wang, Chen Lin, Yeyun Gong, Lionel M. Ni, Heung-Yeung Shum, Jian Guo

AI总结:

本文提出Think-on-Graph(ToG)范式,将LLM作为智能体在知识图谱上迭代执行束搜索以发现推理路径,无需额外训练即可提升深度推理能力并缓解幻觉问题,在6个数据集上达到SOTA。

摘要:

尽管大语言模型(LLM)在各类任务中取得了显著成功,但它们常常受困于幻觉问题,尤其是在需要深度且负责任推理的场景中。在LLM推理中引入外部知识图谱(KG)可以部分缓解这些问题。本文提出了一种新的LLM-KG整合范式“$\hbox{LLM}\otimes\hbox{KG}$”,该范式将LLM视为智能体,在KG上交互式地探索相关实体和关系,并基于检索到的知识进行推理。我们进一步通过引入一种名为Think-on-Graph(ToG)的新方法来实现这一范式,在该方法中,LLM智能体在KG上迭代执行束搜索,发现最有希望的推理路径,并返回最可能的推理结果。我们通过一系列精心设计的实验来验证并阐明ToG的以下优势:1)与LLM相比,ToG具有更强的深度推理能力;2)通过利用LLM推理和专家反馈,ToG具备知识可追溯性和知识可纠正性;3)ToG为不同的LLM、KG和提示策略提供了一个灵活的即插即用框架,无需任何额外训练成本;4)在某些场景下,使用小型LLM模型的ToG性能可以超越如GPT-4等大型LLM,这降低了LLM部署和应用的成本。作为一种无需训练、计算成本更低且通用性更好的方法,ToG在9个数据集中的6个上取得了整体SOTA,而此前大多数SOTA方法依赖于额外训练。

英文摘要:

Although large language models (LLMs) have achieved significant success in various tasks, they often struggle with hallucination problems, especially in scenarios requiring deep and responsible reasoning. These issues could be partially addressed by introducing external knowledge graphs (KG) in LLM reasoning. In this paper, we propose a new LLM-KG integrating paradigm ``$\hbox{LLM}\otimes\hbox{KG}$'' which treats the LLM as an agent to interactively explore related entities and relations on KGs and perform reasoning based on the retrieved knowledge. We further implement this paradigm by introducing a new approach called Think-on-Graph (ToG), in which the LLM agent iteratively executes beam search on KG, discovers the most promising reasoning paths, and returns the most likely reasoning results. We use a number of well-designed experiments to examine and illustrate the following advantages of ToG: 1) compared with LLMs, ToG has better deep reasoning power; 2) ToG has the ability of knowledge traceability and knowledge correctability by leveraging LLMs reasoning and expert feedback; 3) ToG provides a flexible plug-and-play framework for different LLMs, KGs and prompting strategies without any additional training cost; 4) the performance of ToG with small LLM models could exceed large LLM such as GPT-4 in certain scenarios and this reduces the cost of LLM deployment and application. As a training-free method with lower computational cost and better generality, ToG achieves overall SOTA in 6 out of 9 datasets where most previous SOTAs rely on additional training.

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

↑