图神经网络的校准方法、装置及相关设备

    公开(公告)号:CN118153652A

    公开(公告)日:2024-06-07

    申请号:CN202410184886.3

    申请日:2024-02-19

    Inventor: 杨成 石川 杨成栋

    Abstract: 本申请提供的一种图神经网络的校准方法、装置及相关设备,包括:响应于获取到图神经网络的校准图,确定校准图所有边中的任一条边为选中边,确定选中边的两个端节点;通过图神经网络确定端节点的节点嵌入,根据节点嵌入通过完成交叉熵损失最小化训练的多层感知机确定选中边对应的第一权重;确定每条边对应的第一权重生成第一权重集,根据第一权重集对图神经网络输出的预测结果进行调整,生成中间结果;确定中间结果中端节点对应的中间预测结果,对中间预测结果进行温度缩放的归一化计算,确定计算结果之间的欧几里得距离,生成第二权重;确定每条边对应的第二权重生成第二权重集,根据第一权重集和/或第二权重集对图神经网络进行校准调整。

Patent Agency Ranking