arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.22698cs.CLcs.AI
收藏

分词、融合与解耦:弥合大语言模型与知识图谱之间的粒度不匹配

Tokenization, Fusion and Decoupling: Bridging the Granularity Mismatch Between Large Language Models and Knowledge Graphs

Beihang University(北航大学)

Siyue Su, Jian Yang, Bo Li, Guanglin Niu

AI总结:

KGT通过专门的实体token分词、融合预训练特征及解耦预测,有效弥合大语言模型与知识图谱间的粒度不匹配,提升知识图谱补全性能。

摘要:

利用大语言模型(LLMs)进行知识图谱补全(KGC)具有前景,但受到根本性的粒度不匹配的阻碍。LLMs操作于碎片化的token序列,而实体是知识图谱(KGs)场景中的基本单元。现有方法通常限制预测到有限的候选集或通过池化多个token或分解实体为固定长度的token序列来对齐实体与LLM的词汇,这未能捕捉到文本的语义含义和图结构的完整性。为此,我们提出了KGT,一种新的框架,使用专门的实体token来实现高效、全空间预测。具体而言,我们首先引入专门的分词方法来构建在专门实体token层面的特征表示。然后通过关系引导的门控机制将预训练的结构和文本特征融合到这些统一的嵌入中,避免从头开始训练。最后,我们通过独立的头部实现解耦预测,通过分离和结合语义和结构推理。实验结果表明,KGT在多个基准上均优于最先进的方法。

英文摘要:

Leveraging Large Language Models (LLMs) for Knowledge Graph Completion (KGC) is promising but hindered by a fundamental granularity mismatch. LLMs operate on fragmented token sequences, whereas entities are the fundamental units in knowledge graphs (KGs) scenarios. Existing approaches typically constrain predictions to limited candidate sets or align entities with the LLM's vocabulary by pooling multiple tokens or decomposing entities into fixed-length token sequences, which fail to capture both the semantic meaning of the text and the structural integrity of the graph. To address this, we propose KGT, a novel framework that uses dedicated entity tokens to enable efficient, full-space prediction. Specifically, we first introduce specialized tokenization to construct feature representations at the level of dedicated entity tokens. We then fuse pre-trained structural and textual features into these unified embeddings via a relation-guided gating mechanism, avoiding training from scratch. Finally, we implement decoupled prediction by leveraging independent heads to separate and combine semantic and structural reasoning. Experimental results show that KGT consistently outperforms state-of-the-art methods across multiple benchmarks.

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

↑