Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Duality-Induced Regularizer for Tensor Factorization Based Knowledge Graph Completion

About

Tensor factorization based models have shown great power in knowledge graph completion (KGC). However, their performance usually suffers from the overfitting problem seriously. This motivates various regularizers -- such as the squared Frobenius norm and tensor nuclear norm regularizers -- while the limited applicability significantly limits their practical usage. To address this challenge, we propose a novel regularizer -- namely, DUality-induced RegulArizer (DURA) -- which is not only effective in improving the performance of existing models but widely applicable to various methods. The major novelty of DURA is based on the observation that, for an existing tensor factorization based KGC model (primal), there is often another distance based KGC model (dual) closely associated with it. Experiments show that DURA yields consistent and significant improvements on benchmarks.

Zhanqiu Zhang, Jianyu Cai, Jie Wang• 2020

Related benchmarks

TaskDatasetResultRank
Link PredictionFB15k-237
MRR36.8
293
Link PredictionWN18RR
Hits@1057.7
188
Knowledge Graph CompletionFB15k-237 (test)
MRR0.371
179
Knowledge Graph CompletionWN18RR (test)
MRR0.498
177
Knowledge Base CompletionYAGO3-10 (test)
MRR0.584
71
Showing 5 of 5 rows

Other info

Code

Follow for update