arXiv:2412.04948cs.CLcs.AI2024-12中稿 · Frontiers of Compu…被引 1

让大模型更懂知识,通过双视角图对比学习对齐知识图谱

KaLM: Knowledge-aligned Autoregressive Language Modeling via Dual-view Knowledge Graph Contrastive Learning

  • 用双视图知识图谱对比学习显式对齐模型与知识
  • 在知识补全和问答任务上显著提升性能
  • 适合想提升大模型知识能力的研究者

自回归大语言模型在生成任务上表现优异,但在事实查询等知识驱动任务中表现不佳。知识图谱作为高质量结构化知识库,可弥补大模型的知识缺陷。现有方法或未能有效对齐知识表示,或损害生成能力。本文提出KaLM,通过显式与隐式双重对齐目标,联合优化大模型与知识图谱的匹配:显式对齐采用双视图知识图谱对比学习直接优化知识表征;隐式对齐则通过三元组补全语言建模引入知识文本模式。实验表明,该方法在基于嵌入的知识图谱补全和基于生成的知识图谱问答任务上均有显著提升。

原文摘要 · Abstract (English)

Autoregressive large language models (LLMs) pre-trained by next token prediction are inherently proficient in generative tasks. However, their performance on knowledge-driven tasks such as factual knowledge querying remains unsatisfactory. Knowledge graphs (KGs), as high-quality structured knowledge bases, can provide reliable knowledge for LLMs, potentially compensating for their knowledge deficiencies. Aligning LLMs with explicit, structured knowledge from KGs has been a challenge; previous attempts either failed to effectively align knowledge representations or compromised the generative capabilities of LLMs, leading to less-than-optimal outcomes. This paper proposes \textbf{KaLM}, a \textit{Knowledge-aligned Language Modeling} approach, which fine-tunes autoregressive LLMs to align with KG knowledge via the joint objective of explicit knowledge alignment and implicit knowledge alignment. The explicit knowledge alignment objective aims to directly optimize the knowledge representation of LLMs through dual-view knowledge graph contrastive learning. The implicit knowledge alignment objective focuses on incorporating textual patterns of knowledge into LLMs through triple completion language modeling. Notably, our method achieves a significant performance boost in evaluations of knowledge-driven tasks, specifically embedding-based knowledge graph completion and generation-based knowledge graph question answering.

大模型知识对齐知识图谱对比学习

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