图神经网络之池化SAGPool

SAGPool 一个自注意力图池化方法,它是架构在层次池化环境下的。 Self-attention mask 注意力可以让网络关注重要的特征,而轻视不重要的特征。 我们利用图图卷积来进行自注意力打分。 σ \sigma σ为激活函数,如tanh. 下面需要选取top-k个结点 网络结构: (1) 卷积层: (2) READOUT层: (3) 池化层 图池化 层次池化 传送门:https://arx
相关文章
相关标签/搜索