OMNIA: Closing the Loop by Leveraging LLMs for Knowledge Graph Completion
Not on the map. Why?
Abstract (arXiv)
Knowledge Graphs (KGs) are widely used to represent structured knowledge, yet their automatic construction, especially with Large Language Models (LLMs), often results in incomplete or noisy outputs. Knowledge Graph Completion (KGC) aims to infer and add missing triples, but most existing methods either rely on structural embeddings that overlook semantics or language models that ignore the graph's structure and depend on external sources. In this work, we present OMNIA, a two-stage approach that bridges structural and semantic reasoning for KGC. It first generates candidate triples by clustering semantically related entities and relations within the KG, then validates them through lightweight embedding filtering followed by LLM-based semantic validation. OMNIA performs on the internal KG, without external sources, and specifically targets implicit semantics that are most frequent in LLM-generated graphs. Extensive experiments on multiple datasets demonstrate that OMNIA significantly improves F1-score compared to traditional embedding-based models. These results highlight OMNIA's effectiveness and efficiency, as its clustering and filtering stages reduce both search space and validation cost while maintaining high-quality completion.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 32 of its 47 references that are in the map. First posted on arXiv 12 Mar 2026.
Builds on
32 of 47 references.
References · 32
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2023Mistral 7Bcited by 3,164
- 2017Billion-scale similarity search with GPUscited by 2,497
- 2020MPNet: Masked and Permuted Pre-training for Language Understandingcited by 981
- 2014Embedding Entities and Relations for Learning and Inference in Knowledge Basescited by 932
- 2016Complex Embeddings for Simple Link Predictioncited by 902
- 2019RotatE: Knowledge Graph Embedding by Relational Rotation in Complex Spacecited by 898
- 2017Convolutional 2D Knowledge Graph Embeddingscited by 730
- 2019KG-BERT: BERT for Knowledge Graph Completioncited by 324
- 2020CoDEx: A Comprehensive Knowledge Graph Completion Benchmarkcited by 102
- 2021COVID-Fact: Fact Extraction and Verification of Real-World Claims on COVID-19 Pandemiccited by 100
- 2023LLMs for Knowledge Graph Construction and Reasoning: Recent Capabilities and Future Opportunitiescited by 97
20 more
- 2022SimKGC: Simple Contrastive Knowledge Graph Completion with Pre-trained Language Modelscited by 91
- 2020PyKEEN 1.0: A Python Library for Training and Evaluating Knowledge Graph Embeddingscited by 77
- 2022Sequence-to-Sequence Knowledge Graph Completion and Question Answeringcited by 75
- 2017Open-World Knowledge Graph Completioncited by 64
- 2023Making Large Language Models Perform Better in Knowledge Graph Completioncited by 48
- 2022From Discrimination to Generation: Knowledge Graph Completion with Generative Transformercited by 45
- 2023Enhancing Knowledge Graph Construction Using Large Language Modelscited by 44
- 2023Exploring Large Language Models for Knowledge Graph Completioncited by 43
- 2024KICGPT: Large Language Model with Knowledge in Context for Knowledge Graph Completioncited by 36
- 2024Knowledge Graph Large Language Model (KG-LLM) for Link Predictioncited by 36
- 2022Knowledge Is Flat: A Seq2Seq Generative Framework for Various Knowledge Graph Completioncited by 32
- 2022Relational Message Passing for Fully Inductive Knowledge Graph Completioncited by 30
- 2020Reinforced Anytime Bottom Up Rule Learning for Knowledge Graph Completioncited by 24
- 2024Multi-perspective Improvement of Knowledge Graph Completion with Large Language Modelscited by 20
- 2024Contextualization Distillation from Large Language Model for Knowledge Graph Completioncited by 18
- 2024Construction and Application of Materials Knowledge Graph in Multidisciplinary Materials Science via Large Language Modelcited by 15
- 2024Start from Zero: Triple Set Prediction for Automatic Knowledge Graph Completioncited by 7
- 2020Entity Type Prediction in Knowledge Graphs using Embeddingscited by 3
- 2022Knowledge Graph Curation: A Practical Frameworkcited by 2
- 2025Ontology-Enhanced Knowledge Graph Completion using Large Language Modelscited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.