Lune

ACL Findings2022

Do Pre-trained Models Benefit Knowledge Graph Completion? A Reliable Evaluation and a Reasonable Approach

Xin Lv, Yankai Lin, Yixin Cao, Lei Hou, Juanzi Li, Zhiyuan Liu, Peng Li, Jie Zhou

Abstract

In recent years, pre-trained language models (PLMs) have been shown to capture factual knowledge from massive texts, which encourages the proposal of PLM-based knowledge graph completion (KGC) models. However, these models are still quite behind the SOTA KGC models in terms of performance. In this work, we find two main reasons for the weak performance: (1) Inaccurate evaluation setting. The evaluation setting under the closedworld assumption (CWA) may underestimate the PLM-based KGC models since they introduce more external knowledge; (2) Inappropriate utilization of PLMs. Most PLM-based KGC models simply splice the labels of entities and relations as inputs, leading to incoherent sentences that do not take full advantage of the implicit knowledge in PLMs. To alleviate these problems, we highlight a more accurate evaluation setting under the open-world assumption (OWA), which manually checks the correctness of knowledge that is not in KGs. Moreover, motivated by prompt tuning, we propose a novel PLM-based KGC model named PKGC. The basic idea is to convert each triple and its support information into natural prompt sentences, which are further fed into PLMs for classification. Experiment results on two KGC datasets demonstrate OWA is more reliable for evaluating KGC, especially on the link prediction, and the effectiveness of our PKCG model on both CWA and OWA settings.

Ask Lune

Ask anything. The answer quotes the paper.

Lune reads all of it and the papers around it, then shows you the passages behind every answer.

Questions to start from

Lune answers from

the full paper · 17 top-tier papers that cite it · related work

Get the answer

Free to start. No credit card required.

Cited by top-tier papers17

  1. Making Large Language Models Perform Better in Knowledge Graph CompletionYichi Zhang, Zhuo Chen, Lingbing Guo, Yajing Xu et al.ACM MM 2024 · 86 citations
  2. GraphCare: Enhancing Healthcare Predictions with Personalized Knowledge GraphsPengcheng Jiang, Cao Xiao, Adam Cross, Jimeng SunICLR 2024 · 77 citations
  3. KG-FIT: Knowledge Graph Fine-Tuning Upon Open-World KnowledgePengcheng Jiang, Lang Cao, Cao (Danica) Xiao, Parminder Bhatia et al.NeurIPS 2024 · 40 citations
  4. Editing Language Model-Based Knowledge Graph EmbeddingsSiyuan Cheng, Ningyu Zhang, Bozhong Tian, Xi Chen et al.AAAI 2024 · 40 citations
  5. Knowledge Graph Error Detection with Contrastive Confidence AdaptionXiangyu Liu, Yang Liu, Wei HuAAAI 2024 · 16 citations

Builds on7

  1. AutoPrompt: Eliciting Knowledge from Language Models with Automatically Generated PromptsTaylor Shin, Yasaman Razeghi, Robert L. Logan IV, Eric Wallace et al.EMNLP 2020 · 1,162 citations
  2. LUKE: Deep Contextualized Entity Representations with Entity-aware Self-attentionIkuya Yamada, Akari Asai, Hiroyuki Shindo, Hideaki Takeda et al.EMNLP 2020 · 562 citations
  3. Realistic Re-evaluation of Knowledge Graph Completion Methods: An Experimental StudyFarahnaz Akrami, Mohammed Samiul Saeef, Qingheng Zhang, Wei Hu et al.SIGMOD 2020 · 101 citations
  4. CoDEx: A Comprehensive Knowledge Graph Completion BenchmarkTara Safavi, Danai KoutraEMNLP 2020 · 97 citations
  5. OKGIT: Open Knowledge Graph Link Prediction with Implicit TypesChandrahas, Partha P. TalukdarACL Findings 2021 · 14 citations

Related papers

  1. Text Augmented Open Knowledge Graph Completion via Pre-Trained Language ModelsPengcheng Jiang, Shivam Agarwal, Bowen Jin, Xuan Wang et al.ACL Findings 2023 · 20 citations
  2. Path-enhanced Pre-trained Language Model for Knowledge Graph CompletionHao Wang, Dandan Song, Zhijing Wu, Yuhang Tian et al.EMNLP Findings 2025
  3. Pre-trained Language Model with Prompts for Temporal Knowledge Graph CompletionWenjie Xu, Ben Liu, Miao Peng, Xu Jia et al.ACL Findings 2023 · 53 citations
  4. Knowledge Context Modeling with Pre-trained Language Models for Contrastive Knowledge Graph CompletionGuangqian Yang, Yi Liu, Lei Zhang, Licheng Zhang et al.ACL Findings 2024 · 9 citations
  5. MusKGC: A Flexible Multi-source Knowledge Enhancement Framework for Open-World Knowledge Graph CompletionXin Song, Haiyan Liu, Haiyang Wang, Ye Wang et al.EMNLP 2025
Dusk over the sea between two cliffs drawn in fine vertical lines