English

Positive-Unlabeled Learning with Adversarial Data Augmentation for Knowledge Graph Completion

Machine Learning 2022-08-17 v3 Artificial Intelligence

Abstract

Most real-world knowledge graphs (KG) are far from complete and comprehensive. This problem has motivated efforts in predicting the most plausible missing facts to complete a given KG, i.e., knowledge graph completion (KGC). However, existing KGC methods suffer from two main issues, 1) the false negative issue, i.e., the sampled negative training instances may include potential true facts; and 2) the data sparsity issue, i.e., true facts account for only a tiny part of all possible facts. To this end, we propose positive-unlabeled learning with adversarial data augmentation (PUDA) for KGC. In particular, PUDA tailors positive-unlabeled risk estimator for the KGC task to deal with the false negative issue. Furthermore, to address the data sparsity issue, PUDA achieves a data augmentation strategy by unifying adversarial training and positive-unlabeled learning under the positive-unlabeled minimax game. Extensive experimental results on real-world benchmark datasets demonstrate the effectiveness and compatibility of our proposed method.

Keywords

Cite

@article{arxiv.2205.00904,
  title  = {Positive-Unlabeled Learning with Adversarial Data Augmentation for Knowledge Graph Completion},
  author = {Zhenwei Tang and Shichao Pei and Zhao Zhang and Yongchun Zhu and Fuzhen Zhuang and Robert Hoehndorf and Xiangliang Zhang},
  journal= {arXiv preprint arXiv:2205.00904},
  year   = {2022}
}

Comments

IJCAI 2022

R2 v1 2026-06-24T11:04:46.300Z