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

张红梅, 李浩然, 张向利

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

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

  • 张红梅, 李浩然, 张向利
作者信息 +
History +

摘要

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

关键词

图神经网络 / 重要性池化 / 层级表示学习

中图分类号

TP18

引用本文

导出引用
张红梅, 李浩然, 张向利. 基于重要性池化的层级图表示学习方法. 桂林电子科技大学学报. 2020, 40(04): 300-304 https://doi.org/10.16725/j.cnki.cn45-1351/tn.2020.04.007

基金

国家自然科学基金(61461010); 认知无线电与信息处理省部共建教育部重点实验室基金(CRKL170103,CRKL170104); 广西高校云计算与复杂系统重点实验室基金(YF16203)

评论

Accesses

Citation

Detail

段落导航
相关文章

/