计算机研究与发展2024,Vol.61Issue(4) :903-915.DOI:10.7544/issn1000-1239.202221004

基于图核同构网络的图分类方法

Graph Classification Method Based on Graph Kernel Isomorphism Network

徐立祥 葛伟 陈恩红 罗斌
计算机研究与发展2024,Vol.61Issue(4) :903-915.DOI:10.7544/issn1000-1239.202221004

基于图核同构网络的图分类方法

Graph Classification Method Based on Graph Kernel Isomorphism Network

徐立祥 1葛伟 1陈恩红 2罗斌3
扫码查看

作者信息

  • 1. 合肥学院人工智能与大数据学院 合肥 230601
  • 2. 大数据分析与应用安徽省重点实验室(中国科学技术大学计算机科学与技术学院) 合肥 230027
  • 3. 安徽大学计算机科学与技术学院 合肥 230601
  • 折叠

摘要

图表示学习已成为图深度学习领域的一个研究热点.大多数图神经网络存在过平滑现象,这类方法重点关注图节点特征,对图的结构特征关注度不高.为了提升对图结构特征的表征能力,提出了一种基于图核同构网络的图分类方法,即KerGIN.该方法首先通过图同构网络(graph isomorphism network,GIN)对图进行节点特征编码,并使用图核方法对图进行结构编码,进一步利用Nyström方法降低图核矩阵的维度.其次借助MLP将图核矩阵与图特征矩阵对齐,通过注意力机制将图的特征编码和结构编码进行自适应加权融合,进而得到图的最终特征表示,提升了图结构特征信息的表达能力.最后在 7个公开的图分类数据集上对模型进行了实验评估:与现有图表示模型相比,KerGIN模型能够在图分类准确度上有较大幅度提升,它可以增强GIN对图结构特征信息的表达能力.

Abstract

Graph representation learning has become a research hotspot in the field of graph deep learning.Most graph neural networks suffer from oversmoothing,and these methods focus on graph node features and pay little attention to the structural features of graphs.In order to improve the representation of graph structural features,we propose a graph classification method based on graph kernel homomorphic network,namely KerGIN.The method first encodes the node features of the graph through graph isomorphism network(GIN),and then uses the graph kernel method to encode the graph structure.The Nyström method is further used to reduce the dimension of the graph kernel matrix.The graph kernel matrix is aligned with the graph feature matrix with the help of MLP,and the feature encoding and structure encoding of the graph are adaptively weighted and fused through the attention mechanism to obtain the final feature representation of the graph,which enhances the ability to express the structural feature information of the graph.Finally,the model is experimentally evaluated on seven publicly available graph classification datasets:compared with the existing graph representation models,KerGIN model is able to improve the graph classification accuracy substantially,and it can enhance the ability of GIN to represent the graph structural feature information.

关键词

图分类/图神经网络/图核/Nyström方法/图注意力机制

Key words

graph classification/graph neural network/graph kernel/Nyström method/graph attention mechanism

引用本文复制引用

基金项目

国家自然科学基金(62176085)

国家自然科学基金国际合作重点项目(61860206004)

出版年

2024
计算机研究与发展
中国科学院计算技术研究所 中国计算机学会

计算机研究与发展

CSTPCDCSCD北大核心
影响因子:2.649
ISSN:1000-1239
被引量1
参考文献量31
段落导航相关论文