A Relation Is a Rotation On a Knowledge Graph
Summary
The `knowlytix` tool implements a geometric memory system where knowledge graph relations are represented as rotations in a 32-dimensional real vector space. This system, demonstrated on a bank's compliance policy, allows AI agents to accurately assess fact plausibility and workflow steps. For instance, an overdraft fee of "35.0" scores a distance of 0.052, while a wrong fee like "45.0" scores 1.527, effectively distinguishing correct from incorrect information. The system handles 40 entities, 8 relations, and 41 triples, ensuring relations are genuine rotations that preserve length and have inverses. Crucially, it supports non-commutative relation composition, essential for complex reasoning chains like escalation paths, and assigns an "admissible radius" to each relation to define its strictness.
Key takeaway
For AI Engineers developing agents for regulated environments, integrating `knowlytix` can address the "prompt and pray" problem by providing a robust geometric memory. This system ensures factual accuracy and workflow compliance by validating claims against policy with calibrated distances and admissible radii. You can move beyond simple string lookups to a memory that flags inconsistencies and provides definitive yes-or-no answers, crucial for agent reliability in sensitive applications like banking compliance.
Key insights
Knowledge graph relations can be modeled as rotations in high-dimensional space, scoring fact plausibility by geodesic distance.
Principles
- Relations as rotations preserve length and have clean inverses.
- High-dimensional rotations enable non-commutative relation composition.
- Admissible radii define relation strictness for fact validation.
Method
`knowlytix` parses policy documents into typed triples, trains a geometry placing entities on a unit sphere, and makes each relation a rotation (rotor) to score fact plausibility via geodesic distance.
In practice
- Use `knowlytix.score_triple` to validate facts.
- Apply `knowlytix.link_predict` for ranking candidate tails.
- Employ `check_holonomy` to detect reasoning inconsistencies.
Topics
- Knowledge Graphs
- Geometric Embeddings
- Relation Learning
- AI Agent Memory
- Compliance Automation
- knowlytix
Code references
Best for: CTO, VP of Engineering/Data, Director of AI/ML, AI Scientist, Machine Learning Engineer, AI Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by Agus’s Substack.