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