notesum.ai

Published at November 11

UniHR: Hierarchical Representation Learning for Unified Knowledge Graph Link Prediction

cs.CL

cs.AI

Released Date: November 11, 2024

Authors: Zhiqiang Liu¹, Mingyang Chen², Yin Hua¹, Zhuo Chen², Ziqi Liu³, Lei Liang³, Huajun Chen², Wen Zhang¹

Aff.: ¹School of Software Technology, Zhejiang University; ²College of Computer Science and Technology, Zhejiang University; ³Ant Group

Arxiv: http://arxiv.org/abs/2411.07019v1

	WikiPeople							WD50K
Model	# Tr. Params	subject/object			all entities			# Tr. Params	subject/object			all entities
	# Tr. Params	MRR	Hits@1	Hits@10	MRR	Hits@1	Hits@10	# Tr. Params	MRR	Hits@1	Hits@10	MRR	Hits@1	Hits@10
NaLP	-	0.356	0.271	0.499	0.360	0.275	0.503	-	0.230	0.170	0.347	0.251	0.187	0.375
tNaLP	-	0.358	0.288	0.486	0.361	0.290	0.490	-	0.221	0.163	0.331	0.243	0.182	0.360
RAM	27.34M	0.459	0.384	0.584	0.461	0.386	0.585	-	0.276	0.210	0.399	0.296	0.232	0.416
HINGE	-	0.393	0.309	0.547	0.395	0.311	0.549	-	0.264	0.187	0.410	0.277	0.200	0.424
NeuInfer	-	0.357	0.247	0.533	0.357	0.248	0.532	-	0.220	0.154	0.347	0.225	0.158	0.355
StarE	7.84M	0.458	0.364	0.611	-	-	-	10.39M	0.309	0.234	0.452	-	-	-
HyTransformer	-	0.460	0.382	0.594	-	-	-	-	0.304	0.231	0.443	-	-	-
GRAN	15.26M	0.462	0.366	0.610	0.465	0.371	0.613	18.51M	0.330	0.255	0.472	0.361	0.286	0.501
HyNT	23.02M	0.482	0.415	0.602	0.481	0.414	0.603	29.61M	0.333	0.259	0.474	0.360	0.287	0.500
UniHR	8.02M	0.491	0.417	0.618	0.493	0.420	0.621	10.55M	0.348	0.278	0.482	0.382	0.313	0.513