通过分布学习提升不确定知识图谱补全的精度
Uncertain Knowledge Graph Completion via Semi-Supervised Confidence Distribution Learning
- 将置信度转为分布,增强嵌入学习的监督信号
- 在两个数据集上优于现有方法,多指标表现更优
- 适合处理置信度分布不均的知识图谱任务
不确定知识图谱(UKG)为每个三元组赋予置信度得分,以提供更精确的知识表示。由于现实世界中的UKG普遍存在不完整性,不确定知识图谱补全成为研究热点,旨在填补缺失的三元组及其置信度。现有方法尝试学习UKG嵌入来解决该问题,但忽略了三元组置信度极度不平衡的分布特性,导致学习到的嵌入难以支持高质量的补全。为此,本文提出一种新的半监督置信度分布学习(ssCDL)方法,将每个三元组的置信度转换为置信度分布,引入不同置信度的更多监督信息以强化嵌入学习过程。ssCDL通过元学习为未标记数据生成伪标签(即预测的三元组置信度),并在有标签数据与伪标签数据间迭代进行关系学习,从而扩充训练数据并重新平衡置信度分布。在两个UKG数据集上的实验表明,ssCDL在不同评估指标下均持续优于当前最优基线方法。
原文摘要 · Abstract (English)
Uncertain knowledge graphs (UKGs) associate each triple with a confidence score to provide more precise knowledge representations. Recently, since real-world UKGs suffer from the incompleteness, uncertain knowledge graph (UKG) completion attracts more attention, aiming to complete missing triples and confidences. Current studies attempt to learn UKG embeddings to solve this problem, but they neglect the extremely imbalanced distributions of triple confidences. This causes that the learnt embeddings are insufficient to high-quality UKG completion. Thus, in this paper, to address the above issue, we propose a new semi-supervised Confidence Distribution Learning (ssCDL) method for UKG completion, where each triple confidence is transformed into a confidence distribution to introduce more supervision information of different confidences to reinforce the embedding learning process. ssCDL iteratively learns UKG embedding by relational learning on labeled data (i.e., existing triples with confidences) and unlabeled data with pseudo labels (i.e., unseen triples with the generated confidences), which are predicted by meta-learning to augment the training data and rebalance the distribution of triple confidences. Experiments on two UKG datasets demonstrate that ssCDL consistently outperforms state-of-the-art baselines in different evaluation metrics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。