A Generative Adversarial Negative Sampling Method for Knowledge Hypergraph Link Prediction

被引:0
|
作者
Guo Z. [1 ]
Zuo J. [1 ]
Duan L. [1 ]
Li R. [1 ]
He C. [1 ]
Xiao Y. [1 ]
Wang P. [1 ]
机构
[1] School of Computer Science, Sichuan University, Chengdu
基金
中国国家自然科学基金;
关键词
Generative adversarial network; Knowledge hypergraph; Link prediction; N-ary relation; Negative sampling;
D O I
10.7544/issn1000-1239.20220074
中图分类号
学科分类号
摘要
As an extension of the knowledge graph, the knowledge hypergraph has a strong ability to express n-ary relational facts. Using the knowledge hypergraph to model known facts in the real world and discover unknown facts through link prediction has become a current research hotspot. Among existing knowledge hypergraph (or knowledge graph) link prediction methods, constructing the loss function using true labels of samples and their predicted labels is a key step, where negative samples have a great influence on the training of the link prediction model. However, when applying the negative sampling methods for knowledge graph link prediction (e.g., the uniformly random sampling) to the knowledge hypergraph, we may face problems such as low quality of negative samples and high complexity of models. As a result, we design a generative adversarial negative sampling method, named HyperGAN, for knowledge hypergraph link prediction, which generates high-quality negative samples through adversarial training to solve the zero loss problem, thereby improving the accuracy of the link prediction model. Besides, HyperGAN does not require pre-training, which makes it more efficient than previous negative sampling methods in assisting the training of link prediction models. Comparative experiments on multiple real-world datasets show that HyperGAN outperforms the baselines in terms of performance and efficiency. In addition, the case study and quantitative analysis further validate our method in improving the quality of negative samples. © 2022, Science Press. All right reserved.
引用
收藏
页码:1742 / 1756
页数:14
相关论文
共 37 条
  • [1] Bollacker K, Evans C, Paritosh P, Et al., Freebase: A collaboratively created graph database for structuring human knowledge, Proc of the 2008 ACM SIGMOD Int Conf on Management of Data, pp. 1247-1250, (2008)
  • [2] Fatemi B, Taslakian P, Vazquez D, Et al., Knowledge hypergraphs: Prediction beyond binary relations, Proc of the 29th Int Joint Conf on Artificial Intelligence, pp. 2191-2197, (2020)
  • [3] Lande D, Fu Minglei, Guo Wen, Et al., Link prediction of scientific collaboration networks based on information retrieval, World Wide Web-Internet and Web Information Systems, 23, 4, pp. 2239-2257, (2020)
  • [4] Zhang Fuzheng, Yuan N, Lian Defu, Et al., Collaborative knowledge base embedding for recommender systems, Proc of the 22nd ACM SIGKDD Int Conf on Knowledge Discovery and Data Mining, pp. 353-362, (2016)
  • [5] JiShuyi, Feng Yifan, Ji Rongong, Et al., Dual channel hypergraph collaborative filtering, Proc of the 26th ACM SIGKDD Int Conf on Knowledge Discovery and Data Mining, pp. 2020-2029, (2020)
  • [6] Yang Bishan, Mitchell T., Leveraging knowledge bases in LSTMs for improving machine reading, Proc of the 55th Annual Meeting of the Association for Computational Linguistics, pp. 1436-1446, (2017)
  • [7] Abboud R, Ceylan I, Lukasiewicz T, Et al., BoxE: A box embedding model for knowledge base completion, Proc of the 34th Conf on Neural Information Processing Systems, pp. 9649-9661, (2020)
  • [8] Auer S, Bizer C, Kobilarov G, Et al., DBpedia: A nucleus for a Web of open data, Proc of the 6th Int Semantic Web Conf and 2nd Asian Semantic Web Conf, pp. 722-735, (2007)
  • [9] Krompabeta D, Baier S, Tresp V., Type-constrained representation learning in knowledge graphs, Proc of the 14th Int Semantic Web Conf, pp. 640-655, (2015)
  • [10] Cai Liwei, Wang Y W., KBGAN: Adversarial learning for knowledge graph embeddings, Proc of the 2018 Conf of the North American Chapter of the Association for Computational Linguistics, pp. 1470-1480, (2018)