Learning Attention-based Representations from Multiple Patterns for Relation Prediction in Knowledge Graphs

TL;DR

This paper introduces { ext{ extgreek{A}}}MP, an attention-based model that enhances relation prediction in knowledge graphs by capturing local and semantic context through message passing and path analysis, outperforming existing methods.

Contribution

The paper presents a novel attention-based approach that integrates local neighborhood and path-based semantic information for improved relation prediction in knowledge graphs.

Findings

{ ext{ extgreek{A}}}MP outperforms or matches state-of-the-art methods on multiple benchmarks.

Attention mechanisms effectively enhance context representation.

Combining entity and path semantics improves relation prediction accuracy.

Abstract

Knowledge bases, and their representations in the form of knowledge graphs (KGs), are naturally incomplete. Since scientific and industrial applications have extensively adopted them, there is a high demand for solutions that complete their information. Several recent works tackle this challenge by learning embeddings for entities and relations, then employing them to predict new relations among the entities. Despite their aggrandizement, most of those methods focus only on the local neighbors of a relation to learn the embeddings. As a result, they may fail to capture the KGs' context information by neglecting long-term dependencies and the propagation of entities' semantics. In this manuscript, we propose {\AE}MP (Attention-based Embeddings from Multiple Patterns), a novel model for learning contextualized representations by: (i) acquiring entities' context information through an attention-enhanced message-passing scheme, which captures the entities' local semantics while focusing on different aspects of their neighborhood; and (ii) capturing the semantic context, by leveraging the paths and their relationships between entities. Our empirical findings draw insights into how attention mechanisms can improve entities' context representation and how combining entities and semantic path contexts improves the general representation of entities and the relation predictions. Experimental results on several large and small knowledge graph benchmarks show that {\AE}MP either outperforms or competes with state-of-the-art relation prediction methods.