Capturing Knowledge Graphs and Rules with Octagon Embeddings
Victor Charpenay, Steven Schockaert
TL;DR
This work addresses the need for region-based knowledge graph embeddings that support relational composition and explicit rule capture. It introduces axis-aligned octagons as the building blocks in a coordinate-wise region model, proving that regions are closed under intersection and composition and that octagon parameters can be normalised for stability. The authors prove that octagon embeddings can represent arbitrary knowledge graphs and can capture a wide class of rules, while identifying theoretical limits (e.g., hexagons lack expressive power for composition). Empirically, octagon embeddings achieve competitive link prediction performance on standard benchmarks, with ablations showing the importance of the $v$-constraint and attention; the work also discusses practical learning with soft boundaries and potential extensions including cross-coordinate comparisons to broaden rule coverage.
Abstract
Region based knowledge graph embeddings represent relations as geometric regions. This has the advantage that the rules which are captured by the model are made explicit, making it straightforward to incorporate prior knowledge and to inspect learned models. Unfortunately, existing approaches are severely restricted in their ability to model relational composition, and hence also their ability to model rules, thus failing to deliver on the main promise of region based models. With the aim of addressing these limitations, we investigate regions which are composed of axis-aligned octagons. Such octagons are particularly easy to work with, as intersections and compositions can be straightforwardly computed, while they are still sufficiently expressive to model arbitrary knowledge graphs. Among others, we also show that our octagon embeddings can properly capture a non-trivial class of rule bases. Finally, we show that our model achieves competitive experimental results.
