DropGNN: Random Dropouts Increase the Expressiveness of Graph Neural NetworksDownload PDF

21 May 2021, 20:50 (modified: 21 Jan 2022, 20:38)NeurIPS 2021 SpotlightReaders: Everyone
Keywords: graph neural networks, GNN, expressivity, WL, molecular graphs
TL;DR: We devise a new GNN variant (DropGNN) with larger expressive power in both theory and practice
Abstract: This paper studies Dropout Graph Neural Networks (DropGNNs), a new approach that aims to overcome the limitations of standard GNN frameworks. In DropGNNs, we execute multiple runs of a GNN on the input graph, with some of the nodes randomly and independently dropped in each of these runs. Then, we combine the results of these runs to obtain the final result. We prove that DropGNNs can distinguish various graph neighborhoods that cannot be separated by message passing GNNs. We derive theoretical bounds for the number of runs required to ensure a reliable distribution of dropouts, and we prove several properties regarding the expressive capabilities and limits of DropGNNs. We experimentally validate our theoretical findings on expressiveness. Furthermore, we show that DropGNNs perform competitively on established GNN benchmarks.
Supplementary Material: pdf
Code Of Conduct: I certify that all co-authors of this work have read and commit to adhering to the NeurIPS Statement on Ethics, Fairness, Inclusivity, and Code of Conduct.
13 Replies