Feb, 2024

增强图池化与持久同调

TL;DR最近,有一个新兴的趋势是将持久同调(PH)整合到图神经网络(GNNs)中,以丰富其表达能力。然而,将 PH 特征简单地插入 GNN 层中总是导致边际改善且解释性不高。在本文中,我们研究了一种新颖的机制,利用 PH 将全局拓扑不变性注入到池化层中,这受到了 PH 中过滤操作自然地对齐图池化的观察的启发。这种方式下,粗化图中的消息传递沿着持久池化拓扑进行,从而提高了性能。在实验中,我们将这一机制应用于一系列图池化方法,并观察到在几个流行的数据集上一致且显著的性能提升,证明了其广泛适用性和灵活性。