DiffPool层次化池化——图粗化可视化

展示DiffPool如何逐步粗化图结构

S^l = softmax(GNN_l(A^l, H^l)), A^(l+1) = S^T·A·S
学习软分配矩阵S, 将节点分配到超节点
层0: 8节点 → 层1: 4超节点 → 层2: 2超节点
DiffPool端到端可训练, 保留图结构信息