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

• Published in: Data mining and knowledge discovery Vol. 38; no. 2; pp. 343 - 371
• Main Authors: He, Tao, Liu, Ming, Cao, Yixin, Qu, Meng, Zheng, Zihao, Qin, Bing
• Format: Journal Article
• Language: English
• Published: New York Springer US 01.03.2024; Springer Nature B.V
• Subjects: Algorithms; Artificial Intelligence; Chemistry and Earth Sciences; Computer Science; Data Mining and Knowledge Discovery; Distillation; Information Storage and Retrieval; Knowledge; Knowledge acquisition; Knowledge representation; Physics; Statistics for Engineering; Training; Knowledge fusion; Knowledge graph reasoning; Variational inference; Graph neural network
• ISSN: 1384-5810; 1573-756X
• DOI: 10.1007/s10618-023-01001-y

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 VEM 2 L, 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.