一种基于文本信息增强实体嵌入的方法

作者:陆以勤; 洪炜妍; 覃健诚; 谢树禄; 李智鹏; 陈帅豪; 陈嘉睿
来源:2021-03-29, 中国, ZL202110334949.5.

摘要

本发明公开了一种基于文本信息增强实体嵌入的方法,包括预训练知识图谱嵌入模型,得到实体向量和关系向量;将实体向量和关系向量分别加载到实体嵌入矩阵和关系嵌入矩阵中,得到实体结构嵌入以及关系结构嵌入;预训练词向量模型,得到词向量;查询词嵌入矩阵,得到实体描述词向量和关系词向量;将实体描述词向量输入到BiLSTM网络中,在BiLSTM网络的输出层中引入点积注意力;对关系词向量取平均得到关系嵌入;将实体结构嵌入投影到关系空间,得到实体结构嵌入投影,再将实体描述嵌入和实体结构嵌入投影相加,得到实体嵌入。本发明利用网络安全知识库中附带的文本信息来增强实体的表示能力,进而提高实体链接预测的准确率。