Skip to main navigation Skip to search Skip to main content

VEM 2 L: an easy but effective framework for fusing text and structure knowledge on sparse knowledge graph completion

  • Tao He
  • , Ming Liu*
  • , Yixin Cao
  • , Meng Qu
  • , Zihao Zheng
  • , Bing Qin
  • *Corresponding author for this work
  • Harbin Institute of Technology
  • Peng Cheng Laboratory
  • Singapore Management University
  • Mila - Quebec AI Institute

Research output: Contribution to journalArticlepeer-review

Abstract

The task of Knowledge Graph Completion (KGC) is to infer missing links for Knowledge Graphs (KGs) by analyzing graph structures. However, with increasing sparsity in KGs, this task becomes increasingly challenging. In this paper, we propose VEM2L, a joint learning framework that incorporates structure and relevant text information to supplement insufficient features for sparse KGs. We begin by training two pre-existing KGC models: one based on structure and the other based on text. Our ultimate goal is to fuse knowledge acquired by these models. To achieve this, we divide knowledge within the models into two non-overlapping parts: expressive power and generalization ability. We then propose two different joint learning methods that co-distill these two kinds of knowledge respectively. For expressive power, we allow each model to learn from and exchange knowledge mutually on training examples. For the generalization ability, we propose a novel co-distillation strategy using the Variational EM algorithm on unobserved queries. Our proposed joint learning framework is supported by both detailed theoretical evidence and qualitative experiments, demonstrating its effectiveness.

Original languageEnglish
Pages (from-to)343-371
Number of pages29
JournalData Mining and Knowledge Discovery
Volume38
Issue number2
DOIs
StatePublished - Mar 2024

Keywords

  • Graph neural network
  • Knowledge fusion
  • Knowledge graph reasoning
  • Variational inference

Fingerprint

Dive into the research topics of 'VEM 2 L: an easy but effective framework for fusing text and structure knowledge on sparse knowledge graph completion'. Together they form a unique fingerprint.

Cite this