English

Knowledge Graph Embedding with 3D Compound Geometric Transformations

Artificial Intelligence 2023-04-04 v1 Machine Learning

Abstract

The cascade of 2D geometric transformations were exploited to model relations between entities in a knowledge graph (KG), leading to an effective KG embedding (KGE) model, CompoundE. Furthermore, the rotation in the 3D space was proposed as a new KGE model, Rotate3D, by leveraging its non-commutative property. Inspired by CompoundE and Rotate3D, we leverage 3D compound geometric transformations, including translation, rotation, scaling, reflection, and shear and propose a family of KGE models, named CompoundE3D, in this work. CompoundE3D allows multiple design variants to match rich underlying characteristics of a KG. Since each variant has its own advantages on a subset of relations, an ensemble of multiple variants can yield superior performance. The effectiveness and flexibility of CompoundE3D are experimentally verified on four popular link prediction datasets.

Keywords

Cite

@article{arxiv.2304.00378,
  title  = {Knowledge Graph Embedding with 3D Compound Geometric Transformations},
  author = {Xiou Ge and Yun-Cheng Wang and Bin Wang and C. -C. Jay Kuo},
  journal= {arXiv preprint arXiv:2304.00378},
  year   = {2023}
}
R2 v1 2026-06-28T09:44:46.871Z