Contextual Dictionary Lookup for Knowledge Graph Completion
This work addresses the incompleteness of knowledge graphs for applications relying on accurate link prediction, offering an incremental improvement by enhancing existing embedding models with fine-grained semantics.
The paper tackles the problem of knowledge graph completion by addressing the limitation of existing embedding models that overlook fine-grained semantics of relations under different entities, proposing a contextual dictionary lookup method that enables conventional models to learn these semantics end-to-end and achieves substantial performance improvements on benchmark datasets.
Knowledge graph completion (KGC) aims to solve the incompleteness of knowledge graphs (KGs) by predicting missing links from known triples, numbers of knowledge graph embedding (KGE) models have been proposed to perform KGC by learning embeddings. Nevertheless, most existing embedding models map each relation into a unique vector, overlooking the specific fine-grained semantics of them under different entities. Additionally, the few available fine-grained semantic models rely on clustering algorithms, resulting in limited performance and applicability due to the cumbersome two-stage training process. In this paper, we present a novel method utilizing contextual dictionary lookup, enabling conventional embedding models to learn fine-grained semantics of relations in an end-to-end manner. More specifically, we represent each relation using a dictionary that contains multiple latent semantics. The composition of a given entity and the dictionary's central semantics serves as the context for generating a lookup, thus determining the fine-grained semantics of the relation adaptively. The proposed loss function optimizes both the central and fine-grained semantics simultaneously to ensure their semantic consistency. Besides, we introduce two metrics to assess the validity and accuracy of the dictionary lookup operation. We extend several KGE models with the method, resulting in substantial performance improvements on widely-used benchmark datasets.