TuckerDNCaching: high-quality negative sampling with tucker decomposition

Knowledge Graph Embedding (KGE) translates entities and relations of knowledge graphs (KGs) into a low-dimensional vector space, enabling an efficient way of predicting missing facts. Generally, KGE models are trained with positive and negative examples, discriminating positives against negatives. Nevertheless, KGs contain only positive facts; KGE training requires generating negatives from non-observed ones in KGs, referred to as negative sampling. Since KGE models are sensitive to inputs, negative sampling becomes crucial, and the quality of the negatives becomes critical in KGE training. Generative adversarial networks (GAN) and self-adversarial methods have recently been utilized in negative sampling to address the vanishing gradients observed with early negative sampling methods. However, they introduce the problem of false negatives with high probability. In this paper, we extend the idea of reducing false negatives by adopting a Tucker decomposition representation, i.e., TuckerDNCaching, to enhance the semantic soundness of latent relations among entities by introducing a relation feature space. TuckerDNCaching ensures the quality of generated negative samples, and the experimental results reflect that our proposed negative sampling method outperforms the existing state-of-the-art negative sampling methods.

PDF Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Benchmark
Link Prediction FB15k TuckerDNCaching + ComplEx MR 121 # 11
MRR 0.9363 # 1
Hits@10 0.9505 # 1
Link Prediction FB15k TuckerDNCaching + TransE MR 51 # 6
MRR 0.8477 # 4
Hits@10 0.9245 # 2
Link Prediction WN18 TuckerDNCaching + ComplEx MRR 0.9551 # 2
Hits@10 0.9575 # 13
MR 790 # 19
Link Prediction WN18 TuckerDNCaching + TransE MRR 0.8084 # 29
Hits@10 0.9506 # 22
MR 454 # 17
Link Prediction WN18RR TuckerDNCaching + TransE MRR 0.2563 # 68
Hits@10 0.5383 # 54
Hits@3 0.4314 # 46
Hits@1 0.0571 # 60
MR 3062 # 21
Link Prediction WN18RR TuckerDNCaching +. ComplEx MRR 0.4961 # 13
Hits@10 0.5227 # 61
Hits@3 0.5034 # 27
Hits@1 0.4817 # 6
MR 8792 # 30

Methods