Self-Supervised Hyperboloid Representations from Logical Queries over Knowledge Graphs

TL;DR

This paper introduces HypE, a self-supervised hyperboloid embedding method for knowledge graphs that captures hierarchical and complex query operations, significantly improving reasoning and anomaly detection performance.

Contribution

The paper proposes a novel hyperboloid embedding framework, HypE, for modeling complex logical queries over knowledge graphs in a self-supervised manner, capturing hierarchical and geometric relations.

Findings

HypE outperforms state-of-the-art methods on KG reasoning tasks.

HypE achieves significant improvements in anomaly detection on e-commerce and web data.

Learned embeddings are interpretable within the Poincaré ball geometry.

Abstract

Knowledge Graphs (KGs) are ubiquitous structures for information storagein several real-world applications such as web search, e-commerce, social networks, and biology. Querying KGs remains a foundational and challenging problem due to their size and complexity. Promising approaches to tackle this problem include embedding the KG units (e.g., entities and relations) in a Euclidean space such that the query embedding contains the information relevant to its results. These approaches, however, fail to capture the hierarchical nature and semantic information of the entities present in the graph. Additionally, most of these approaches only utilize multi-hop queries (that can be modeled by simple translation operations) to learn embeddings and ignore more complex operations such as intersection and union of simpler queries. To tackle such complex operations, in this paper, we formulate KG representation learning as a self-supervised logical query reasoning problem that utilizes translation, intersection and union queries over KGs. We propose Hyperboloid Embeddings (HypE), a novel self-supervised dynamic reasoning framework, that utilizes positive first-order existential queries on a KG to learn representations of its entities and relations as hyperboloids in a Poincar\'e ball. HypE models the positive first-order queries as geometrical translation, intersection, and union. For the problem of KG reasoning in real-world datasets, the proposed HypE model significantly outperforms the state-of-the art results. We also apply HypE to an anomaly detection task on a popular e-commerce website product taxonomy as well as hierarchically organized web articles and demonstrate significant performance improvements compared to existing baseline methods. Finally, we also visualize the learned HypE embeddings in a Poincar\'e ball to clearly interpret and comprehend the representation space.