首页|基于多尺度卷积的胶囊网络知识图谱嵌入模型

基于多尺度卷积的胶囊网络知识图谱嵌入模型

扫码查看
该文基于胶囊神经网络出色的维度信息挖掘能力,加入多尺度卷积以进一步增强其特征提取和交互能力,提出了基于多尺度卷积的胶囊网络知识图谱嵌入模型.首先,通过TransE算法训练得到实体和关系的初始化嵌入向量;其次,通过多尺度卷积生成不同的特征图,将得到的特征图进行特征融合,融合后得到的特征图重组为相对应的胶囊;最后,利用动态路由指定从第一层胶囊到第二层胶囊的连接,经过路由得到的第二层胶囊利用squash函数得到最终向量长度,该向量长度决定三元组的置信度.知识图谱链接预测任务的实验结果表明,较嵌入模型CapsE,本文提出的模型在WN18RR数据集上指标Hit@10提高1.8%,MRR提高 1.4%,在 FB15k-237 数据集上 Hit@10 提高 2.2%,MR 提高 4.8%.
Knowledge graph embedding model based on multi-scale convolution of capsule network
This paper is based on the excellent dimensional information mining ability of capsule neural networks,and incorporates multi-scale convolution to further enhance their feature extraction and interac-tion capabilities.A capsule network knowledge graph embedding model based on multi-scale convolution is proposed.Firstly,the initialization embedding vectors for entities and relationships are trained using the TransE algorithm.Secondly,different feature maps are generated through multi-scale convolution,and the resulting feature maps are fused to form corresponding capsules.Finally,dynamic routing is used to speci-fy the connection from the first layer capsule to the second layer capsule.The second layer capsule obtained through routing is then used to obtain the final vector length using the squash function,which determines the confidence level of the triplet.Compared with the embedding models CapsE,the proposed model in this paper has a 1.8%improvement in Hit@10 and 1.4%improvement in MRR metrics on the WN18RR dataset,and a 2.2%improvement in Hit@10 and 4.8%improvement in MR on the FB15k-237 dataset.

knowledge graphmulti-scale convolutioncapsule networkknowledge graph embed-dingneural network

周淑霄、王艳娜、周子力、王妍、董兆安

展开 >

曲阜师范大学网络空间安全学院,273165,曲阜市

曲阜师范大学计算机学院,276826,山东省日照市

知识图谱 多尺度卷积 胶囊网络 知识图谱嵌入 神经网络

山东省自然科学基金山东省高等学校科技计划教育部产学合作协同育人项目

ZR2020MF149J18KB161202102291003

2024

曲阜师范大学学报(自然科学版)
山东曲阜师范大学

曲阜师范大学学报(自然科学版)

影响因子:0.299
ISSN:1001-5337
年,卷(期):2024.50(2)
  • 21