在审
图自监督学习方法、电子设备及存储介质
摘要
本发明提供一种基于图自监督学习方法、电子设备及存储介质。所述方法包括:获取原图数据,原图数据包括原图节点特征;将原图数据输入至图掩码对比学习模型,图掩码对比学习模型包括生成模块和对比模块;通过生成模块对原图数据进行掩码和编码操作,并从操作后的数据中解码出局部重建节点特征和整体恢复节点特征;将局部重建节点特征与原图节点特征进行对比学习,得到第一损失函数;通过对比模块对原图数据进行编码操作,并从操作后的数据中提取图判别特征;将整体恢复节点特征与图判别特征进行对比学习,得到第二损失函数;根据第一损失函数和第二损失函数,对图掩码对比学习模型进行参数优化。本发明能够提高图自监督学习的学习能力。
暂无引用专利



