KG-CF: Knowledge Graph Completion with Context Filtering under the Guidance of Large Language Models
Large Language Models (LLMs) have shown impressive performance in various tasks, including knowledge graph completion (KGC). However, current studies mostly apply LLMs to classification tasks, like identifying missing triplets, rather than ranking-based tasks, where the model ranks candidate entities based on plausibility. This focus limits the practical use of LLMs in KGC, as real-world applications prioritize highly plausible triplets. Additionally, while graph paths can help infer the existence of missing triplets and improve completion accuracy, they often contain redundant information. To address these issues, we propose KG-CF, a framework tailored for ranking-based KGC tasks. KG-CF leverages LLMs’ reasoning abilities to filter out irrelevant contexts, achieving superior results on real-world datasets. The code and datasets are available at https://anonymous.4open.science/r/KG-CF.
Authors
- Jundong Li (ORCID: https://orcid.org/0000-0002-1878-817X)
- Haochen Liu (ORCID: https://orcid.org/0009-0006-7515-2166)
- Qi Wang (ORCID: https://orcid.org/0000-0003-3249-8459)
- Zaiyi Zheng (ORCID: https://orcid.org/0009-0003-0685-0057)
- Song Wang (ORCID: https://orcid.org/0000-0003-1273-7694)
- Yushun Dong
Institutions
- Florida State University (US)
- Boston University (US)
- Northeastern University (US)
- University of Virginia (US)
Publication Details
- Published
- 2024-12-15
- DOI
- https://doi.org/10.1109/bigdata62323.2024.10826107
- Citations
- 1
- Primary Topic
- Advanced Graph Neural Networks
- Type
- article
- Field-Weighted Citation Impact
- 0.35
Funders
- National Science Foundation