Equivariance, proved and broken
proof▲▲△Prove that the sum layer , with applied entrywise, is permutation-equivariant, and that the readout is invariant. Then show that the layer “take the feature of your lowest-numbered neighbour” is not equivariant, using the paw graph with and the relabelling that swaps nodes and .
Hint
, an entrywise function commutes with reordering rows, and .
Solution
Equivariance. Relabel: and . The pre-activation becomes
An entrywise applied to a matrix with reordered rows gives the same rows reordered, so the output is .
Invariance. A permutation matrix’s columns each contain one , so and .
A layer that breaks it. On the paw the lowest-numbered neighbours of nodes to are , , and , so the layer returns . After swapping nodes and , the features are and the lowest-numbered neighbours are , , and : the layer returns . Equivariance would require the original output with its first and last entries swapped, . The second and third entries differ: node ‘s output changed although nothing about node did. The layer reads the labelling, which carries no information about the graph.