Knowledge Sheaves: A Sheaf-Theoretic Framework for Knowledge Graph Embedding

Abstract

Knowledge graph embedding involves learning representations of entities -- the vertices of the graph -- and relations -- the edges of the graph -- such that the resulting representations encode the known factual information represented by the knowledge graph and can be used in the inference of new relations. We show that knowledge graph embedding is naturally expressed in the topological and categorical language of \textit{cellular sheaves}: a knowledge graph embedding can be described as an approximate global section of an appropriate \textit{knowledge sheaf} over the graph, with consistency constraints induced by the knowledge graph's schema. This approach provides a generalized framework for reasoning about knowledge graph embedding models and allows for the expression of a wide range of prior constraints on embeddings. Further, the resulting embeddings can be easily adapted for reasoning over composite relations without special training. We implement these ideas to highlight the benefits of the extensions inspired by this new perspective.

AISTATS 2023

References in corpus (9)

• RotatE: Knowledge Graph Embedding by Relational Rotation in Complex Space
• Quaternion Knowledge Graph Embeddings
• Bringing Light Into the Dark: A Large-scale Evaluation of Knowledge Graph Embedding Models Under a Unified Framework
• PyKEEN 1.0: A Python Library for Training and Evaluating Knowledge Graph Embeddings
• Query2box: Reasoning over Knowledge Graphs in Vector Space using Box Embeddings
• Seven Sketches in Compositionality: An Invitation to Applied Category Theory
• Understanding Double Descent Requires a Fine-Grained Bias-Variance Decomposition
• Complex and Holographic Embeddings of Knowledge Graphs: A Comparison
• Knowledge Representation in Bicategories of Relations