公开(公告)号：CN117892807A

公开(公告)日：2024-04-16

申请号：CN202410041466.X

申请日：2024-01-11

申请人： 中国人民解放军国防科技大学

发明人： 周斌 ,  宋鑫 ,  王晔 ,  汪海洋 ,  黄颜熠 ,  田磊 ,  谢锋 ,  高立群 ,  丁锐东 ,  李爱平 ,  江荣 ,  涂宏魁

IPC分类号： G06N5/02 ,  G06F40/289 ,  G06N3/0455 ,  G06N3/0499

摘要： 本发明提供了基于预训练语言模型提示微调的开放知识图谱补全方法，其充分利用预训练语言模型中的开放知识来协助知识图谱补全，方法包括提取所有关系并构造关系特定模板，根据关系特定模板将三元组数据转换成自然语言句子，采用分词器处理成token序列并采用token[MASK]来替换尾实体；获得缺失尾实体的三元组的头实体h的邻居三元组集合，将邻居三元组中的尾实体替换成token[MASK]，输入编码器获得头实体h的邻居嵌入向量；设置软token[con]作为软连接符，以形成整体的软提示，基于Bert模型和MLP网络构建尾实体预测模型，将构建的整体的软提示输入最优的尾实体预测模型，输出尾实体以补全知识图谱。