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

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.

PDF Abstract

Datasets


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Link Property Prediction ogbl-wikikg2 InterHT+ (256dim) Validation MRR 0.7370 ± 0.0022 # 4
Test MRR 0.7257 ± 0.0018 # 5
Number of params 148000738 # 9
Ext. data No # 1
Link Property Prediction ogbl-wikikg2 InterHT+ Validation MRR 0.7391 ± 0.0023 # 3
Test MRR 0.7293 ± 0.0018 # 4
Number of params 156332770 # 10
Ext. data No # 1
Link Property Prediction ogbl-wikikg2 InterHT Validation MRR 0.6893 ± 0.0015 # 9
Test MRR 0.6779 ± 0.0018 # 9
Number of params 19215402 # 5
Ext. data No # 1

Methods


No methods listed for this paper. Add relevant methods here