Embedding Entities and Relations for Learning and Inference in Knowledge Bases
Open accessAbstract
We consider learning representations of entities and relations in KBs using the neural-embedding approach. We show that most existing models, including NTN (Socher et al., 2013) and TransE (Bordes et al., 2013b), can be generalized under a unified learning framework, where entities are low-dimensional vectors learned from a neural network and relations are bilinear and/or linear mapping functions. Under this framework, we compare a variety of embedding models on the link prediction task. We show that a simple bilinear formulation achieves new state-of-the-art results for the task (achieving a top-10 accuracy of 73.2% vs. 54.7% by TransE on Freebase). Furthermore, we introduce a novel approach that utilizes the learned relation embeddings to mine logical rules such as "BornInCity(a,b) and CityInCountry(b,c) => Nationality(a,c)". We find that embeddings learned from the bilinear objective are particularly good at capturing relational semantics and that the composition of relations is characterized by matrix multiplication. More interestingly, we demonstrate that our embedding-based rule extraction approach successfully outperforms a state-of-the-art confidence-based rule mining approach in mining Horn rules that involve compositional reasoning.
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Integrity signals
Registry
Cites
Statistics
Claims the paper makes
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Claims other papers make about this one
Each is checked against this paper’s own text: does it say what the citing paper says it does?
The second group of multiplicative models includes DistMult and Complex, which can outperform additive models by capturing more semantic information.
“The second group of multiplicative models includes DistMult [20] and Complex [21], which can outperform the additive models by capturing more semantic information [19].”
From A Survey on Knowledge Graph Embeddings for Link Prediction (2021) · cites this paper as [20]
The Results (p. 5) report that DistMult outperformed the additive DistAdd model on the evaluated datasets. The Introduction (p. 1) says bilinear embeddings capture the compositional semantics of relations, but the paper does not evaluate Complex or establish that both named models outperform additive models by capturing more semantic information. (LLM, conf=0.86)
Experimental — accuracy not yet measuredTraditional semantic-based models focus only on the information of the triple itself and do not fuse any additional information.
“The traditional semantic-based models focus only on the information of the triple itself and do not fuse any additional information, as DistMult [20] and Complex [21] do.”
From A Survey on Knowledge Graph Embeddings for Link Prediction (2021) · cites this paper as [20]
DistMult can only model symmetric relations.
“Since DistMult does not distinguish between head and tail entities, it can only model symmetric relations.”
From SimplE Embedding for Link Prediction in Knowledge Graphs (2018) · cites this paper as [46]
Self-check
Percentages, sums and reported differences re-computed from the paper’s own numbers.
The self-check needs the paper’s full text.
Cites
No references could be resolved yet.
Research works printed but not linked: 32
- Auer, S¨ oren, Bizer, Christian, Kobilarov, Georgi, Lehmann, Jens, Cyganiak, Richard, and Ives, Zachary. Dbpedia: A nucleus for a web of open data. In The semantic web , pp. 722-735. Springer, 2007.
- Bordes, Antoine, Weston, Jason, Collobert, Ronan, and Bengio, Yoshua. Learning structured embeddings of knowledge bases. In AAAI , 2011.
- Bordes, Antoine, Glorot, Xavier, Weston, Jason, and Bengio, Yoshua. A semantic matching energy function for learning with multi-relational data. Machine Learning , pp. 1-27, 2013a.
- Bordes, Antoine, Usunier, Nicolas, Garcia-Duran, Alberto, Weston, Jason, and Yakhnenko, Oksana. Translating embeddings for modeling multi-relational data. In NIPS , 2013b.
- Bowman, Samuel R. Can recursive neural tensor networks learn logical reasoning? In ICLR , 2014.
- Chang, Kai-Wei, Yih, Wen-tau, Yang, Bishan, and Meek, Chris. Typed tensor decomposition of knowledge bases for relation extraction. In EMNLP , 2014.
- Deng, Li, Hinton, G., and Kingsbury, B. New types of deep neural network learning for speech recognition and related applications: An overview. In in ICASSP , 2013.
- Duchi, John, Hazan, Elad, and Singer, Yoram. Adaptive subgradient methods for online learning and stochastic optimization. The Journal of Machine Learning Research , 12:2121-2159, 2011.
- Gao, Jianfeng, Pantel, Patrick, Gamon, Michael, He, Xiaodong, Deng, Li, and Shen, Yelong. Modeling interestingness with deep neural networks. In EMNLP , 2014.
- Garc´ ıa-Dur´ an, Alberto, Bordes, Antoine, and Usunier, Nicolas. Effective blending of two and threeway interactions for modeling multi-relational data. In Machine Learning and Knowledge Discovery in Databases , pp. 434-449. Springer, 2014.
- Getoor, Lise and Taskar, Ben (eds.). Introduction to Statistical Relational Learning . The MIT Press, 2007.
- Hinton, Geoff, Deng, L., Yu, D., Dahl, G., Mohamed, A., Jaitly, N., Senior, A., Vanhoucke, V., Nguyen, P., Sainath, T., and Kingsbury, B. Deep neural networks for acoustic modeling in speech recognition. IEEE Sig. Proc. Mag. , 29:82-97, 2012.
- Huang, Po-Sen, He, Xiaodong, Gao, Jianfeng, Deng, Li, Acero, Alex, and Heck, Larry. Learning deep structured semantic models for Web search using clickthrough data. In CIKM , 2013.
- Hutchinson, B, Deng, L., and Yu, D. Tensor deep stacking networks. IEEE Transactions on Pattern Analysis and Machine Intelligence , 35(8):1944-1957, 2013.
- Jenatton, Rodolphe, Le Roux, Nicolas, Bordes, Antoine, and Obozinski, Guillaume. A latent factor model for highly multi-relational data. In NIPS , 2012.
- Kemp, Charles, Tenenbaum, Joshua B, Griffiths, Thomas L, Yamada, Takeshi, and Ueda, Naonori. Learning systems of concepts with an infinite relational model. In AAAI , volume 3, pp. 5, 2006.
- Mikolov, Tomas, Sutskever, Ilya, Chen, Kai, Corrado, Greg S, and Dean, Jeff. Distributed representations of words and phrases and their compositionality. In NIPS , pp. 3111-3119, 2013.
- Nickel, Maximilian, Tresp, Volker, and Kriegel, Hans-Peter. A three-way model for collective learning on multi-relational data. In ICML , pp. 809-816, 2011.
- Nickel, Maximilian, Tresp, Volker, and Kriegel, Hans-Peter. Factorizing YAGO: scalable machine learning for linked data. In WWW , pp. 271-280, 2012.
- Paccanaro, Alberto and Hinton, Geoffrey E. Learning distributed representations of concepts using linear relational embedding. IEEE Transactions on Knowledge and Data Engineering , 13(2): 232-244, 2001.
- Richardson, Matthew and Domingos, Pedro. Markov logic networks. Machine learning , 62(1-2): 107-136, 2006.
- Rockt¨ aschel, Tim, Boˇ snjak, Matko, Singh, Sameer, and Riedel, Sebastian. Low-dimensional embeddings of logic. In ACL Workshop on Semantic Parsing , 2014.
- Schoenmackers, Stefan, Etzioni, Oren, Weld, Daniel S, and Davis, Jesse. Learning first-order horn clauses from web text. In EMNLP , 2010.
- Shen, Yelong, He, Xiaodong, Gao, Jianfeng, Deng, Li, and Mesnil, Gregoire. A latent semantic model with convolutional-pooling structure for information retrieval. In CIKM , 2014a.
- Shen, Yelong, He, Xiaodong, Gao, Jianfeng, Deng, Li, and Mesnil, Gr´ egoire. Learning semantic representations using convolutional neural networks for Web search. In WWW , pp. 373-374, 2014b.
- Singh, Ajit P and Gordon, Geoffrey J. Relational learning via collective matrix factorization. In KDD , pp. 650-658. ACM, 2008.
- Socher, Richard, Huval, Brody, Manning, Christopher D., and Ng, Andrew Y. Semantic compositionality through recursive matrix-vector spaces. In EMNLP-CoNLL , 2012.
- Socher, Richard, Chen, Danqi, Manning, Christopher D., and Ng, Andrew Y. Reasoning with neural tensor networks for knowledge base completion. In NIPS , 2013.
- Sutskever, Ilya, Tenenbaum, Joshua B, and Salakhutdinov, Ruslan. Modelling relational data using Bayesian clustered tensor factorization. In NIPS , pp. 1821-1828, 2009.
- Vinyals, O., Jia, Y ., Deng, L., and Darrell, T. Learning with recursive perceptual representations. In NIPS , 2012.
- Yih, Wen-tau, He, Xiaodong, and Meek, Christopher. Semantic parsing for single-relation question answering. In ACL , 2014.
- Yu, D., Deng, L., and Seide, F. The deep tensor neural network with applications to large vocabulary speech recognition. IEEE Trans. Audio, Speech and Language Proc. , 21(2):388 -396, 2013.
Cited by
Sorted by citations
Modeling Relational Data with Graph Convolutional Networks
Schlichtkrull, M., Kipf, T. N., Bloem, P., van den Berg, R., Titov, I., & Welling, M. (2018) · Lecture Notes in Computer Science · doi:10.1007/978-3-319-93417-4_38 · Cited by 5,570 · see citing works
A Survey on Knowledge Graphs: Representation, Acquisition, and Applications
Ji, S., Pan, S., Cambria, E., Marttinen, P., & Yu, P. S. (2022) · IEEE Transactions on Neural Networks and Learning Systems · doi:10.1109/tnnls.2021.3070843 · Cited by 2,947 · see citing works
Knowledge Graph Embedding: A Survey of Approaches and Applications
Wang, Q., Mao, Z., Wang, B., & Guo, L. (2017) · IEEE Transactions on Knowledge and Data Engineering · doi:10.1109/tkde.2017.2754499 · Cited by 2,654 · see citing works
Complex Embeddings for Simple Link Prediction
Trouillon, T., Welbl, J., Riedel, S., Gaussier, É., & Bouchard, G. (2016) · arXiv · doi:10.48550/arxiv.1606.06357 · Cited by 2,100 · see citing works
RotatE: Knowledge Graph Embedding by Relational Rotation in Complex Space
Sun, Z., Deng, Z.-H., Nie, J.-Y., & Tang, J. (2019) · arXiv · doi:10.48550/arxiv.1902.10197 · Cited by 1,414 · see citing works
Unifying Large Language Models and Knowledge Graphs: A Roadmap
Pan, S., Luo, L., Wang, Y., Chen, C., Wang, J., & Wu, X. (2024) · IEEE Transactions on Knowledge and Data Engineering · doi:10.1109/tkde.2024.3352100 · Cited by 1,376 · see citing works
A review: Knowledge reasoning over knowledge graph
Chen, X., Jia, S., & Xiang, Y. (2020) · Expert Systems with Applications · doi:10.1016/j.eswa.2019.112948 · Cited by 1,035 · see citing works
Representation Learning of Knowledge Graphs with Entity Descriptions
Xie, R., Liu, Z., Jia, J., Luan, H., & Sun, M. (2016) · Proceedings of the AAAI Conference on Artificial Intelligence · doi:10.1609/aaai.v30i1.10329 · Cited by 725 · see citing works
Knowledge Enhanced Contextual Word Representations
Peters, M. E., Neumann, M., Logan, R., Schwartz, R., Joshi, V., Singh, S., & Smith, N. A. (2019) · Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Language Processing (EMNLP-IJCNLP) · doi:10.18653/v1/d19-1005 · Cited by 661 · see citing works
SimplE Embedding for Link Prediction in Knowledge Graphs
Kazemi, S. M., & Poole, D. (2018) · arXiv · doi:10.48550/arxiv.1802.04868 · Cited by 476 · see citing works
“DistMult (Yang et al., 2015), one of the simplest multiplicative approaches, considers the embeddings for each entity and each relation to be a single vector of size k (v(e) and v(r)) and defines its similarity function as: f(E(h),E(r),E(t)) = 〈v(h), v(r), v(t)〉.”
BackgroundLow-Dimensional Hyperbolic Knowledge Graph Embeddings
Chami, I., Wolf, A., Juan, D.-C., Sala, F., Ravi, S., & Ré, C. (2020) · Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics · doi:10.18653/v1/2020.acl-main.617 · Cited by 384 · see citing works
Utilizing graph machine learning within drug discovery and development
Gaudelet, T., Day, B., Jamasb, A. R., Soman, J., Regep, C., Liu, G., Hayter, J. B. R., Vickers, R., Roberts, C., Tang, J., et al. (2021) · Briefings in Bioinformatics · doi:10.1093/bib/bbab159 · Cited by 313 · see citing works
Entity Embeddings of Categorical Variables
Guo, C., & Berkhahn, F. (2016) · arXiv · doi:10.48550/arxiv.1604.06737 · Cited by 308 · see citing works
Ye, Q., Hsieh, C.-Y., Yang, Z., Kang, Y., Chen, J., Cao, D., He, S., & Hou, T. (2021) · Nature Communications · doi:10.1038/s41467-021-27137-3 · Cited by 261 · see citing works
Toward better drug discovery with knowledge graph
Zeng, X., Tu, X., Liu, Y., Fu, X., & Su, Y. (2022) · Current Opinion in Structural Biology · doi:10.1016/j.sbi.2021.09.003 · Cited by 229 · see citing works
A Survey of Data-driven and Knowledge-aware eXplainable AI
Li, X.-H., Cao, C. C., Shi, Y., Bai, W., Gao, H., Qiu, L., Wang, C., Gao, Y., Zhang, S., Xue, X., et al. (2020) · IEEE Transactions on Knowledge and Data Engineering · doi:10.1109/tkde.2020.2983930 · Cited by 224 · see citing works
A Survey on Knowledge Graph Embeddings for Link Prediction
Wang, M., Qiu, L., & Wang, X. (2021) · Symmetry · doi:10.3390/sym13030485 · Cited by 212 · see citing works
CoLAKE: Contextualized Language and Knowledge Embedding
Sun, T., Shao, Y., Qiu, X., Guo, Q., Hu, Y., Huang, X., & Zhang, Z. (2020) · Proceedings of the 28th International Conference on Computational Linguistics · doi:10.18653/v1/2020.coling-main.327 · Cited by 185 · see citing works
TeMP: Temporal Message Passing for Temporal Knowledge Graph Completion
Wu, J., Cao, M., Cheung, J. C. K., & Hamilton, W. L. (2020) · Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing (EMNLP) · doi:10.18653/v1/2020.emnlp-main.462 · Cited by 167 · see citing works
Enhancing Pre-Trained Language Representations with Rich Knowledge for Machine Reading Comprehension
Yang, A., Wang, Q., Liu, J., Liu, K., Lyu, Y., Wu, H., She, Q., & Li, S. (2019) · Proceedings of the 57th Annual Meeting of the Association for Computational Linguistics · doi:10.18653/v1/p19-1226 · Cited by 155 · see citing works
Neural-symbolic integration and the Semantic Web
Hitzler, P., Bianchi, F., Ebrahimi, M., & Sarker, M. K. (2020) · Semantic Web · doi:10.3233/sw-190368 · Cited by 70 · see citing works
Pretrain-KGE: Learning Knowledge Representation from Pretrained Language Models
Zhang, Z., Liu, X., Zhang, Y., Su, Q., Sun, X., & He, B. (2020) · Findings of the Association for Computational Linguistics: EMNLP 2020 · doi:10.18653/v1/2020.findings-emnlp.25 · Cited by 52 · see citing works
Probabilistic Box Embeddings for Uncertain Knowledge Graph Reasoning
Chen, X., Boratko, M., Chen, M., Dasgupta, S. S., Li, X. L., & McCallum, A. (2021) · Proceedings of the 2021 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies · doi:10.18653/v1/2021.naacl-main.68 · Cited by 47 · see citing works
Natural Language Processing for Information Extraction
Singh, S. (2018) · arXiv · doi:10.48550/arxiv.1807.02383 · Cited by 45 · see citing works
Zhao, C., Jiang, J., Guan, Y., Guo, X., & He, B. (2018) · Artificial Intelligence in Medicine · doi:10.1016/j.artmed.2018.03.005 · Cited by 41 · see citing works
RHO: Reducing Hallucination in Open-domain Dialogues with Knowledge Grounding
Ji, Z., Liu, Z., Lee, N., Yu, T., Wilie, B., Zeng, M., & Fung, P. (2023) · Findings of the Association for Computational Linguistics: ACL 2023 · doi:10.18653/v1/2023.findings-acl.275 · Cited by 37 · see citing works
A General Framework for Information Extraction using Dynamic Span Graphs
Luan, Y., Wadden, D., He, L., Shah, A., Ostendorf, M., & Hajishirzi, H. (2019) · arXiv · doi:10.48550/arxiv.1904.03296 · Cited by 34 · see citing works
“Knowledge graph completion (Yang et al., 2015; Bordes et al., 2013) is one prominent example.”
BackgroundModeling Heterogeneous Hierarchies with Relation-specific Hyperbolic Cones
Bai, Y., Ying, R., Ren, H., & Leskovec, J. (2021) · arXiv · doi:10.48550/arxiv.2110.14923 · Cited by 31 · see citing works
“Prominent examples include TransE [5], DistMult [24], ComplEx [25], RotatE [7] and TuckER [14].”
BackgroundDevarakonda, M. V., Mohanty, S., Sunkishala, R. R., Mallampalli, N., & Liu, X. (2024) · Journal of Biomedical Informatics · doi:10.1016/j.jbi.2024.104627 · Cited by 9 · see citing works
Semantic embeddings of generic objects for zero-shot learning
Hascoet, T., Ariki, Y., & Takiguchi, T. (2019) · EURASIP Journal on Image and Video Processing · doi:10.1186/s13640-018-0371-x · Cited by 9 · see citing works
Explainable knowledge graph embeddings for industrial process monitoring & control
Weyns, M., Blyau, T., Steenwinckel, B., De Turck, F., Van Hoecke, S., & Ongenae, F. (2025) · Information Fusion · doi:10.1016/j.inffus.2025.103242 · Cited by 7 · see citing works
Med-KGMA: A novel AI-driven medical support system leveraging knowledge graphs and medical advisors
Varshney, S., Jain, B., Singh, P., Rani, D., & Mehra, S. (2025) · Computers in Biology and Medicine · doi:10.1016/j.compbiomed.2025.110929
- Authors
- Yang, B., Yih, W.-T., He, X., Gao, J., & Deng, L.
- Affiliations
- Cornell University (US)
- Microsoft (United States) (US)
- Venue
- arXiv
- Publisher
- arXiv
- Published
- 2014
- ISSN
- 2331-8422 (ISSN-L)
- arXiv
- 1412.6575
- OpenAlex
- W1533230146
- Cited by
- 2,252
- References
- 0
- Citations by year
- 2026: 17 · 2025: 169 · 2024: 352 · 2023: 331 · 2022: 322 · 2021: 396 · 2020: 290 · 2019: 192 · 2018: 94 · 2017: 46 · 2016: 33 · 2015: 9 · 2012: 1
- Type
- posted content
- Language
- en
- Open access
- green
- arxiv.org
- Locations
- arXiv (Cornell University) (submittedVersion)
- arXiv (Cornell University)
- Topics
- Physical Sciences › Computer Science › Artificial Intelligence › Advanced Graph Neural Networks
- Physical Sciences › Computer Science › Artificial Intelligence › Logic, Reasoning, and Knowledge
- Physical Sciences › Computer Science › Artificial Intelligence › Semantic Web and Ontologies
- Keywords
- knowledge graphs, first-order logic rules, inductive logic programming, knowledge graph completion, fact-checking, rule-based reasoning, knowledge base inference
About these checks
Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.
Statistics and arithmetic are recomputed, not judged.
Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.
What counts as a claim ›How we measure confidence ›How we check this ›
Are you an author, or do you see a problem with this page? Request a review