【技术实现步骤摘要】
基于图表示学习的身份保持对抗训练方法、装置、介质
[0001]本专利技术涉及图数据挖掘
,尤其涉及一种基于图表示学习的身份保持对抗训练方法、装置、介质。
技术介绍
[0002]图表示学习成为分析图结构数据的热门研究领域。在软件层面,图表示学习旨在学习一种编码函数,该函数充分利用图数据的优势,将具有复杂结构的图数据转换为保留多样化图属性和结构特征的低维空间中的密集表示。目前,图表示学习方法广泛应用于节点分类、异常检测、连边预测、标签推荐等各种图挖掘任务中。同时为实际生活中的大量应用问题带来了突破性的进展,例如在商品推荐场景中,用户对于商品的喜好程度的预测可以形式化为连边预测问题,其中节点代表用户或者商品,连边表示其喜好程度。药物靶点预测是连边预测任务对应的另一个重要的实际应用,其中节点表示药物或者蛋白质,连边表示两者之间是否能起作用。在金融风险控制场景中,已知借款人之间的交易关系,需要判断每个借款人的风险程度即其还款能力,防止将贷款发放给高风险人群。该场景是典型的节点分类问题,不同的类别表示不同的风险程度,每个借款人是一个节点,用连边表示借款人间的交易信息。
[0003]然而这些方法都忽略了现实世界中图数据的噪声,例如在推荐场景中用户可能存在误点击,对不喜欢的商品点了赞。同时在算法训练的过程中可能面临着过拟合的问题。基于对抗训练的图表示学习方法,通过在图表示学习方法之上引入对抗训练,将对抗样本及其对应的原始样本一同用于训练图表示学习模型,以提升训练数据的多样性从而解决上述问题。但该方法由于未保证对抗样本的 ...
【技术保护点】
【技术特征摘要】
1.一种基于图表示学习的身份保持对抗训练方法,其特征在于,包括:获取训练场景的图数据,定义所述图数据的每一个节点为用于表征所述训练场景的一个原样本,定义所述原样本的样本身份信息;生成每一个所述原样本对应的对抗样本;通过为所述对抗样本添加身份保持约束,将所述对抗样本保持所述原样本的样本身份信息;将所述对抗样本作为第一输入变量,输入至初始图表示学习模型,执行身份保持对抗训练;更新所述初始图表示学习模型,得到目标图表示学习模型,利用所述目标图表示学习模型预测所述训练场景中所述原样本在不同图挖掘任务下的输出。2.根据权利要求1所述的方法,其特征在于,所述生成每一个所述原样本对应的对抗样本,包括:将每一个所述原样本作为第二输入变量,输入至所述初始图表示学习模型进行训练,得到每一个所述原样本的表示;为每一个所述原样本的表示添加扰动因子,生成每一个所述原样本对应的对抗样本。3.根据权利要求2所述的方法,所述为每一个所述原样本的表示添加扰动因子,生成每一个所述原样本对应的对抗样本之前,还包括:生成所述扰动因子,所述扰动因子包括扰动强度与扰动方向,所述扰动强度用于确定所述对抗样本与所述原样本之间的距离,所述扰动方向用于确定所述对抗样本的生成方向;对于每一个所述原样本,生成的对抗样本的表示为:h
′
i
=h
i
+r
i
*d
i
其中,r
i
、d
i
、h
i
分别为第i个所述原样本对应的扰动强度、扰动方向、表示。4.根据权利要求3所述的方法,其特征在于,所述生成所述扰动因子,包括:确定所述扰动强度,包括:将每一个节点的邻居节点的表示作为输入,输入至感知机模型,通过自适应学习得到每一个所述节点对应的扰动强度;所述扰动强度表示为:r
i
=ReLU(W2*ReLU(W1h
i
))其中,W1、W2是可学习的参数,ReLU是激活函数;构造所述扰动强度满足的约束条件,得到第一损失函数;所述第一损失函数表示为:其中,N表示所述节点的数量。5.根据权利要求4所述的方法,其特征在于,所述生成所述扰动因子,还包括:确定所述扰动方向,包括:通过最大化所述初始图表示学习模型的原损失函数方向,确定所述扰动方向,所述扰动方向表示为:
其中,L
basic
表示所述初始图表示学习模型的原损失函数。6.根据权利要求5所述的方法,其特征在于,所述通过为所述对抗样本添加身份保持约束,将所述对抗样本保持所述原样本的样本身份信息,包括:构造所述身份保持约束,包括:负采样每一个所述节点,构造所述身份保持约束,得到第二损失函数;所述第二损失函数表示为:其中,h
k
为第k个所述节点的表示,i≠k;p(n)表示每一个所述节点被采样到的概率,按均匀分布负采样;K是负采样的个数;σ(h
′
i
,h
i
)表示原样本与其相对应的对抗样本之间的关系;σ(h
′
i
...
【专利技术属性】
技术研发人员:沈华伟,岑科廷,曹婍,徐冰冰,程学旗,
申请(专利权)人:中国科学院计算技术研究所,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。