arXiv:2506.03576cs.CLcs.AI2025-06被引 3

用双向语言模型融合知识图谱结构与语义,提升关系推理能力

KG-BiLM: Knowledge Graph Embedding via Bidirectional Language Models

  • 双向注意力机制让实体与文本充分交互
  • 在大规模多跳关系图上链接预测显著优于基线
  • 适合需要联合建模结构与语义的任务场景

知识表示学习的最新进展凸显了将符号化知识图谱(KGs)与语言模型(LMs)统一以实现更丰富的语义理解的迫切需求。然而,现有方法通常侧重于图结构或文本语义,缺乏同时捕捉全局图连通性、细微语言上下文和判别性推理语义的统一框架。为此,我们提出KG-BiLM,一种融合图谱结构线索与生成式Transformer语义表达的双向语言模型框架。KG-BiLM包含三个关键组件:(i) 双向知识注意力,去除因果掩码以实现所有标记与实体间的全交互;(ii) 知识掩码预测,促使模型利用局部语义上下文与全局图连通性;(iii) 对比图语义聚合,通过采样子图表示的对比对齐保持图结构。在标准基准上的大量实验表明,KG-BiLM在链接预测任务中优于强基线,尤其在具有复杂多跳关系的大规模图上表现突出,验证了其在统一结构信息与文本语义方面的有效性。

原文摘要 · Abstract (English)

Recent advances in knowledge representation learning (KRL) highlight the urgent necessity to unify symbolic knowledge graphs (KGs) with language models (LMs) for richer semantic understanding. However, existing approaches typically prioritize either graph structure or textual semantics, leaving a gap: a unified framework that simultaneously captures global KG connectivity, nuanced linguistic context, and discriminative reasoning semantics. To bridge this gap, we introduce KG-BiLM, a bidirectional LM framework that fuses structural cues from KGs with the semantic expressiveness of generative transformers. KG-BiLM incorporates three key components: (i) Bidirectional Knowledge Attention, which removes the causal mask to enable full interaction among all tokens and entities; (ii) Knowledge-Masked Prediction, which encourages the model to leverage both local semantic contexts and global graph connectivity; and (iii) Contrastive Graph Semantic Aggregation, which preserves KG structure via contrastive alignment of sampled sub-graph representations. Extensive experiments on standard benchmarks demonstrate that KG-BiLM outperforms strong baselines in link prediction, especially on large-scale graphs with complex multi-hop relations - validating its effectiveness in unifying structural information and textual semantics.

知识图谱双向模型语义融合

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