[发明专利]安全性预测方法及装置、介质、设备

权利要求书

1.一种基于节点和边注意力机制的安全性预测方法，其特征在于，包括：

获取待预测主体的第一交易信息，以及与所述待预测主体关联的其他主体的第二交易信息，并根据所述待预测主体以及与所述待预测主体关联的其他主体，生成初始特征图；

从所述初始特征图中确定与所述待预测主体所在的待预测节点对应的邻居节点，并利用预设的安全性预测模型中的第一处理单元对与所述待预测节点对应的第一交易信息以及与所述邻居节点对应的第二交易信息处理，得到待预测节点特征；

从所述初始特征图中确定待预测节点与所述邻居节点对应的待预测边，以及与所述待预测边对应的邻居边，并根据第一交易信息以及第二交易信息计算所述邻居边的第一边信息以及待预测边的第二边信息；用预设的安全性预测模型中的第二处理单元对所述第一边信息以及第二边信息进行处理，得到待预测边特征；

对所述待预测节点特征以及待预测边特征进行拼接，得到目标待预测特征，并根据所述目标待预测特征对所述待预测主体在未来时间段内处于安全状态的概率进行预测。

2.根据权利要求1所述的安全性预测方法，其特征在于，根据所述待预测主体以及与所述待预测主体关联的其他主体，生成初始特征图，包括：

对所述待预测主体以及与所述待预测主体关联的其他主体进行抽象处理，得到多个节点；

对所述待预测主体与所述其他主体之间的关联关系进行抽象处理，得到多个边；

对各所述节点以及各所述边进行连接，生成所述初始特征图。

3.根据权利要求1所述的安全性预测方法，其特征在于，所述第一处理单元中包括第一全连接层、第一归一化层以及第一加权求和层；

其中，利用预设的安全性预测模型中的第一处理单元对与所述待预测节点对应的第一交易信息以及与所述邻居节点对应的第二交易信息处理，得到待预测节点特征，包括：

利用所述第一全连接层对所述第一交易信息以及第二交易信息进行计算，得到所述邻居节点对所述待预测节点的节点重要性系数；

利用所述第一归一化层对所述节点重要性系数进行计算，得到每一个邻居节点在所有邻居节点中的权重值；

利用所述第一加权求和层对每一个邻居节点的权重值以及该邻居节点的第二交易信息进行加权求和，得到所述待预测节点特征。

4.根据权利要求3所述的安全性预测方法，其特征在于，所述待预测节点特征包括第一层级节点特征以及第二层级节点特征；

所述第一层级节点特征是与所述待预测节点直接相连的第一层级邻居节点的特征，所述第二层级节点特征是与所述第一层级邻居节点直接相连的第二层级邻居节点的特征；

所述待预测主体包括用户和/或设备。

5.根据权利要求1所述的安全性预测方法，其特征在于，所述第二处理单元包括第二全连接层、第二归一化层以及第二加权求和层；

其中，利用预设的安全性预测模型中的第二处理单元对所述第一边信息以及第二边信息进行处理，得到待预测边特征，包括：

利用所述第二全连接层对所述第一边信息以及第二边信息进行计算，得到所述邻居边对所述待预测边的边重要性系数；

利用所述第二归一化层对所述待预测边的边重要性系数进行计算，得到每一个第一邻居节边在所有的第一邻居节边中的权重值；

利用所述第二加权求和层对每一个第一邻居节边的权重值以及该邻居边的边信息进行加权求和，得到所述待预测边特征。

6.根据权利要求5所述的安全性预测方法，其特征在于，所述待预测边特征包括第一层级边特征以及第二层级边特征；

所述第一层级边特征是与所述待预测节点直接相连的第一层级边的特征，所述第二层级边特征是与所述第一层级边的终止节点直接相连的第二层级边的特征。