InterHT: Knowledge Graph Embeddings by Interaction between Head and Tail Entities

Baoxin Wang, Qingye Meng, Ziyue Wang, Dayong Wu, Wanxiang Che, Shijin Wang, Zhigang Chen, Cong Liu

Main figure for InterHT: Knowledge Graph Embeddings by Interaction between Head and Tail Entities

Abstract

Knowledge graph embedding (KGE) models learn the representation of entities and relations in knowledge graphs. Distance-based methods show promising performance on link prediction task, which predicts the result by the distance between two entity representations. However, most of these methods represent the head entity and tail entity separately, which limits the model capacity. We propose two novel distance-based methods named InterHT and InterHT+ that allow the head and tail entities to interact better and get better entity representation. Experimental results show that our proposed method achieves the best results on ogbl-wikikg2 dataset.

Benchmark Results

Tested on 1 dataset

Tables show presented results in global context (best 3 papers + GCN + MLP).

OGB

#ModelMRR ↑Paper
1
InterHT + DigPiece
0.7293 ±0.0018
This paper
2
StarGraph + TripleRE'
0.7290
3
TripleRE + StarGraph
0.7201 ±0.0011
This paper
5
InterHT + NodePiece
0.6779 ±0.0018
This paper
7
TripleRE + NodePiece
0.6582 ±0.0020
This paper
11
ComplEx-RP (50dim)
0.6392 ±0.0045
This paper
21
TripleRE
0.5794 ±0.0020
This paper
24
AutoSF + NodePiece
0.5703 ±0.0035
This paper
25
AutoSF
0.5458 ±0.0052
This paper
26
PairRE (200dim)
0.5208 ±0.0027
This paper
27
ComplEx (250dim)
0.5027 ±0.0027
This paper
31
RotatE (250dim)
0.4332 ±0.0025
This paper
33
TransE (500dim)
0.4256 ±0.0030
This paper
SOTA performance (not significantly different from the best) Proposed method (this paper) Baseline / comparison (this paper) Global context (other papers)