摘要
本申请公开了一种风险识别方法及相关装置,涉及人工智能技术领域,获取目标用户的目标知识图谱的特征数据;将特征数据输入至预先训练的风险识别模型,得到风险识别模型输出的风险识别结果;风险识别模型顺次包括多头注意力模型和分类模型,注意力头由空间注意力网络和通道注意力网络顺次连接构成,风险识别模型基于样本知识图谱的样本特征数据和风险标签预先训练得到,风险标签用于指示是否存在目标风险类型的风险。本申请结合多头注意力、通道注意力以及空间注意力机制。在注意力头中,通过空间注意力先行聚焦关键区域,通过通道注意力仅在空间注意力选中的高风险节点优化特征通道权重,降低计算复杂度提高金风险预测的准确性。