论文标题
通过子图同构计数提高图神经网络的表达
Improving Graph Neural Network Expressivity via Subgraph Isomorphism Counting
论文作者
论文摘要
尽管图神经网络(GNN)在各种应用中取得了显着的结果,但最近的研究揭示了其捕获基础图结构的能力的重要缺点。已经表明,标准GNN的表达能力受Weisfeiler-Leman(WL)图同构测试的界定,它们从中继承了诸如无法检测和计算图形子结构的诸如无法检测到的限制。另一方面,有大量的经验证据,例如在网络科学和生物信息学中,该子结构通常与下游任务密切相关。为此,我们提出了基于基于子结构编码的拓扑意识的消息传递方案“ Graph Subscure Networks”(GSN)。我们从理论上分析了我们体系结构的表现力,表明它比WL检验更具表现力,并为普遍性提供了足够的条件。重要的是,我们不会试图遵守WL层次结构。这使我们能够保留标准GNN的多个有吸引力的属性,例如局部性和线性网络复杂性,同时能够消除同构象构象的歧义。我们对图形分类和回归任务进行了广泛的实验评估,并在包括分子图和社交网络在内的各种现实世界中获得最先进的结果。该代码可在https://github.com/gbouritsas/graph-substructure-networks上公开获取。
While Graph Neural Networks (GNNs) have achieved remarkable results in a variety of applications, recent studies exposed important shortcomings in their ability to capture the structure of the underlying graph. It has been shown that the expressive power of standard GNNs is bounded by the Weisfeiler-Leman (WL) graph isomorphism test, from which they inherit proven limitations such as the inability to detect and count graph substructures. On the other hand, there is significant empirical evidence, e.g. in network science and bioinformatics, that substructures are often intimately related to downstream tasks. To this end, we propose "Graph Substructure Networks" (GSN), a topologically-aware message passing scheme based on substructure encoding. We theoretically analyse the expressive power of our architecture, showing that it is strictly more expressive than the WL test, and provide sufficient conditions for universality. Importantly, we do not attempt to adhere to the WL hierarchy; this allows us to retain multiple attractive properties of standard GNNs such as locality and linear network complexity, while being able to disambiguate even hard instances of graph isomorphism. We perform an extensive experimental evaluation on graph classification and regression tasks and obtain state-of-the-art results in diverse real-world settings including molecular graphs and social networks. The code is publicly available at https://github.com/gbouritsas/graph-substructure-networks.