Our new X account is live! Follow @wizwand_team for updates
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
280
Knowledge Graph CompletionFB15k-237 (test)
MRR0.371
179
Knowledge Graph CompletionWN18RR (test)
MRR0.498
177
Link PredictionWN18RR
Hits@1057.7
175
Knowledge Base CompletionYAGO3-10 (test)
MRR0.584
71
Showing 5 of 5 rows

Other info

Code

Follow for update