Geometric Structural Knowledge Graph Foundation Model
By: Ling Xin , Mojtaba Nayyeri , Zahra Makki Nayeri and more
Structural knowledge graph foundation models aim to generalize reasoning to completely new graphs with unseen entities and relations. A key limitation of existing approaches like Ultra is their reliance on a single relational transformation (e.g., element-wise multiplication) in message passing, which can constrain expressiveness and fail to capture diverse relational and structural patterns exhibited on diverse graphs. In this paper, we propose Gamma, a novel foundation model that introduces multi-head geometric attention to knowledge graph reasoning. Gamma replaces the single relational transformation with multiple parallel ones, including real, complex, split-complex, and dual number based transformations, each designed to model different relational structures. A relational conditioned attention fusion mechanism then adaptively fuses them at link level via a lightweight gating with entropy regularization, allowing the model to robustly emphasize the most appropriate relational bias for each triple pattern. We present a full formalization of these algebraic message functions and discuss how their combination increases expressiveness beyond any single space. Comprehensive experiments on 56 diverse knowledge graphs demonstrate that Gamma consistently outperforms Ultra in zero-shot inductive link prediction, with a 5.5% improvement in mean reciprocal rank on the inductive benchmarks and a 4.4% improvement across all benchmarks, highlighting benefits from complementary geometric representations.
Similar Papers
Knowledge Graph-Infused Fine-Tuning for Structured Reasoning in Large Language Models
Computation and Language
Helps computers understand facts and connect ideas.
RiemannGFM: Learning a Graph Foundation Model from Riemannian Geometry
Machine Learning (CS)
Teaches computers to understand any graph's structure.
Why Relational Graphs Will Save the Next Generation of Vision Foundation Models?
CV and Pattern Recognition
Helps computers understand how things relate and move.