1 图嵌入概述
1.1 同质网络嵌入
1.2 异构网络嵌入
1.3 多层异构网络嵌入
2 模型定义
3 模型框架
表1 模型算法 |
输入:多层异构兵棋联合作战态势实体知识图谱G=(V, E, O, R, A),实体嵌入向量维度d,边嵌入向量维度s,学习率η,负采样数量L,系数α和β 输出:所有实体节点对于不同边类型r汇聚后的节点嵌入向量vi,r |
---|
Step1:初始化模型参数θ Step2:为每种边类型生成随机游走序列 Step3:从每种边类型的随机游走序列中生成训练样本集{(vk, vj, r)} Step4:while 不收敛 do for 训练样本集中的每一个样本{(vk, vj, r)} do 计算vi 负采样L个样本,计算目标函数E 更新模型参数θ |
4 实验分析
表2 数据集统计情况 |
数据集 | 实体数 | 关系数 | 实体类型 | 关系类型 |
---|---|---|---|---|
亚马逊数据集 | 10,166 | 148,865 | 1 | 2 |
兵棋数据集 | 10,602 | 330,649 | 68 | 5 |
表3 不同模型在数据集上的性能比较 |
数据集 | 评估标准 | metapath2vec | GATNE-I | 本文模型 |
---|---|---|---|---|
亚马逊 数据集 | ROC-AUC | 0.9415 | 0.9625 | 0.9668 |
PR-AUC | 0.8748 | 0.9477 | 0.9502 | |
F1 | 0.9401 | 0.9136 | 0.9196 | |
兵棋数据集 | ROC-AUC | 0.7732 | 0.8002 | 0.8565 |
PR-AUC | 0.7404 | 0.7915 | 0.8197 | |
F1 | 0.7587 | 0.8107 | 0.8292 |