基于金字塔图池化Transformer的图级别数据分类方法技术

技术编号:44330031 阅读:14 留言:0更新日期:2025-02-18 20:37
本发明专利技术涉及一种基于金字塔图池化Transformer的图级别数据分类方法,包括如下步骤:获取输入的图级别数据,利用图神经网络提取图的局部邻域信息,得到编码特征;基于编码特征,通过多个并行的图池化操作得到多个池化图并分别进行上采样,通过将图对应位置的节点特征叠加得到金字塔特征;将金字塔特征与预设的token拼接,利用transformer得到表示整个图的嵌入向量;基于嵌入向量,利用多层感知机制得到图级别数据分类结果。与现有技术相比,本发明专利技术通过采用金字塔池化transformer,不仅能够捕获远近距离节点之间的依赖关系,而且还能够减少transformer中节点序列的长度和提取多尺度信息。

【技术实现步骤摘要】

本专利技术涉及计算机,尤其是涉及一种基于金字塔图池化transformer的图级别数据分类方法。


技术介绍

1、在计算机视觉领域,卷积神经网络由于其强大的特征提取能力,已经被广泛应用于处理图片、视频和音频等数据,在卷积神经网络中,对于图级别数据,通过增加网络的层数来增加感受野。然而,随着网络层数的增加,网络层数的增加会带来过平滑和过度挤压的问题,导致图神经网络的性能下降。

2、在图级别任务中,需要预测一个描述整个图的目标值,这个目标值可能依赖于远距离节点的依赖关系,图卷积神经网络有限的感受野不足以捕获远距离节点关系,增加网络的层数会带来过平滑和过度挤压的问题。

3、因此,设计一种新的图级别数据分类架构突破消息传递限制来解决过平滑和过度挤压的问题至关重要。


技术实现思路

1、本专利技术的目的就是为了克服上述现有技术存在的缺陷而提供一种基于金字塔图池化transformer的图级别数据分类方法,以实现图级别任务下,图级别数据的分类。

2、本专利技术的目的可以通过以下技术方案来实本文档来自技高网...

【技术保护点】

1.一种基于金字塔图池化Transformer的图级别数据分类方法,其特征在于,包括如下步骤:

2.根据权利要求1所述的一种基于金字塔图池化Transformer的图级别数据分类方法,其特征在于,所述的步骤S1中,通过聚合图中当前节点邻域内节点的消息更新当前节点的隐藏嵌入,实现图的局部邻域信息的提取。

3.根据权利要求2所述的一种基于金字塔图池化Transformer的图级别数据分类方法,其特征在于,聚合图中当前节点邻域内节点的消息更新当前节点的隐藏嵌入的过程采用下式实现:

4.根据权利要求1所述的一种基于金字塔图池化Transformer的图级别数据...

【技术特征摘要】

1.一种基于金字塔图池化transformer的图级别数据分类方法,其特征在于,包括如下步骤:

2.根据权利要求1所述的一种基于金字塔图池化transformer的图级别数据分类方法,其特征在于,所述的步骤s1中,通过聚合图中当前节点邻域内节点的消息更新当前节点的隐藏嵌入,实现图的局部邻域信息的提取。

3.根据权利要求2所述的一种基于金字塔图池化transformer的图级别数据分类方法,其特征在于,聚合图中当前节点邻域内节点的消息更新当前节点的隐藏嵌入的过程采用下式实现:

4.根据权利要求1所述的一种基于金字塔图池化transformer的图级别数据分类方法,其特征在于,所述的步骤s2包括如下子步骤:

5.根据权利要求4所述的一种基于金字塔图池化transformer的图级别数据分类方法,其特征在于,所述的步骤s202中,通过图卷积和反池化操作得到恢复的池化图的过程采用下式实现:

6.根据权利要求1所述的一种基于金字塔图池化trans...

【专利技术属性】
技术研发人员:黄德双李志鹏王思果张钦虎
申请(专利权)人:宁波数字孪生东方理工研究院
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1