Beyond GNNs: An Efficient Architecture for Graph Problems

Pranjal Awasthi, Abhimanyu Das, Sreenivas Gollapudi

[AAAI-22] Main Track
Abstract: Despite their popularity for graph structured data, existing {\em Graph Neural Networks}~(GNNs) have inherent limitations for fundamental graph problems such as shortest paths, $k$-connectivity, minimum spanning tree and minimum cuts. In these instances, it is known that one needs GNNs of high depth, scaling at a polynomial rate with the number of nodes $n$, to provably encode the solution space, in turn affecting their statistical efficiency.



In this work we propose a new hybrid architecture to overcome this limitation. Our proposed architecture that we call as {\GNNplus} networks involve a combination of multiple parallel low depth GNNs along with simple pooling layers involving low depth fully connected networks. We provably demonstrate that for many graph problems, the solution space can be encoded by {\GNNplus} networks using depth that scales only {\em poly-logarithmically} in the number of nodes. This also has statistical advantages that we demonstrate via generalization bounds for {\GNNplus} networks.

Introduction Video

Sessions where this paper appears

  • Poster Session 3

    Fri, February 25 8:45 AM - 10:30 AM (+00:00)
    Blue 2
    Add to Calendar

  • Poster Session 8

    Sun, February 27 12:45 AM - 2:30 AM (+00:00)
    Blue 2
    Add to Calendar