9 天前

图网络中的主邻域聚合

Gabriele Corso, Luca Cavalleri, Dominique Beaini, Pietro Liò, Petar Veličković
图网络中的主邻域聚合
摘要

图神经网络(Graph Neural Networks, GNNs)已被证明在处理图结构数据的各类预测任务中具有优异性能。近期关于其表达能力的研究主要聚焦于同构性判定任务以及可数特征空间。本文将这一理论框架拓展至包含连续特征的情形——这类特征在现实世界的输入数据中频繁出现,同时也广泛存在于GNN的隐藏层中,并在此背景下揭示了在多聚合函数需求上的必要性。基于此,我们提出了一种新型架构——主邻域聚合(Principal Neighbourhood Aggregation, PNA),该架构结合了多种聚合函数与度数缩放机制(degree-scalers),后者是对求和聚合器的广义化。最后,我们通过一个新颖的基准测试体系,系统评估了不同模型在捕捉与利用图结构方面的性能,该基准涵盖来自经典图论的多个任务,以及来自真实世界领域的现有基准。实验结果一致表明,所提出的PNA模型展现出显著优势。本研究旨在推动图神经网络领域的研究方向,朝向新型聚合方法的探索,我们认为这些方法对于构建强大且鲁棒的图神经网络模型至关重要。