公开(公告)号：CN118245805A

公开(公告)日：2024-06-25

申请号：CN202410452256.X

申请日：2024-04-15

Applicant: 中国联合网络通信集团有限公司

Inventor： 佟晶 ,  范永安 ,  张智勇 ,  汪悦 ,  谢文锋 ,  钟颋 ,  杨军 ,  邓俊琼

IPC: G06F18/214 ,  G06N3/042 ,  G06N3/0475 ,  G06N3/088

Abstract: 本申请提供基于双视角的对比学习模型训练方法、装置、设备及介质，该方法包括：对原始图数据进行边节点互换处理，得到对照图数据，其中，原始图数据用于指示多个数据节点和多个数据节点之间的连接关系，对照图数据用于指示多个边节点和多个边节点之间的连接关系，边节点是由两个数据节点连接得到的；根据原始图数据和对照图数据对基于图神经网络的对比学习模型进行训练处理，得到对比学习模型输出的第一对比损失；若第一对比损失小于预设损失，则确定基于图神经网络的对比学习模型训练完成。本申请的方法，实现了通过将节点和边信息结合形成双视角的图数据，以使对比学习模型能够深入学习，从而提高模型训练效率和有效性的技术效果。