Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Generalizing Knowledge Graph Embedding with Universal Orthogonal Parameterization

About

Recent advances in knowledge graph embedding (KGE) rely on Euclidean/hyperbolic orthogonal relation transformations to model intrinsic logical patterns and topological structures. However, existing approaches are confined to rigid relational orthogonalization with restricted dimension and homogeneous geometry, leading to deficient modeling capability. In this work, we move beyond these approaches in terms of both dimension and geometry by introducing a powerful framework named GoldE, which features a universal orthogonal parameterization based on a generalized form of Householder reflection. Such parameterization can naturally achieve dimensional extension and geometric unification with theoretical guarantees, enabling our framework to simultaneously capture crucial logical patterns and inherent topological heterogeneity of knowledge graphs. Empirically, GoldE achieves state-of-the-art performance on three standard benchmarks. Codes are available at https://github.com/xxrep/GoldE.

Rui Li, Chaozhuo Li, Yanming Shen, Zeyu Zhang, Xu Chen• 2024

Related benchmarks

TaskDatasetResultRank
Link PredictionFB15k-237 (test)
Hits@1052
419
Link PredictionWN18RR (test)
Hits@1060.7
380
Knowledge Base CompletionWebQSP 30% KB
MRR0.162
16
Knowledge Base CompletionWebQSP 50% KB
MRR20.7
16
Knowledge Base CompletionCWQ (30% KB)
MRR16.3
16
Knowledge Base CompletionCWQ 50% KB
MRR20
16
Showing 6 of 6 rows

Other info

Follow for update