-
公开(公告)号:CN118013123A
公开(公告)日:2024-05-10
申请号:CN202410172821.7
申请日:2024-02-07
Applicant: 东北大学
IPC: G06F16/9535 , G06N3/0895 , G06N3/0464
Abstract: 本发明提供一种基于跨尺度图对比学习的好友推荐方法,涉及网络科学技术领域。该方法具体包括:获取用户数据并对用户数据进行建模,构建图结构数据;利用随机采样增强方法对建模后的图结构数据进行数据增强;构建由两层图卷积层堆叠构成的图卷积神经网络作为编码器,利用编码器对数据增强得到的视图的节点向量进行编码,得到节点的向量表示;利用读出函数将节点的向量表示映射成图的向量表示;分别构建正负样本对,并采用信息最大化非参数互信息估计作为损失函数更新编码器和读出函数的参数,得到训练好的编码器和读出函数;利用训练好的编码器和读出函数对给定用户进行好友推荐,解决了图对比学习中忽略图的整体结构信息的问题。