基于重要性池化的层级图表示学习方法

张红梅; 李浩然; 张向利

doi:10.16725/j.cnki.cn45-1351/tn.2020.04.007

桂林电子科技大学学报 ›› 2020, Vol. 40 ›› Issue (04) : 300 -304. DOI: 10.16725/j.cnki.cn45-1351/tn.2020.04.007

基于重要性池化的层级图表示学习方法

张红梅, 李浩然, 张向利

作者信息 +

Author information +

文章历史 +

摘要

为解决传统图神经网络在图分类任务上存在训练过程中噪声信息过多以及不能完整地挖掘图的层次表征信息等问题,提出一种端到端的基于重要性池化的层级图表示学习方法。该方法以层内-层间联合特征提取结构为基础,主要包括层内特征提取模块和层间特征提取模块2个部分。利用池化方法将图粗粒化为高级子图结构,以缩减特征图的尺寸;利用循环单元,以抑制层间噪声的传播,并自适应地聚集层级表征。实验结果表明,在合理的时间复杂度下,该方法能使损失函数收敛于更小的值,且模型精度有明显的提升。