CN112633481B - 一种多跳图卷积神经网络模型系统及训练方法 - Google Patents

一种多跳图卷积神经网络模型系统及训练方法 Download PDF

Info

Publication number
CN112633481B
CN112633481B CN202011642703.6A CN202011642703A CN112633481B CN 112633481 B CN112633481 B CN 112633481B CN 202011642703 A CN202011642703 A CN 202011642703A CN 112633481 B CN112633481 B CN 112633481B
Authority
CN
China
Prior art keywords
graph
graph convolution
hop
neural network
network model
Prior art date
Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
Active
Application number
CN202011642703.6A
Other languages
English (en)
Other versions
CN112633481A (zh
Inventor
刘勋
宗建华
夏国清
陈华珍
于明清
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
South China Institute Of Software Engineering Gu
Original Assignee
South China Institute Of Software Engineering Gu
Priority date (The priority date is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the date listed.)
Filing date
Publication date
Application filed by South China Institute Of Software Engineering Gu filed Critical South China Institute Of Software Engineering Gu
Priority to CN202011642703.6A priority Critical patent/CN112633481B/zh
Publication of CN112633481A publication Critical patent/CN112633481A/zh
Application granted granted Critical
Publication of CN112633481B publication Critical patent/CN112633481B/zh
Active legal-status Critical Current
Anticipated expiration legal-status Critical

Links

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/04Architecture, e.g. interconnection topology
    • G06N3/045Combinations of networks
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F16/00Information retrieval; Database structures therefor; File system structures therefor
    • G06F16/30Information retrieval; Database structures therefor; File system structures therefor of unstructured textual data
    • G06F16/35Clustering; Classification
    • G06F16/353Clustering; Classification into predefined classes
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F18/00Pattern recognition
    • G06F18/20Analysing
    • G06F18/21Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
    • G06F18/214Generating training patterns; Bootstrap methods, e.g. bagging or boosting
    • G06F18/2155Generating training patterns; Bootstrap methods, e.g. bagging or boosting characterised by the incorporation of unlabelled data, e.g. multiple instance learning [MIL], semi-supervised techniques using expectation-maximisation [EM] or naïve labelling
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F18/00Pattern recognition
    • G06F18/20Analysing
    • G06F18/24Classification techniques
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F18/00Pattern recognition
    • G06F18/20Analysing
    • G06F18/25Fusion techniques
    • G06F18/253Fusion techniques of extracted features
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/04Architecture, e.g. interconnection topology
    • G06N3/048Activation functions
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/08Learning methods
    • YGENERAL TAGGING OF NEW TECHNOLOGICAL DEVELOPMENTS; GENERAL TAGGING OF CROSS-SECTIONAL TECHNOLOGIES SPANNING OVER SEVERAL SECTIONS OF THE IPC; TECHNICAL SUBJECTS COVERED BY FORMER USPC CROSS-REFERENCE ART COLLECTIONS [XRACs] AND DIGESTS
    • Y02TECHNOLOGIES OR APPLICATIONS FOR MITIGATION OR ADAPTATION AGAINST CLIMATE CHANGE
    • Y02DCLIMATE CHANGE MITIGATION TECHNOLOGIES IN INFORMATION AND COMMUNICATION TECHNOLOGIES [ICT], I.E. INFORMATION AND COMMUNICATION TECHNOLOGIES AIMING AT THE REDUCTION OF THEIR OWN ENERGY USE
    • Y02D30/00Reducing energy consumption in communication networks
    • Y02D30/70Reducing energy consumption in communication networks in wireless communication networks

Abstract

本发明提供了一种多跳图卷积神经网络模型及其训练方法,所述模型依次包括输入层、同时捕捉多阶邻域信息、引入自连接和注意力乘数的多跳图卷积层、融合多跳邻域信息的SP信息融合池化层和softmax函数多分类输出层,所述模型对应的训练方法为先预处理得到预处理特征后,将其输入模型训练的高效训练方法。本实施例采用多阶图卷积层增加模型宽度、减少了模型计算复杂度和参数量的同时,还能聚集多阶邻域节点信息、赋予自身节点更高权重,且通过分组注意结合注意力乘数调整不同阶邻域节点分类贡献度,不仅保持了更大的感受野,还提高了模型的有效性、稳定性和分类表现。

Description

一种多跳图卷积神经网络模型系统及训练方法
技术领域
本发明涉及图像处理及深度学习技术领域,特别是涉及一种多跳图卷积神经网络模型及其训练方法。
背景技术
图卷积神经网络中的高阶图卷积网络因其能够聚集更丰富节点信息、扩宽模型感受野及提升分类表现而在图像处理及深度学习技术领域得以广泛应用。如,高阶图卷积网络模型设计了一种包括使用不同权重参数的一阶图卷积到P阶图卷积,利用两个高阶图卷积层学习高阶节点之间的关系和聚集不同距离邻域节点信息的高阶图卷积,且在每个高阶图卷积聚集了不同距离的邻域信息后,利用列连接将这些邻域信息进行拼接,再通过一个全连接层进行邻域信息融合的网络架构,
通过聚集多阶邻域信息,在一定程度上扩宽了模型的感受野、提升了分类精度,但由于其采用不同阶卷积不同的权重参数、堆叠多个高阶图卷积层,及全连接融合不同距离邻域信息的技术必要会造成复杂度增大,参数量成倍增加,进而增加了过拟合风险;高低阶图卷积网络模型设计了一种既包括一阶的低阶图卷积,也包括二阶到P阶的高阶图卷积,所有阶数图卷积采用权重共享机制,且在每个高低阶图卷积聚集不同距离的邻域信息后,利用信息融合层进行融合的网络架构,在一定程度上较少了参数量,但其同样堆叠了多层高低阶图卷积层使得模型的参数量和复杂度并没有显著减少,同样不能避免过拟合风险。
虽然,随后提出的简化图卷积通过反复移除图卷积层之间的非线性激活函数并折叠正则化邻接矩阵,及将参数矩阵重参数化为单一参数矩阵的方法减少了高阶图卷积模型的参数量和复杂度,但由于其过渡简化而只能捕捉k阶邻域信息,造成了大量节点信息的丢失,限制了模型的表达能力。
因此,如何在保留现有高阶图卷积能够捕捉更多更丰富节点信息及全局结构的分类优势的同时,对其进行适当简化,减少计算复杂度和参数量,避免过拟合的风险,进而提高分类的表现是非常有意义的。
发明内容
本发明的目的是通过对现有高阶图卷积的适当简化,减少计算复杂度和参数量,避免过拟合风险的同时,还能够捕捉更多更丰富节点信息及全局结构,进一步提高分类表现。
为了实现上述目的,有必要针对上述技术问题,提供一种多跳图卷积神经网络模型及其训练方法。
第一方面,本发明实施例提供了一种多跳图卷积神经网络模型,所述多跳图卷积神经网络模型模型依次包括输入层、多跳图卷积层、信息融合池化层和输出层;
所述输入层,用于接收训练数据集的图特征;
所述多阶图卷积层,用于根据所述图特征进行零阶到k阶的多跳图卷积运算,得到图卷积数据;
所述信息融合池化层,用于根据所述图卷积数据进行多跳邻域的特征融合,得到融合数据;
所述输出层,用于根据所述融合数据输出模型结果。
进一步地,多阶图卷积层通过如下方式产生:
对不同阶数图卷积按照相邻三阶为一组进行分组;
组内图卷积采用注意融合,组间图卷积采用注意力乘数调整权重。
进一步地,在所述多跳图卷积层的任一阶数图卷积引入新的自连接。
进一步地,所述多跳图卷积层包括相邻组权重共享的一阶图卷积到k阶图卷积,表示为:
其中,X是图的输入矩阵,是参数矩阵,/>是图的正则化邻接矩阵,/>是图卷积的最大阶数,/>,/>为注意融合函数,/>为对应不同组图卷积的注意力乘数。
进一步地,所述多跳图卷积神经网络模型的输出层的输出MulStepNET表示为:
其中,为激活函数,/>为信息融合函数,/>为多分类输出函数。
进一步地,所述注意融合函数SA的注意融合公式为:
其中,为第i-2阶图卷积和第i阶图卷积的注意融合输出。
进一步地,所述激活函数为ReLU非线性激活函数。
进一步地,所述信息融合池化层采用SP求和信息融合池化,其计算公式如下:
第二方面,本发明实施例提供了一种多跳图卷积神经网络模型的训练方法,所述训练方法的步骤包括:
获取所述训练数据集,并根据所述训练数据集的类型得到所述训练数据集的图特征,所述图特征包括图的输入矩阵和正则化邻接矩阵;
将所有不同阶数的所述图的正则化邻接矩阵采用组内注意融合和组间加权作和的方法,得到多幂邻接矩阵,并将所述多幂邻接矩阵与所述图的输入矩阵作积,得到预处理特征;
将所述预处理特征输入所述多跳图卷积神经网络模型,进行特征训练,得到训练结果。
进一步地,所述将所述预处理特征输入所述多跳图卷积神经网络模型,进行特征训练,得到训练结果的步骤包括:
将所述多跳图卷积神经网络模型的参数矩阵进行随机初始化,并将所述注意力乘数初始化为特定值;
将所述预处理特征输入所述多跳图卷积神经网络模型,根据学习率优化结合所述训练数据集的属性调整所述注意力乘数,并采用损失函数和梯度下降法进行训练,得到收敛的参数矩阵。
上述本申请提供了一种多跳图卷积神经网络模型及其训练方法,通过所述模型及其训练方法,实现了采用仅有输入层、多跳图卷积层、SP求和信息融合池化层及softmax函数输出层的多跳图卷积神经网络模型,结合该模型训练前的特征预处理方法,并依此得到精准分类的效果。与现有技术相比,该模型及其训练方法在实际分类应用上,通过对现有高阶图卷积的适当简化,在减少模型的计算复杂度和参数量,提升模型训练效率,避免过拟合风险的同时,还能够捕捉更多更丰富节点信息及全局结构,进一步提高模型的稳定性、学习能力和分类精度。
附图说明
图1是本发明实施例中多跳图卷积神经网络模型及其训练方法的应用场景示意图;
图2是多跳图卷积神经网络模型的示意图;
图3是采用SP信息融合池化层的多跳图卷积神经网络模型的示意图;
图4是图3多跳图卷积神经网络模型的训练方法的流程示意图;
图5是图4中步骤S13将预处理特征输入多跳图卷积神经网络模型进行特征训练的流程示意图;
图6是不同跳数的无自连接、无注意力乘数的多跳图卷积神经网络模型与有自连接、无注意力乘数的多跳图卷积神经网络模型基于半监督数据集的分类精度示意图;
图7是本发明实施例中计算机设备的内部结构图。
具体实施方式
为了使本申请的目的、技术方案和有益效果更加清楚明白,下面结合附图及实施例,对本发明作进一步详细说明,显然,以下所描述的实施例是本发明实施例的一部分,仅用于说明本发明,但不用来限制本发明的范围。基于本发明中的实施例,本领域普通技术人员在没有作出创造性劳动前提下所获得的所有其他实施例,都属于本发明保护的范围。
本发明提供的多跳图卷积神经网络模型是对现有高阶图卷积的简化,该模型及其训练方法,可以应用于如图1所示的终端或服务器上。其中,终端可以但不限于是各种个人计算机、笔记本电脑、智能手机、平板电脑和便携式可穿戴设备,服务器可以用独立的服务器或者是多个服务器组成的服务器集群来实现。服务器可采用多跳图卷积神经网络模型(MulStepNET)及对其对应的训练方法完成现有高阶图卷积神经网络模型和高低阶图卷积神经网络模型能够完成的分类任务,并将模型的分类预测结果发送至终端,供终端的使用者查看分析。
在一个实施例中,如图2所示,提供了一种多跳图卷积神经网络模型包括输入层1、多跳图卷积层2、信息融合池化层3和输出层4;
所述输入层1,用于接收训练数据集的图特征;
所述多跳图卷积层2,用于根据所述图特征进行零阶到k阶的多跳图卷积运算,得到图卷积数据;
所述信息融合池化层3,用于根据所述图卷积数据进行多跳邻域的特征融合,得到融合数据;
所述输出层4,用于根据所述融合数据输出模型结果。
其中,多跳图卷积层2和信息融合池化层3都只有1个,即多跳图卷积神经网络模型的结构为:输入层1与跳阶图卷积层2相连,多跳图卷积层2与信息融合池化层3相连,信息融合池化层3再与采用softmax函数进行多分类输出的输出层4相连。
本实施例中的多跳图卷积层由对不同阶数图卷积按照相邻三阶为一组进行分组,组内图卷积采用注意融合,组间图卷积采用注意力乘数调整权重得到,且在任一阶数图卷积引入新的自连接得到一种新的高阶图卷积。其引入新的自连接是指将任一阶数的图的正则化邻接矩阵都加上同维数的单位矩阵作为新的图的正则化邻接矩阵输入模型。需要注意的是,考虑到自身节点不存在自连接(即自身节点与自身节点有一条边)的邻接矩阵的对角线位置的元素值为0,将该邻接矩阵输入模型训练时可能会因忽略掉自身节点信息而影响模型的分类效果的情况,本实施例中原有的图的正则化邻接矩阵就是含自连接的,而本申请强调的自连接是指在原有的自连接正则化邻接矩阵基础上再次引入自连接,进一步增加自身节点的权重,即:
其中,为第i阶含自连接的正则化邻接矩阵,/>为零阶含自连接的正则化邻接矩阵,/>为与/>同维数的单位矩阵,且/>
为图的邻接矩阵,/>为/>的度矩阵,/>。如在二阶图卷积引入新的自连接时的效果为/>,当然也可以选择在其他阶数图卷积引入自连接,其他阶情况依此类推可以得到,在此不详述。其对引入自连接的不同阶数图卷积按照相邻邻域节点权重相似,而不相邻邻域节点权重不同的原则采用除0阶以外的相邻三阶为一组的方法进行如下分组:/>
得到不同阶图卷积的分组后,采用SA注意融合对组内的不同阶图卷积进行注意融合得到新的融合图卷积,并利用一系列注意力乘数来调整不同组图卷积的权重,通过这些注意力乘数给予分类更重要的组更高的权重,不重要的组较小的权重的方法调整不同组邻域节点的分类贡献度,则最大阶数为k的多跳图卷积分别表示为:
其中,注意融合函数SA的注意融合公式为:
为第i-2阶图卷积和第i阶图卷积的注意融合输出。
上述多跳图卷积包括包括相邻组权重共享的一阶图卷积到k阶图卷积,使得多跳图卷积的参数量与一阶图卷积的参数量保持一致,实现了权重共享和降低参数量的效果,具体体现在模型公式中参数W的选择。该多跳图卷积不仅通过赋予自身节点更高的权重,还通过/>来获得两阶和更高阶邻域节点信息和通过分组注意乘数灵活调整相邻阶图卷积邻域节点的贡献度差异。为了弥补一层架构的学习能力,在实际应用中可以根据需求将选取较大的k值即设置较大的宽度,即通过增加阶数(即增加跳数、增加宽度)来代替增加图卷积层数(增加深度)的方式增加模型的感受野,进而增强模型的学习能力。其中,多跳图卷积层的阶数k可以为二阶及其以上阶数中的一种,或者任意复数种阶数的组合。假设上述多跳图卷积神经网络模型的输出层的输出为MulStepNET,则表示为:
其中,X是图的输入矩阵,是参数矩阵,/>是图的正则化邻接矩阵,/>是图卷积的最大阶数(最大跳数),/>,/>为注意融合函数,/>为不同组图卷积的注意力乘数,/>为激活函数,/>为信息融合函数,/>为多分类输出函数。
本实施例中通过构建只有一层且同时引入分组调整权重和自连接的多跳图卷积层的网络架构,不仅减少了模型的计算复杂度和参数量,有效的提升了模型训练效率,而且通过赋予自身节点更高权重和不同组图卷积采用组内注意融合,组间设置与重要性成正比的注意力乘数来调整分类贡献度的方法,使得模型聚集更多阶更丰富的邻域信息的同时,能合理区分不同阶邻域的分类贡献度,进而很好的保证了模型的实用性和有效性,达到进一步提高分类精度的目的。
在实际应用于大规模的分类训练时,需要先计算出,由于/>通常是一个具有m个非零元素的稀疏矩阵,且多跳图卷积均采用权重共享机制,采用从右到左的乘法来计算,如,当k=2,用/>相乘来求/>,同理,/>,依此类推通过左乘k-1阶图卷积来计算k阶图卷积,即/>,进而,利用矩阵的加法和数乘容易得到多阶图卷积的分组融合结果,这一计算方法有效的降低了计算复杂度。此外,由于不同阶图卷积采用权重共享机制,多阶图卷积的参数量和一阶图卷积的参数量是相同的,在一定程度上减少了计算的复杂度,保证了多跳图卷积计算的高效性。
优选地,激活函数为/>非线性激活函数。
其中,ReLU激活函数是用来进行非线性处理的。由于线性模型的表达力不够,且有些数据特征不一定是线性可分的,为了解决这一问题就在信息融合池化层之后采用激活函数进行非线性处理,常用的非线性激活函数包括sigmoid、tanh和ReLU、ElU、PReLU等,上述激活函数都可使用,但各有优劣,而本实施例中采用神经网络中用的最多的ReLU函数,它的公式定义如下:
即,保留大于等于0的值,其余所有小于0的数值直接改写为0。通过这种方法对卷积后产生的特征图中的值进行映射,就可以在特征提取时直接舍弃掉那些不相关联的数据,操作更方便。
由于非线性激活函数可以提高模型的表达能力,但对图分类任务用处不是很大,尤其是对于本实施例中只有一层多跳图卷积层的宽度图卷积神经网络模型而言,是可以省略非线性激活的处理,进一步减少模型计算的复杂度,但可以能会损失一点精度,总体来说对模型的分类精度影响不大,因此,在该模型的实际应用中可以根据分类任务的具体情况决定是否需要使用非线性激活函数,若对精度上要求比较高可以选择使用非线性激活函数,若希望减少模型的计算复杂度提升模型的性能,可以省略非线性激活的处理。
优选地,信息融合池化层采用SP求和信息融合池化来融合从零阶到k阶的不同跳邻域节点信息,具体的池化公式为:
对应的采用SP信息融合的多跳图卷积能聚集更多更丰富的邻域信息获得全局图结构信息的同时,还考虑了自身节点在分类预测时更加重要,及不同组的邻域节点的分类贡献度差异等因素,如图3所示的模型表达式为:
其中,H为多跳图卷积层的输出值,即为该模型的softmax函数输出层的输入值。
以一个具体的三阶(三跳)实施例来说明上述本实施例中的信息融合方式,多阶的情况类似。假设邻域的阶数k=3,假设其零阶邻域为,一阶邻域为/>,二阶邻域为/>,三阶邻域为/>,/>分别为/>和/>图卷积组的注意力乘数,则SP求和信息融合过程为:
,/>,/>,/>,/>,则
本实施例采用SP信息融合的多跳图卷积算法实现过程如下:
输入:
卷积运算:
信息融合:
非线性激活:
本实施例中图网络先输入到多跳图卷积进行上述的算法处理,再使用SP求和信息融合池化层来混合不同邻域的零阶到高阶的特征,经过非线性激活后输入softmax函数输出层得到分类概率结果的方法,能够在学习过程中保留更多更丰富的特征信息进行全局图拓扑的学习的同时,还考虑了预测时自身节点更重要的作用,且区分了不同组邻域节点的分类贡献度差异,进而很好地提升了模型的有效性及学习效果。
在一个实施例中,如图4所示,任一上述多跳图卷积神经网络模型的训练方法的步骤包括:
S11、获取所述训练数据集,并根据所述训练数据集的类型得到所述训练数据集的图特征,所述图特征包括图的输入矩阵和正则化邻接矩阵;
其中,训练数据集根据实际的分类需求进行选择,比如文本分类数据集、半监督分类数据集、多视图分类3D数据集等。每一种分类任务的数据集内容不相同,其类型也就不同,对应的预处理后得到图特征矩阵(即模型的图的输入矩阵)和正则化邻接矩阵的方法也是不同的,如当需要进行文本分类时,就需要将包括文档和标题的语料集数据进行处理构建对应的语料文本图网络,根据语料文本图网络得到模型训练使用的图的输入矩阵和图的正则化邻接矩阵。对于其它情形的数据集,如半监督数据集或多视图分类数据集等都有对应类型的预处理方法,在使用本实例中的模型进行分类时只需按照分类任务类型对应的常规方法将该任务对应的数据集转换为图的输入矩阵和图的正则化邻接矩阵即可。本申请后续实施例中,均以如表1所示的半监督数据集为例进行相关说明。
表1 半监督分类经典数据集信息表
S12、将所有不同阶数的所述图的正则化邻接矩阵采用组内注意融合和组间加权作和的方法,得到多幂邻接矩阵,并将所述多幂邻接矩阵与所述图的输入矩阵作积,得到预处理特征;
其中,模型训练之前,先得到多幂邻接矩阵是多跳图卷积神经网络模型计算的一大优势,由于本申请构建的是只有一层多跳图卷积层,没有多层图积层,则上述对应的SP求和信息融合计算式可优化为:
且由于正则化邻接矩阵、/>(即/>)和/>都已知,则由矩阵乘法易得/>,则由矩阵数乘及加法易得/>,可见/>是通过逐元素运算得到的,其每个元素的空间位置和/>是相同的,是保留了图拓扑的算子,即可将/>当做一个预处理后的图的邻接矩阵用于后续的模型训练使用。在经过上述预处理得到多幂邻接矩阵/>后,由于输入矩阵X已知的,即可作多跳特征变换得到/>作为预处理特征矩阵,再经过多跳线性变换可得多跳图卷积的输出为/>。显然,将上述的预处理特征矩阵送入模型进行训练,在一定程度上减少了机器训练的复杂度和难度,从而保证了模型训练的高效性。
S13、将所述预处理特征输入所述多跳图卷积神经网络模型,进行特征训练,得到训练结果。
其中,如图5所示,所述将所述预处理特征输入所述多跳图卷积神经网络模型,进行特征训练,得到训练结果的步骤S13包括:
S131、将所述多跳图卷积神经网络模型的参数矩阵进行随机初始化,并将所述注意力乘数初始化为特定值;
其中,对模型参数矩阵进行随机初始化的方法有:权重服从高斯分布的Gaussian初始化、权重为均匀分布的Xavier初始化,及均值为0、方差为2/n的MSRA初始化。本实施例中的多跳图卷积神经网络模型参数矩阵的随机初始化时,可以根据实际分类需求结合上述三种初始化的特点进行选择,不会影响模型的应用效果。需要注意的是模型注意力乘数的初始化值都设为1,后续在训练的过程中根据学习率优化结合所述训练数据集的属性进行调整,不同数据集对应的最大图卷积阶数不相同、各阶图卷积的注意力乘数也不同的。本实施例中要先确定Pubmed、Cora和Citeseer数据集对应的最大图卷积阶数后,在不同数据集的训练过程中以分类精度为依据分别在对应阶数的模型上进行注意力乘数的调整。
S132、将所述预处理特征输入所述多跳图卷积神经网络模型,,根据学习率优化结合所述训练数据集的属性调整所述注意力乘数,并采用损失函数和梯度下降法进行训练,得到收敛的参数矩阵。
其中,多跳图卷积神经网络模型训练的过程为:(1)对选用的训练数据集中的有效特征数据进行预处理得到的预处理特征并输入不同阶数的模型,使用初始化注意力乘数和初始化参数矩阵及最大学习率进行正向传播得到分类结果,选取分类精度最高的阶数的模型作为该训练数据集后续训练的基准模型,如Pubmed、Cora和Citeseer数据集上最大阶数分别为21、8和4的多跳图卷积神经网络模型;(2)确定多跳图卷积神经网络模型的宽度值(最大阶数)后,依据低阶邻域比高阶邻域更重要,且相邻邻域重要性相似的原则对不同阶邻域的注意力乘数值依次进行调整,并将调整后的注意力乘数输入模型进行训练,经过正向传播得到分类结果,再通过损失函数计算交叉熵使用反向传播更新参数矩阵的梯度下降算法进行训练直至收敛,得到当前注意力乘数下的收敛时的参数矩阵,并记录对应的分类精度;(3)重复步骤2的操作,不断调整注意力乘数进行训练,直至得到分类精度更高的参数矩阵,作为该模型对应注意力乘数下的收敛的参数矩阵,用作后续的分类测试使用,基于Pubmed、Cora和Citeseer数据集最大阶数的MulStepNet模型对应的注意力乘数如表2所示。
表2 MulStepNet基于Pubmed、Cora和Citeseer数据集的测试精度比对表
表2说明:表中k是图卷积的最大阶数,模型的准确率以百分比表示,且该数字是10次运行的平均值。
本实施例中,模型训练使用的训练数据集为半监督分类节点的训练数据集,根据该类训练数据集的特点,选用的损失函数为:
为带有标签的顶点(节点)的集合,M是类的数目,/>表示标签节点的实标签,并表示softmax(输入标签节点)预测的0到1之间的概率值。当初始化参数矩阵后,基于所有的训练数据集得到初始的损失函数值,若损失函数较大,则说明神经网络性能不好,需要使用梯度下降法,即通过计算上述损失函数关于该神经网络模型参数的偏导数,再不断地调整更新权值参数再训练,直至损失函数缩小至一个可接受范围,结束模型的整个训练过程,即可达到收敛的参数矩阵。
本申请实施例中,设计了只有一层图卷积的多跳图卷积神经网络模型及其特征预处理的模型训练方法,采用多跳图卷积增加跳数代替深度图卷积,在没有多层图卷积层的情况下,既减少了模型的计算复杂度、参数量及训练难度,还能学习到图的全局信息和更丰富的邻域信息、突出自身节点邻域信息重要性及区分不同组邻域节点的贡献度差异,再结合模型训练时的特征预处理方法,不仅保持了更大的感受野,还提高了模型的有效性、稳定性和分类表现。
本申请实施例中模型基于半监督分类数据集分别进行分类训练,并分别与现有的图卷积神经模型的测试效果进行比对,结果如下表3所示:
表3 MulStepNet与现有图卷积模型基于相同半监督数据集的分类精度比对表
表3说明:表中的准确率以百分比表示,且该数字是10次运行的平均值。
基于上述实验结果表3可知,本实施例提出了一种只具有一层既能同时聚集多阶邻域节点信息和加大自身节点权重,且区分不同组邻域节点分类贡献度的多跳图卷积,及混合不同跳邻域特征的SP信息融合池化层的多跳图卷积网络模型MulStepNet,不仅在分类学习中保留更多更丰富的邻域特征信息,学习了全局图拓扑,扩宽了感受野,还简化了现有高阶图卷积模型,减少了模型的计算复杂度、参数量和训练难度,进一步提升了学习效率,避免了过拟合风险,还通过引入新的自连接加大自身节点权重,和对不同阶图卷积引入分组及注意力乘数区分不同组邻域节点贡献度差异的方法,进一步提高了模型表达能力、学习能力和有效性,基于三组基准半监督分类数据集上的实验结果表明应用基于多跳度图卷积网络模型与采用现有的其他基线方法相比,在分类精度、参数量、复杂度及稳定性等方面都有着较大的优势。
为了得到上述实施例中三组数据分别对应多跳图卷积模型的最大阶数(跳数)及确定自身节点对模型分类预测的重要性,本申请还基于不同阶数的无注意力乘数、无自连接的多跳图卷积和无注意力乘数、有自连接的多跳图卷积的分类精度进行研究。如图6(a)-6(c)所示,研究确定了无注意力乘数、无自连接的多跳图卷积(MulStepNET-a)基于Pubmed、Cora和Citeseer数据集上最大阶数分别为21、5和4,无注意力乘数、有自连接的多跳图卷积(MulStepNET-b)基于Pubmed、Cora和Citeseer数据集上最大阶数分别为21、8和4,且超过对应的最大阶数后分类精度会有所下降;另外研究还表明中引入自连接的多跳图卷积分类精度有所提升,进而确定了通过引入新的自连接赋予自身节点更高权重是更贴合实际、更有意义的。
此外,本申请还分别基于三组数据对应的上述最大阶数模型做了保持其他多跳图卷积的条件不变的情况下是否引入自连接,及保持其他多跳图卷积的条件不变的情况下是否引入注意力乘数的两组对比实验,如表3-4所示,研究结果都表明同时引入自连接和注意力乘数的多跳图卷积模型分类效果最好。
表3无自连接MulStepNET与MulStepNET基于半监督数据集的分类精度比对
表3说明:表中的准确率以百分比表示,且该数字是10次运行的平均值,基于Pubmed、Cora和Citeseer数据集上,有自连接的多跳图卷积神经网络模型的分类精度比无自连接多跳图卷积神经网络模型的分类精度分别提升1.4%、1.3%和1.9%。
表4无注意力乘数MulStepNET与MulStepNET基于半监督数据集的分类精度比对
表4说明:表中的准确率以百分比表示,且该数字是10次运行的平均值。
在本申请上述实施例的实际应用中,根据实际需求,可以对模型的注意力乘数和自连接进行合理的取舍,若只想引入注意力乘数调整不同组邻域节点分类贡献度的方法来聚集多阶邻域信息和区分不同阶邻域的分类贡献度,而不需要引入自连接进一步增加自身节点的权重时,可以将本申请上述各实施例中引入自连接的部分去掉,只保留其他部分的技术方案及可以实现,此处不再赘述。
需要说明的是,虽然上述流程图中的各个步骤按照箭头的指示依次显示,但是这些步骤并不是必然按照箭头指示的顺序依次执行。除非本文中有明确的说明,这些步骤的执行并没有严格的顺序限制,这些步骤可以以其它的顺序执行。而且,上述流程图中的至少一部分步骤可以包括多个子步骤或者多个阶段,这些子步骤或者阶段并不必然是在同一时刻执行完成,而是可以在不同的时刻执行,这些子步骤或者阶段的执行顺序也不必然是依次进行,而是可以与其它步骤或者其它步骤的子步骤或者阶段的至少一部分轮流或者交替地执行。
图7示出一个实施例中计算机设备的内部结构图,该计算机设备具体可以是终端或服务器。如图7所示,该计算机设备包括通过系统总线连接的处理器、存储器、网络接口、显示器和输入装置。其中,该计算机设备的处理器用于提供计算和控制能力。该计算机设备的存储器包括非易失性存储介质、内存储器。该非易失性存储介质存储有操作系统和计算机程序。该内存储器为非易失性存储介质中的操作系统和计算机程序的运行提供环境。该计算机设备的网络接口用于与外部的终端通过网络连接通信。该计算机程序被处理器执行时以实现一种基于分组注意的宽度图卷积网络模型的训练方法。该计算机设备的显示屏可以是液晶显示屏或者电子墨水显示屏,该计算机设备的输入装置可以是显示屏上覆盖的触摸层,也可以是计算机设备外壳上设置的按键、轨迹球或触控板,还可以是外接的键盘、触控板或鼠标等。
本领域普通技术人员可以理解,图7中示出的结构,仅仅是与本申请方案相关的部分结构的框图,并不构成对本申请方案所应用于其上的计算机设备的限定,具体的计算设备可以包括比图中所示更多或更少的部件,或者组合某些部件,或者具有同的部件布置。
在一个实施例中,提供了一种计算机设备,包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序,处理器执行计算机程序时实现上述基于分组注意的宽度图卷积神经网络模型的训练方法的步骤。
在一个实施例中,提供了一种计算机可读存储介质,其上存储有计算机程序,计算机程序被处理器执行时实现上述基于分组注意的宽度图卷积神经网络模型的训练方法的步骤。
综上,本发明实施例提供的一种基于分组注意的宽度图卷积神经网络模型及其训练方法,其基于对现有高阶图卷积及高低阶图卷积的适当简化,提出了一种包括能捕捉多跳邻域信息、加大自身节点权重和引入注意力乘数的多跳图卷积层、混合不同跳邻域特征的SP信息融合层及softmax分类输出层的多跳图卷积神经网络模型,及与该模型对应的先进行特征预处理再进行训练的高效模型训练方法。该模型及其训练方法应用于实际分类测试时,采用多跳图卷积层增加模型宽度、降低模型深度、减少参数量和计算复杂度的同时,还能同时聚集多跳邻域信息、赋予自身节点更高权重,且采用注意力乘数调整不同邻域节点的分类贡献度,不仅保持了更大的感受野,避免了模型过拟合风险,还提高了模型的有效性、稳定性和分类表现。
本领域普通技术人员可以理解实现上述实施例方法中的全部或部分流程,是可以通过计算机程序来指令相关的硬件来完成,所述的计算机程序可存储于一非易失性计算机可读取存储介质中,该计算机程序在执行时,可包括如上述各方法的实施例的流程。
本说明书中的各个实施例均采用递进的方式描述,各个实施例直接相同或相似的部分互相参见即可,每个实施例重点说明的都是与其他实施例的不同之处。需要说明的是,上述实施例的各技术特征可以进行任意的组合,为使描述简洁,未对上述实施例中的各个技术特征所有可能的组合都进行描述,然而,只要这些技术特征的组合不存在矛盾,都应当认为是本说明书记载的范围。
以上所述实施例仅表达了本申请的几种优选实施方式,其描述较为具体和详细,但并不能因此而理解为对发明专利范围的限制。应当指出的是,对于本技术领域的普通技术人员来说,在不脱离本发明技术原理的前提下,还可以做出若干改进和替换,这些改进和替换也应视为本申请的保护范围。因此,本申请专利的保护范围应以所述权利要求的保护范围为准。

Claims (7)

1.一种多跳图卷积神经网络模型系统,其特征在于,所述系统包括多跳图卷积神经网络模型;所述多跳图卷积神经网络模型依次包括输入层、多跳图卷积层、信息融合池化层和输出层;
所述输入层,用于接收训练数据集的图特征;
所述多跳图卷积层,用于根据所述图特征进行零阶到k阶的多跳图卷积运算,得到图卷积数据;
所述信息融合池化层,用于根据所述图卷积数据进行多跳邻域的特征融合,得到融合数据;
所述输出层,用于根据所述融合数据输出模型结果;
其中,所述多跳图卷积层通过如下方式产生:
对不同阶数图卷积按照相邻三阶为一组进行分组;
组内图卷积采用注意融合,组间图卷积采用注意力乘数调整权重;
所述多跳图卷积层包括相邻组权重共享的一阶图卷积到k阶图卷积,表示为:
其中,X是图的输入矩阵,是参数矩阵,/>是图的正则化邻接矩阵,/>是图卷积的最大阶数,/>,/>为注意融合函数,/>为对应不同组图卷积的注意力乘数;
所述多跳图卷积神经网络模型的输出层的输出MulStepNET表示为:
其中,为激活函数,/>为信息融合函数,/>为多分类输出函数。
2.如权利要求1所述的多跳图卷积神经网络模型系统,其特征在于,在所述多跳图卷积层的任一阶数图卷积引入新的自连接。
3.如权利要求1所述的多跳图卷积神经网络模型系统,其特征在于,所述注意融合函数的注意融合公式为:
其中,为第i-2阶图卷积和第i阶图卷积的注意融合输出。
4.如权利要求1所述的多跳图卷积神经网络模型系统,其特征在于,所述激活函数为ReLU非线性激活函数。
5.如权利要求4所述的多跳图卷积神经网络模型系统,其特征在于,所述信息融合池化层采用SP求和信息融合池化,其计算公式如下:
6.如权利要求1-5任一项所述的多跳图卷积神经网络模型系统中多跳图卷积神经网络模型的训练方法,其特征在于,所述训练方法的步骤包括:
获取所述训练数据集,并根据所述训练数据集的类型得到所述训练数据集的图特征,所述图特征包括图的输入矩阵和正则化邻接矩阵;
将所有不同阶数的所述图的正则化邻接矩阵采用组内注意融合和组间加权作和的方法,得到多幂邻接矩阵,并将所述多幂邻接矩阵与所述图的输入矩阵作积,得到预处理特征;
将所述预处理特征输入所述多跳图卷积神经网络模型,进行特征训练,得到训练结果。
7.如权利要求6所述的多跳图卷积神经网络模型的训练方法,其特征在于,所述将所述预处理特征输入所述多跳图卷积神经网络模型,进行特征训练,得到训练结果的步骤包括:
将所述多跳图卷积神经网络模型的参数矩阵进行随机初始化,并将所述注意力乘数初始化为特定值;
将所述预处理特征输入所述多跳图卷积神经网络模型,根据学习率优化结合所述训练数据集的属性调整所述注意力乘数,并采用损失函数和梯度下降法进行训练,得到收敛的参数矩阵。
CN202011642703.6A 2020-12-30 2020-12-30 一种多跳图卷积神经网络模型系统及训练方法 Active CN112633481B (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN202011642703.6A CN112633481B (zh) 2020-12-30 2020-12-30 一种多跳图卷积神经网络模型系统及训练方法

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202011642703.6A CN112633481B (zh) 2020-12-30 2020-12-30 一种多跳图卷积神经网络模型系统及训练方法

Publications (2)

Publication Number Publication Date
CN112633481A CN112633481A (zh) 2021-04-09
CN112633481B true CN112633481B (zh) 2023-11-28

Family

ID=75290617

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202011642703.6A Active CN112633481B (zh) 2020-12-30 2020-12-30 一种多跳图卷积神经网络模型系统及训练方法

Country Status (1)

Country Link
CN (1) CN112633481B (zh)

Families Citing this family (4)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN113361334B (zh) * 2021-05-18 2022-07-22 山东师范大学 基于关键点优化和多跳注意图卷积行人重识别方法及系统
CN113343942B (zh) * 2021-07-21 2023-05-23 西安电子科技大学 一种遥感图像缺陷检测方法
CN113535904B (zh) * 2021-07-23 2022-08-09 重庆邮电大学 一种基于图神经网络的方面级情感分析方法
CN116824282B (zh) * 2023-08-31 2023-11-10 中国石油大学(华东) 融合多跳图卷积和注意力的半监督高光谱图像分类方法

Citations (3)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN111370073A (zh) * 2020-02-27 2020-07-03 福州大学 一种基于深度学习的药物互作规则预测方法
CN111540198A (zh) * 2020-04-17 2020-08-14 浙江工业大学 基于有向图卷积神经网络的城市交通态势识别方法
CN111538848A (zh) * 2020-04-29 2020-08-14 华中科技大学 一种融合多源信息的知识表示学习方法

Family Cites Families (1)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
US11544535B2 (en) * 2019-03-08 2023-01-03 Adobe Inc. Graph convolutional networks with motif-based attention

Patent Citations (3)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN111370073A (zh) * 2020-02-27 2020-07-03 福州大学 一种基于深度学习的药物互作规则预测方法
CN111540198A (zh) * 2020-04-17 2020-08-14 浙江工业大学 基于有向图卷积神经网络的城市交通态势识别方法
CN111538848A (zh) * 2020-04-29 2020-08-14 华中科技大学 一种融合多源信息的知识表示学习方法

Non-Patent Citations (1)

* Cited by examiner, † Cited by third party
Title
《Hybrid Low-order and Higher-order Graph Convolutional Networks》;刘勋等;Machine Learning;全文 *

Also Published As

Publication number Publication date
CN112633481A (zh) 2021-04-09

Similar Documents

Publication Publication Date Title
CN112633481B (zh) 一种多跳图卷积神经网络模型系统及训练方法
CN112598080B (zh) 基于注意力的宽度图卷积神经网络模型系统及训练方法
US20230169140A1 (en) Graph convolutional networks with motif-based attention
Abu-El-Haija et al. Mixhop: Higher-order graph convolutional architectures via sparsified neighborhood mixing
CN112633482B (zh) 一种高效宽度图卷积神经网络模型系统及训练方法
Nie et al. The constrained laplacian rank algorithm for graph-based clustering
EP3564866A1 (en) Computation method
CN112529168B (zh) 一种基于gcn的属性多层网络表示学习方法
Qu et al. An end-to-end neighborhood-based interaction model for knowledge-enhanced recommendation
CN113657578A (zh) 高效卷积神经网络
CN112529069B (zh) 一种半监督节点分类方法、系统、计算机设备和存储介质
Zheng Gradient descent algorithms for quantile regression with smooth approximation
CN109685204B (zh) 图像处理方法及装置、存储介质及电子设备
CN110826700B (zh) 建模邻居交互的双线性图神经网络模型的实现及分类方法
WO2022105108A1 (zh) 一种网络数据分类方法、装置、设备及可读存储介质
CN114637923B (zh) 基于层次注意力图神经网络的数据信息推荐方法和装置
CN112668700B (zh) 一种基于分组注意的宽度图卷积网络模型系统及训练方法
JP2019016159A (ja) 畳み込みニューラルネットワーク
Venturelli et al. A Kriging-assisted multiobjective evolutionary algorithm
Liang et al. Survey of graph neural networks and applications
CN108737491B (zh) 信息推送方法和装置以及存储介质、电子装置
Sun et al. Graph force learning
CN112651492B (zh) 一种自连接宽度图卷积神经网络模型系统及训练方法
Abu-El-Haija et al. A higher-order graph convolutional layer
Zheng et al. Multi-class indoor semantic segmentation with deep structured model

Legal Events

Date Code Title Description
PB01 Publication
PB01 Publication
SE01 Entry into force of request for substantive examination
SE01 Entry into force of request for substantive examination
GR01 Patent grant
GR01 Patent grant