Fast knowledge graph completion using graphics processing units

被引:0
作者
Lee, Chun-Hee [1 ,2 ]
Kang, Dong-oh [1 ]
Song, Hwa Jeon [1 ]
机构
[1] ETRI, Intelligence Informat Res Div, Daejeon 34129, South Korea
[2] Kyungpook Natl Univ, Sch Comp Sci & Engn, Daegu 41566, South Korea
关键词
Knowledge graph embedding; Knowledge graph completion; TransE; Similarity join; GPU processing; FAST SIMILARITY JOIN; ALGORITHM;
D O I
10.1016/j.jpdc.2024.104885
中图分类号
TP301 [理论、方法];
学科分类号
081202 ;
摘要
Knowledge graphs can be used in many areas related to data semantics such as question -answering systems, knowledge based systems. However, the currently constructed knowledge graphs need to be complemented for better knowledge in terms of relations. It is called knowledge graph completion. To add new relations to the existing knowledge graph by using knowledge graph embedding models, we have to evaluate N x N x R vector operations, where N is the number of entities and R is the number of relation types. It is very costly. In this paper, we provide an efficient knowledge graph completion framework on GPUs to get new relations using knowledge graph embedding vectors. In the proposed framework, we first define transformable to a metric space and then provide a method to transform the knowledge graph completion problem into the similarity join problem for a model which is transformable to a metric space . After that, to efficiently process the similarity join problem, we derive formulas using the properties of a metric space. Based on the formulas, we develop a fast knowledge graph completion algorithm. Finally, we experimentally show that our framework can efficiently process the knowledge graph completion problem.
引用
收藏
页数:15
相关论文
共 43 条
[1]   Bringing Light Into the Dark: A Large-Scale Evaluation of Knowledge Graph Embedding Models Under a Unified Framework [J].
Ali, Mehdi ;
Berrendorf, Max ;
Hoyt, Charles Tapley ;
Vermue, Laurent ;
Galkin, Mikhail ;
Sharifzadeh, Sahand ;
Fischer, Asja ;
Tresp, Volker ;
Lehmann, Jens .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2022, 44 (12) :8825-8845
[2]  
[Anonymous], 2008, US
[3]  
Böhm C, 2001, SIGMOD RECORD, V30, P379, DOI 10.1145/376284.375714
[4]  
Bohm C., 2009, Index-supported similarity join on graphics processors, P57
[5]  
Bollacker Kurt., P 2008 ACM SIGMOD IN
[6]  
Bordes A., 2011, AAAI, P301, DOI [DOI 10.1609/AAAI.V25I1.7917, 10.1609/aaai.v25i1.7917]
[7]  
Bordes A., 2013, P 26 INT C NEURAL IN, P2787
[8]  
Brinkhoff T., 1994, SIGMOD Record, V23, P197, DOI 10.1145/191843.191880
[9]  
Broscheit S, 2020, PROCEEDINGS OF THE 2020 CONFERENCE ON EMPIRICAL METHODS IN NATURAL LANGUAGE PROCESSING: SYSTEM DEMONSTRATIONS, P165
[10]  
Cai L., 2018, P 2018 C N AM CHAPT, V1, P1470, DOI [DOI 10.18653/V1/N18-1133, 10.18653/v1/N18-1133]