arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2402.11441cs.CLcs.AIcs.LG
收藏

InfuserKI:通过注入器引导的知识集成增强大语言模型与知识图谱

InfuserKI: Enhancing Large Language Models with Knowledge Graphs via Infuser-Guided Knowledge Integration

Pennsylvania State University(宾夕法尼亚州立大学);GE Healthcare(GE医疗);NEC Laboratories America(美国NEC实验室)

Fali Wang, Runxue Bao, Suhang Wang, Wenchao Yu, Yanchi Liu, Wei Cheng, Haifeng Chen

AI总结:

针对大语言模型知识密集型任务中知识集成效率低与遗忘风险,提出InfuserKI框架,利用Transformer内部状态引导知识注入,在UMLS-2.5k和MetaQA上分别降低9%和6%的知识遗忘。

摘要:

大语言模型(LLMs)在跨领域的开放生成任务中展现出卓越能力,但在需要密集知识的任务中仍面临困难。为应对这些挑战,研究者开发了知识集成方法,通过外部模块将领域特定知识图谱增强到LLMs中。然而,这些方法面临数据效率低下的问题,因为它们需要对已知和未知知识都进行处理以进行微调。因此,我们的研究聚焦于一个新问题:高效地将未知知识集成到LLMs中,同时避免对已知知识的不必要重复处理。引入新知识的一个风险是可能遗忘已有知识。为降低这一风险,我们提出了创新的{\method}框架。该框架利用Transformer内部状态来决定何时用额外信息丰富LLM输出,从而有效防止知识遗忘。使用UMLS-2.5k和MetaQA领域知识图谱的性能评估表明,{\method}不仅成功集成了新知识,还优于最先进的基线方法,分别将知识遗忘降低了9%和6%。

英文摘要:

Large Language Models (LLMs) have achieved exceptional capabilities in open generation across various domains, yet they encounter difficulties with tasks that require intensive knowledge. To address these challenges, methods for integrating knowledge have been developed, which augment LLMs with domain-specific knowledge graphs through external modules. These approaches, however, face data inefficiency issues as they necessitate the processing of both known and unknown knowledge for fine-tuning. Thus, our research focuses on a novel problem: efficiently integrating unknown knowledge into LLMs without unnecessary overlap of known knowledge. A risk of introducing new knowledge is the potential forgetting of existing knowledge. To mitigate this risk, we propose the innovative {\method} framework. This framework employs transformer internal states to determine when to enrich LLM outputs with additional information, effectively preventing knowledge forgetting. Performance evaluations using the UMLS-2.5k and MetaQA domain knowledge graphs reveal that {\method} not only successfully integrates new knowledge but also outperforms state-of-the-art baselines, reducing knowledge forgetting by 9\% and 6\%, respectively.

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

↑