MQuinE: a cure for "Z-paradox" in knowledge graph embedding models (2402.03583v3)
Abstract: Knowledge graph embedding (KGE) models achieved state-of-the-art results on many knowledge graph tasks including link prediction and information retrieval. Despite the superior performance of KGE models in practice, we discover a deficiency in the expressiveness of some popular existing KGE models called \emph{Z-paradox}. Motivated by the existence of Z-paradox, we propose a new KGE model called \emph{MQuinE} that does not suffer from Z-paradox while preserves strong expressiveness to model various relation patterns including symmetric/asymmetric, inverse, 1-N/N-1/N-N, and composition relations with theoretical justification. Experiments on real-world knowledge bases indicate that Z-paradox indeed degrades the performance of existing KGE models, and can cause more than 20\% accuracy drop on some challenging test samples. Our experiments further demonstrate that MQuinE can mitigate the negative impact of Z-paradox and outperform existing KGE models by a visible margin on link prediction tasks.
- Boxe: A box embedding model for knowledge base completion. Advances in Neural Information Processing Systems, 33:9649–9661, 2020.
- Tucker: Tensor factorization for knowledge graph completion. In Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Language Processing (EMNLP-IJCNLP), pp. 5185–5194, 2019.
- Hopfe: Knowledge graph representation learning using inverse hopf fibrations. In Proceedings of the 30th ACM International Conference on Information & Knowledge Management, pp. 89–99, 2021.
- A neural probabilistic language model. J. Mach. Learn. Res., 3:1137–1155, 2003.
- Latent dirichlet allocation. J. Mach. Learn. Res., 3:993–1022, 2003.
- Freebase: a collaboratively created graph database for structuring human knowledge. In Proceedings of the 2008 ACM SIGMOD international conference on Management of data, pp. 1247–1250, 2008.
- Translating embeddings for modeling multi-relational data. Advances in neural information processing systems, 26, 2013.
- Dual quaternion knowledge graph embeddings. In Proceedings of the AAAI conference on artificial intelligence, volume 35, pp. 6894–6902, 2021.
- Fuzzy logic based logical query answering on knowledge graphs. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 36, pp. 3939–3948, 2022.
- Convolutional 2d knowledge graph embeddings. In Proceedings of the AAAI conference on artificial intelligence, volume 32, 2018.
- Christiane Fellbaum. WordNet: An Electronic Lexical Database. Bradford Books, 1998.
- Knowledge graph embedding based question answering. In Proceedings of the twelfth ACM international conference on web search and data mining, pp. 105–113, 2019.
- Integration of knowledge graph embedding into topic modeling with hierarchical dirichlet process. In Proceedings of the 2019 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, Volume 1 (Long and Short Papers), pp. 940–950, 2019.
- House: Knowledge graph embedding with householder parameterization. In International Conference on Machine Learning, pp. 13209–13224. PMLR, 2022.
- Dense: An enhanced non-abelian group representation for knowledge graph embedding. arXiv preprint arXiv:2008.04548, 2020.
- Neural network-based question answering over knowledge graphs on word and character level. In Proceedings of the 26th international conference on World Wide Web, pp. 1211–1220, 2017.
- Jointly learning explainable rules for recommendation with knowledge graph. In The world wide web conference, pp. 1210–1221, 2019.
- Yago3: A knowledge base from multilingual wikipedias. In 7th biennial conference on innovative data systems research. CIDR Conference, 2014.
- Efficient estimation of word representations in vector space. In 1st International Conference on Learning Representations, ICLR 2013, Scottsdale, Arizona, USA, May 2-4, 2013, Workshop Track Proceedings.
- Distributed representations of words and phrases and their compositionality. Advances in neural information processing systems, 26, 2013.
- Learning attention-based embeddings for relation prediction in knowledge graphs. In Proceedings of the 57th Annual Meeting of the Association for Computational Linguistics, pp. 4710–4723, 2019.
- A three-way model for collective learning on multi-relational data. In ICML, 2011.
- Expressive: A spatio-functional embedding for knowledge graph completion. In International Conference on Learning Representations, 2023.
- CoDEx: A Comprehensive Knowledge Graph Completion Benchmark. In Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing (EMNLP), pp. 8328–8350, Online, November 2020. Association for Computational Linguistics. doi: 10.18653/v1/2020.emnlp-main.669. URL https://www.aclweb.org/anthology/2020.emnlp-main.669.
- Logician and orator: Learning from the duality between language and knowledge in open domain. In Proceedings of the 2018 Conference on Empirical Methods in Natural Language Processing, pp. 2119–2130, 2018.
- Rotate: Knowledge graph embedding by relational rotation in complex space. In International Conference on Learning Representations, 2019.
- Orthogonal relation transforms with graph context modeling for knowledge graph embedding. In Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics, pp. 2713–2722, 2020.
- Observed versus latent features for knowledge base and text inference. In Proceedings of the 3rd workshop on continuous vector space models and their compositionality, pp. 57–66, 2015.
- Complex embeddings for simple link prediction. In International conference on machine learning, pp. 2071–2080. PMLR, 2016.
- Composition-based multi-relational graph convolutional networks. In International Conference on Learning Representations, 2020.
- Dkn: Deep knowledge-aware network for news recommendation. In Proceedings of the 2018 world wide web conference, pp. 1835–1844, 2018.
- Mixed-curvature multi-relational graph neural network for knowledge graph completion. In Proceedings of the Web Conference 2021, pp. 1761–1771, 2021.
- Kgat: Knowledge graph attention network for recommendation. In Proceedings of the 25th ACM SIGKDD international conference on knowledge discovery & data mining, pp. 950–958, 2019.
- Relation embedding with dihedral group in knowledge graph. In Proceedings of the 57th Annual Meeting of the Association for Computational Linguistics, pp. 263–272, 2019.
- Seek: Segmented embedding of knowledge graphs. In Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics, pp. 3888–3897, 2020.
- Embedding entities and relations for learning and inference in knowledge bases. arXiv preprint arXiv:1412.6575, 2014.
- Mquade: a unified model for knowledge fact embedding. In Proceedings of the Web Conference 2021, pp. 3442–3452, 2021.
- Collaborative knowledge base embedding for recommender systems. In Proceedings of the 22nd ACM SIGKDD international conference on knowledge discovery and data mining, pp. 353–362, 2016.
- Quaternion knowledge graph embeddings. Advances in neural information processing systems, 32, 2019.
- Learning hierarchy-aware knowledge graph embeddings for link prediction. In Thirty-Fourth AAAI Conference on Artificial Intelligence, pp. 3065–3072. AAAI Press, 2020.
- Jointe: Jointly utilizing 1d and 2d convolution for knowledge graph embedding. Knowledge-Based Systems, 240:108100, 2022.
- Neural bellman-ford networks: A general graph neural network framework for link prediction. Advances in Neural Information Processing Systems, 34, 2021.