论文标题
具有快速傅立叶变换的复杂双曲知识图嵌入
Complex Hyperbolic Knowledge Graph Embeddings with Fast Fourier Transform
论文作者
论文摘要
知识图(KG)嵌入的几何空间的选择会对KG完成任务的性能产生重大影响。双曲线几何形状已被证明可以捕获其类似树状的指标引起的层次模式,从而解决了欧几里得嵌入模型的局限性。对复杂双曲几何形状的最新探索进一步改善了捕获各种分层结构的双曲线嵌入。但是,非传播关系的双曲线嵌入模型的性能仍然没有主张,而复杂的双曲线嵌入并不涉及多重关系。本文旨在利用多关系KG嵌入中复杂双曲几何形状的表示能力。为了应用复杂双曲线空间中不同关系和注意力机制的几何变换,我们建议将快速傅立叶变换(FFT)用作真实和复杂双曲线空间之间的转换。在复杂空间中构建基于注意力的转换非常具有挑战性,而所提出的基于傅立叶变换的复杂双曲线方法则提供了一种简单有效的解决方案。实验结果表明,我们的方法的表现优于基准,包括欧几里得和真正的双曲线嵌入模型。
The choice of geometric space for knowledge graph (KG) embeddings can have significant effects on the performance of KG completion tasks. The hyperbolic geometry has been shown to capture the hierarchical patterns due to its tree-like metrics, which addressed the limitations of the Euclidean embedding models. Recent explorations of the complex hyperbolic geometry further improved the hyperbolic embeddings for capturing a variety of hierarchical structures. However, the performance of the hyperbolic KG embedding models for non-transitive relations is still unpromising, while the complex hyperbolic embeddings do not deal with multi-relations. This paper aims to utilize the representation capacity of the complex hyperbolic geometry in multi-relational KG embeddings. To apply the geometric transformations which account for different relations and the attention mechanism in the complex hyperbolic space, we propose to use the fast Fourier transform (FFT) as the conversion between the real and complex hyperbolic space. Constructing the attention-based transformations in the complex space is very challenging, while the proposed Fourier transform-based complex hyperbolic approaches provide a simple and effective solution. Experimental results show that our methods outperform the baselines, including the Euclidean and the real hyperbolic embedding models.