Oct, 2023

GRAPES: 学习对于可扩展图神经网络进行图样本采样

TL;DR图神经网络通过以各种方式聚合邻域信息来学习图中节点的表示,然而随着网络深度的增加,其感受野会呈指数增长,导致内存开销过大。为了解决图神经网络的内存问题,图采样通过在图中对节点进行采样,从而使得图神经网络能够适应更大规模的图。本文引入了 GRAPES 这一自适应图采样方法,通过学习识别用于训练图神经网络分类器的有影响力的节点集合,并利用 GFlowNet 学习节点采样概率。我们在多个小规模和大规模图数据集上评估了 GRAPES 方法,并展示了其在准确性和可扩展性方面的有效性。与现有的采样方法相比,GRAPES 方法即使在样本规模较小的情况下仍能保持高准确性,因此可以适用于非常大的图。我们的代码公开在此 https URL。