Entity Similarity-Based Negative Sampling for Knowledge Graph Embedding

Naimeng Yao,Qing Liu,Xiang Li,Yi Yang,Quan Bai
DOI: https://doi.org/10.1007/978-3-031-20865-2_6
2022-01-01
Abstract:Knowledge graph embedding (KGE) models optimize loss functions to maximize the total plausibility of positive triples and minimize the plausibility of negative triples. Negative samples are essential in KGE training since they are not as observable as positive samples. Currently, most negative sampling methods apply different techniques to keep track of negative samples with high scores that are regarded as quality negative samples. While, we found entities with similar semantic contexts are easier to be deceptive and misclassified, contributing to quality negative samples. This is not considered in most negative sampling approaches. Besides, the unequal effectiveness of quality negative samples in different loss functions is usually ignored. In this paper, we propose an Entity Similarity-based Negative Sampling framework (ESNS). The framework takes semantic similarities among entities into consideration with a shift-based logistic loss function. Comprehensive experiments on the five benchmark datasets have been conducted, and the experimental results demonstrate that ESNS outperforms the state-of-the-art negative sampling methods in the link prediction task.
What problem does this paper attempt to address?