Persistent Identifier
|
doi:10.11588/data/YYULL2 |
Publication Date
|
2019-08-19 |
Title
| Negative Sampling for Learning Knowledge Graph Embeddings |
Subtitle
| Analysis of the Impact of Negative Sampling on Link Prediction in Knowledge Graphs |
Alternative URL
| https://github.com/bhushank/kge-rl |
Author
| Kotnis, Bhushan (Department of Computational Linguistics, Heidelberg University, Germany (2016-2018), NEC Laboratories Europe GmbH (since 2018)) |
Point of Contact
|
Use email button above to contact.
Kotnis, Bhushan (NEC Laboratories Europe GmbH) |
Description
| Reimplementation of four KG factorization methods and six negative sampling methods. Abstract
Knowledge graphs are large, useful, but incomplete knowledge repositories. They encode knowledge through entities and relations which define each other through the connective structure of the graph. This has inspired methods for the joint embedding of entities and relations in continuous low-dimensional vector spaces, that can be used to induce new edges in the graph, i.e., link prediction in knowledge graphs. Learning these representations relies on contrasting positive instances with negative ones. Knowledge graphs include only positive relation instances, leaving the door open for a variety of methods for selecting negative examples. In this paper we present an empirical study on the impact of negative sampling on the learned embeddings, assessed through the task of link prediction. We use state-of-the-art knowledge graph embeddings -- \rescal , TransE, DistMult and ComplEX -- and evaluate on benchmark datasets -- FB15k and WN18. We compare well known methods for negative sampling and additionally propose embedding based sampling methods. We note a marked difference in the impact of these sampling methods on the two datasets, with the "traditional" corrupting positives method leading to best results on WN18, while embedding based methods benefiting the task on FB15k. |
Subject
| Arts and Humanities; Computer and Information Science |
Keyword
| knowledge graphs
negative sampling
embedding models
linkprediction |
Topic Classification
| knowledge discovery in knowledge graphs |
Related Publication
| Bhushan Kotnis and Vivi Nastase. 2018. Analysis of the Impact of Negative Sampling on Link Prediction in Knowledge Graphs. In Proceedings of Workshop on Knowledge Base Construction, Reasoning and Mining, Los Angeles, California USA, Feb 2018 (KBCOM'18). arXiv: 1708.06816 https://arxiv.org/pdf/1708.06816.pdf |
Production Date
| 2018 |
Production Location
| Heidelberg University |
Data Type
| program source code |
Related Material
| FB15K datasets
"This FREEBASE FB15k DATA consists of a collection of triplets (synset, relation_type, triplet) extracted from Freebase (http://www.freebase.com). This data set can be seen as a 3-mode tensor depicting ternary relationships between synsets." (see README, in FB15k)
Link to the FB15K dataset: https://everest.hds.utc.fr/lib/exe/fetch.php?media=en:fb15k.tgz
License: Creative Commons Attribution 2.5 License.
When using this data, one should cite the original paper:
Bordes, A., Usunier, N., García-Durán, A., Weston, J., & Yakhnenko, O. (2013). Translating Embeddings for Modeling Multirelational Data. In Proceedings of Neural Information Processing Systems (NIPS 26), Lake Taho, NV, USA. Dec. 2013. |
Related Dataset
| Kotnis, Bhushan, 2019, "KGE Algorithms", https://doi.org/10.11588/data/CSXYSS, heiDATA |
Data Source
| M. Nickel, K. Murphy, V. Tresp, and E. Gabrilovich. 2016. A Review of Relational Machine Learning for Knowledge Graphs. In Proc. IEEE 104, 1 (Jan 2016), 11–33. https://doi.org/10.1109/JPROC.2015.2483592
Maximilian Nickel, Volker Tresp, and Hans-Peter Kriegel. 2012. Factorizing YAGO: Scalable Machine Learning for Linked Data. In Proceedings of the 21st International Conference on World Wide Web (WWW'12). ACM, New York, NY,USA, 271–280. https://doi.org/10.1145/2187836.2187874
Antoine Bordes, Nicolas Usunier, Alberto Garcia-Duran, Jason Weston, and Oksana Yakhnenko. 2013. Translating Embeddings for Modeling Multirelational Data. In Advances in Neural Information Processing Systems 26, C. J. C. Burges, L. Bottou, M. Welling, Z. Ghahramani, and K. Q. Weinberger (Eds.). Curran Associates, Inc., 2787–2795. https://papers.nips.cc/paper/5071-translating-embeddings-for-modeling-multi-relational-data.pdf
Maximilian Nickel, Lorenzo Rosasco, and Tomaso Poggio. 2016. Holographic Embeddings of Knowledge Graphs. In Proceedings of the Thirtieth AAAI Conference on Artificial Intelligence (AAAI'16). AAAI Press, 1955–1961. https://dl.acm.org/citation.cfm?id=3016100.3016172
PyTorch. https://pytorch.org/ |