In-Context Learning with Topological Information for Knowledge Graph Completion

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Sehwag, Udari Madhushani, Papasotiriou, Kassiani, Vann, Jared, Ganesh, Sumitra
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909425242996736
author Sehwag, Udari Madhushani
Papasotiriou, Kassiani
Vann, Jared
Ganesh, Sumitra
author_facet Sehwag, Udari Madhushani
Papasotiriou, Kassiani
Vann, Jared
Ganesh, Sumitra
contents Knowledge graphs (KGs) are crucial for representing and reasoning over structured information, supporting a wide range of applications such as information retrieval, question answering, and decision-making. However, their effectiveness is often hindered by incompleteness, limiting their potential for real-world impact. While knowledge graph completion (KGC) has been extensively studied in the literature, recent advances in generative AI models, particularly large language models (LLMs), have introduced new opportunities for innovation. In-context learning has recently emerged as a promising approach for leveraging pretrained knowledge of LLMs across a range of natural language processing tasks and has been widely adopted in both academia and industry. However, how to utilize in-context learning for effective KGC remains relatively underexplored. We develop a novel method that incorporates topological information through in-context learning to enhance KGC performance. By integrating ontological knowledge and graph structure into the context of LLMs, our approach achieves strong performance in the transductive setting i.e., nodes in the test graph dataset are present in the training graph dataset. Furthermore, we apply our approach to KGC in the more challenging inductive setting, i.e., nodes in the training graph dataset and test graph dataset are disjoint, leveraging the ontology to infer useful information about missing nodes which serve as contextual cues for the LLM during inference. Our method demonstrates superior performance compared to baselines on the ILPC-small and ILPC-large datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2412_08742
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle In-Context Learning with Topological Information for Knowledge Graph Completion
Sehwag, Udari Madhushani
Papasotiriou, Kassiani
Vann, Jared
Ganesh, Sumitra
Computation and Language
Artificial Intelligence
68T37 (Primary), 68T05, 68P20 (Secondary)
Knowledge graphs (KGs) are crucial for representing and reasoning over structured information, supporting a wide range of applications such as information retrieval, question answering, and decision-making. However, their effectiveness is often hindered by incompleteness, limiting their potential for real-world impact. While knowledge graph completion (KGC) has been extensively studied in the literature, recent advances in generative AI models, particularly large language models (LLMs), have introduced new opportunities for innovation. In-context learning has recently emerged as a promising approach for leveraging pretrained knowledge of LLMs across a range of natural language processing tasks and has been widely adopted in both academia and industry. However, how to utilize in-context learning for effective KGC remains relatively underexplored. We develop a novel method that incorporates topological information through in-context learning to enhance KGC performance. By integrating ontological knowledge and graph structure into the context of LLMs, our approach achieves strong performance in the transductive setting i.e., nodes in the test graph dataset are present in the training graph dataset. Furthermore, we apply our approach to KGC in the more challenging inductive setting, i.e., nodes in the training graph dataset and test graph dataset are disjoint, leveraging the ontology to infer useful information about missing nodes which serve as contextual cues for the LLM during inference. Our method demonstrates superior performance compared to baselines on the ILPC-small and ILPC-large datasets.
title In-Context Learning with Topological Information for Knowledge Graph Completion
topic Computation and Language
Artificial Intelligence
68T37 (Primary), 68T05, 68P20 (Secondary)
url https://arxiv.org/abs/2412.08742