mlsanityData to 7 Oct 2026

Pre-training and Diagnosing Knowledge Base Completion Models

arXiv 2401.15439 ↗First posted 27 Jan 2024cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

In this work, we introduce and analyze an approach to knowledge transfer from one collection of facts to another without the need for entity or relation matching. The method works for both canonicalized knowledge bases and uncanonicalized or open knowledge bases, i.e., knowledge bases where more than one copy of a real-world entity or relation may exist. The main contribution is a method that can make use of large-scale pre-training on facts, which were collected from unstructured text, to improve predictions on structured data from a specific domain. The introduced method is most impactful on small datasets such as ReVerb20k, where a 6% absolute increase of mean reciprocal rank and 65% relative decrease of mean rank over the previously best method was achieved, despite not relying on large pre-trained models like Bert. To understand the obtained pre-trained models better, we then introduce a novel dataset for the analysis of pre-trained models for Open Knowledge Base Completion, called Doge (Diagnostics of Open knowledge Graph Embeddings). It consists of 6 subsets and is designed to measure multiple properties of a pre-trained model: robustness against synonyms, ability to perform deductive reasoning, presence of gender stereotypes, consistency with reverse relations, and coverage of different areas of general knowledge. Using the introduced dataset, we show that the existing OKBC models lack consistency in the presence of synonyms and inverse relations and are unable to perform deductive reasoning. Moreover, their predictions often align with gender stereotypes, which persist even when presented with counterevidence. We additionally investigate the role of pre-trained word embeddings and demonstrate that avoiding biased word embeddings is not a sufficient measure to prevent biased behavior of OKBC models.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 27 of its 49 references that are in the map. First posted on arXiv 27 Jan 2024.

Builds on

27 of 49 references.

References · 27

  1. 2020Language Models are Few-Shot Learnerscited by 32,199
  2. 2017Graph Attention Networkscited by 8,854
  3. 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,016
  4. 2019Language Models as Knowledge Bases?cited by 1,922
  5. 2016Man is to Computer Programmer as Woman is to Homemaker? Debiasing Word Embeddingscited by 1,304
  6. 2013Spectral Networks and Locally Connected Networks on Graphscited by 1,097
  7. 2019RotatE: Knowledge Graph Embedding by Relational Rotation in Complex Spacecited by 893
  8. 2017Convolutional 2D Knowledge Graph Embeddingscited by 729
  9. 2018Gender Bias in Coreference Resolution: Evaluation and Debiasing Methodscited by 657
  10. 2020Pre-trained Models for Natural Language Processing: A Surveycited by 504
  11. 2018Gender Bias in Coreference Resolutioncited by 438
  12. 2017Supervised Learning of Universal Sentence Representations from Natural Language Inference Datacited by 428
15 more
  1. 2019KG-BERT: BERT for Knowledge Graph Completioncited by 323
  2. 2018Canonical Tensor Decomposition for Knowledge Base Completioncited by 150
  3. 2019PyTorch-BigGraph: A Large-scale Graph Embedding Systemcited by 118
  4. 2020BoxE: A Box Embedding Model for Knowledge Base Completioncited by 105
  5. 2020Pre-training via Paraphrasingcited by 92
  6. 2017Knowledge Graph Completion via Complex Tensor Factorizationcited by 80
  7. 2020Bringing Light Into the Dark: A Large-scale Evaluation of Knowledge Graph Embedding Models Under a Unified Frameworkcited by 70
  8. 2015Modeling Relation Paths for Representation Learning of Knowledge Basescited by 56
  9. 2019CESI: Canonicalizing Open Knowledge Bases using Embeddings and Side Informationcited by 28
  10. 2020Duality-Induced Regularizer for Tensor Factorization Based Knowledge Graph Completioncited by 28
  11. 2020MLMLM: Link Prediction with Mean Likelihood Masked Language Modelcited by 22
  12. 20205* Knowledge Graph Embeddings with Projective Transformationscited by 19
  13. 2021OKGIT: Open Knowledge Graph Link Prediction with Implicit Typescited by 6
  14. 2021Knowledge Base Completion Meets Transfer Learningcited by 4
  15. 2016Improved Knowledge Base Completion by Path-Augmented TransR Modelcited by 1

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2022Reasoning Through Memorization: Nearest Neighbor Knowledge Graph Embeddings
  2. 2024Open Knowledge Base Canonicalization with Multi-task Learning
  3. 2026Conditional Diffusion Guided Knowledge Transfer for Multi-Domain Knowledge Graph Completion
  4. 2024MUSE: Multi-Knowledge Passing on the Edges, Boosting Knowledge Graph Completion
  5. 2023Knowledge Graphs are not Created Equal: Exploring the Properties and Structure of Real KGs
  6. 2023Open Knowledge Base Canonicalization with Multi-task Unlearning
  7. 2026Predicate Importance Estimation and Decoupled Rationale-Score Distillation for Entity Alignment
  8. 2022LinkBERT: Pretraining Language Models with Document Links