Graph neural network

Graph neural networks (GNNs) are designed to handle graph-structured data, which lacks a canonical node ordering. GNNs achieve permutation equivariance by updating node representations through pairwise message passing, ensuring that reordering nodes results in the same node representations. This property is crucial for tasks where the order of nodes does not matter, such as predicting molecular efficacy.

For graph-level prediction tasks, GNNs use a permutation-invariant readout function. This function aggregates node representations into a single vector that remains unchanged regardless of node ordering. This approach simplifies the process of making predictions based on the entire graph, rather than individual nodes.

An example of this is molecular drug design, where molecules are represented as graphs with nodes for atoms and edges for bonds. The graph-level task involves predicting the efficacy of a molecule for a specific application, such as eliminating E. coli bacteria. By reducing the graph to a single vector, GNNs can effectively handle varying input sizes and provide a unified representation for prediction tasks.