arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.02269stat.MLcond-mat.dis-nncs.LG

图神经网络并不总是过平滑

Graph Neural Networks Do Not Always Oversmooth

RWTH Aachen University(亚琛工业大学);Department of Physics, RWTH Aachen University(物理系,亚琛工业大学);Institute for Advanced Simulation (IAS-6), Computational and Systems Neuroscience, Jülich Research Centre(先进模拟研究所(IAS-6),计算与系统神经科学,耶鲁研究中心)

Bastian Epping, Alexandre René, Moritz Helias, Michael T. Schaub

AI总结:

本研究利用高斯过程等价性分析图卷积网络的过平滑问题,发现初始权重方差足够大时可避免过平滑,还推广了信息传播深度概念,在相变点附近初始化能得到兼具深度与表达能力的网络。

摘要:

图神经网络(GNNs)已成为在各类应用中处理关系型数据的强大工具。然而,GNNs存在过平滑问题,即所有节点的特征会随层数增加呈指数级收敛到同一向量,这阻碍了深度GNNs的设计。在本研究中,我们利用图卷积网络(GCNs)在隐藏特征数量趋于无穷极限下的高斯过程(GP)等价性,研究GCNs中的过平滑现象。通过推广传统深度神经网络(DNNs)中的方法,我们能够用高斯过程描述深度GCNs输出层的特征分布:正如预期,我们发现文献中典型的参数选择会导致过平滑。但该理论也让我们识别出一种全新的非过平滑相:如果网络的初始权重具有足够大的方差,GCNs就不会发生过平滑,即便在深度很大的情况下节点特征仍能保持信息性。我们通过在有限规模GCNs的输出上训练线性分类器,验证了这一预测的有效性。此外,利用GCN高斯过程的线性化,我们将DNNs中的信息传播深度概念推广到了GCNs中。该传播深度在过平滑相与非过平滑相的转变点处发散。我们测试了所提方法的预测结果,发现其与有限规模GCNs的表现高度吻合。在接近非过平滑相转变点的位置初始化GCNs,我们得到了既具备深度又富有表达能力的网络。

英文摘要:

Graph neural networks (GNNs) have emerged as powerful tools for processing relational data in applications. However, GNNs suffer from the problem of oversmoothing, the property that the features of all nodes exponentially converge to the same vector over layers, prohibiting the design of deep GNNs. In this work we study oversmoothing in graph convolutional networks (GCNs) by using their Gaussian process (GP) equivalence in the limit of infinitely many hidden features. By generalizing methods from conventional deep neural networks (DNNs), we can describe the distribution of features at the output layer of deep GCNs in terms of a GP: as expected, we find that typical parameter choices from the literature lead to oversmoothing. The theory, however, allows us to identify a new, non-oversmoothing phase: if the initial weights of the network have sufficiently large variance, GCNs do not oversmooth, and node features remain informative even at large depth. We demonstrate the validity of this prediction in finite-size GCNs by training a linear classifier on their output. Moreover, using the linearization of the GCN GP, we generalize the concept of propagation depth of information from DNNs to GCNs. This propagation depth diverges at the transition between the oversmoothing and non-oversmoothing phase. We test the predictions of our approach and find good agreement with finite-size GCNs. Initializing GCNs near the transition to the non-oversmoothing phase, we obtain networks which are both deep and expressive.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑