有效

一种基于权重因子和非采样损失函数的知识图谱训练方法

张鹏飞、袁雪美、马武彬、赵翔、胡艳丽、谭真、肖卫东
中国人民解放军国防科技大学

摘要

本发明属于数字化装备技术领域,公开了基于权重因子和非采样损失函数的知识图谱训练方法,包括以下步骤:构造知识图谱嵌入模型;将基于矩阵/张量分解的知识图谱中所有实体和关系组成的非正三元组作为负三元组全部用于训练;在基于矩阵/张量分解的知识图谱嵌入模型中,通过基于概率预测函数的非采样损失函数,缓解基于平方的损失函数带来的模型的性能削弱问题;采用权重因子对负三元组的损失函数进行动态加权,减少易分负三元组对模型训练造成的不良影响;划分训练集和测试集,对知识图谱嵌入模型进行训练,优化模型参数,发现装备潜在问题。本申请降低易分负三元组对模型训练造成的不利影响,从而提高模型训练的精度。