Feb, 2024

使用分布式边布局进行图形学习

TL;DR图神经网络(GNNs)通过基于特定的拓扑布局在邻居节点之间通过边缘传递局部消息来学习图结构化数据。本文首次提出使用具备显式物理能量的玻尔兹曼分布所搭配的 Langevin 动力学全局采样拓扑布局,以提升其在物理世界中的适用性。我们认为这样的采样 / 优化布局能够捕捉广泛的能量分布,并在 WL-test 之上提供额外的表达能力,从而简化下游任务。为此,我们提出分布式边缘布局(DELs)作为各种 GNNs 的补充。DEL 是一种预处理策略,独立于后续的 GNN 变体,因此非常灵活。实验结果表明,DELs 一致且显著地提升了一系列 GNN 基准算法,在多个数据集上取得了最先进的性能。