首页|基于归纳学习图卷积和自注意力池化的图分类网络

基于归纳学习图卷积和自注意力池化的图分类网络

扫码查看
针对图神经网络在大规模图上的分类表现不佳,无法快速形成未知节点和边的嵌入,并且容易丢失图重要特征等问题.提出一种基于归纳学习和自注意力池化相结合的图分类网络模型,一方面采用改进聚合函数后的归纳式学习方法对图的节点特征形成快速地嵌入,另一方面采用 自注意力池化方法保留图的重要特征,最终采用适于提取大规模图信息的层次化结构框架进行下游图分类任务.实验结果表明,该网络模型在相同的公共数据集下,对比其他图分类模型有2%~10%左右精度的提高.
GRAPH CLASSIFICATION NETWORK BASED ON INDUCTIVE LEARNING GRAPH CONVOLUTION AND SELF-ATTENTION POOLING
In view of the poor classification performance of graph neural network on large-scale graphs,it is unable to quickly form the embedding of unknown nodes and edges,and it is easy to lose the important features of the graph.A graph classification network model based on inductive learning and self-attention pooling is proposed.On the one hand,the inductive learning method after the improved aggregation function was used to quickly embed the node features of the graph.On the other hand,the self-attention pooling method was used to retain the important features of the graph.A hierarchical framework suitable for extracting large-scale graph information was used for downstream graph classification task.The experimental results show that the accuracy of this method is about 2%~10%higher than that of other graph classification models under the same common data set.

Graph neural networkGraph classificationSelf-attention poolingGraph convolutional network

倪瑞智、王永平、张晓琳、叶金辉、陶雪晴

展开 >

内蒙古科技大学信息工程学院 内蒙古包头 014010

图神经网络 图分类 自注意力池化 图卷积神经网络

国家自然科学基金项目内蒙古自治区自然科学基金项目

615620652019MS06001

2024

计算机应用与软件
上海市计算技术研究所 上海计算机软件技术开发中心

计算机应用与软件

CSTPCD北大核心
影响因子:0.615
ISSN:1000-386X
年,卷(期):2024.41(10)