Geometric Deep Learning II: Every Linear Permutation-Equivariant Map on ℝⁿ Is a Linear Combination of the Identity and the Averaging Map (1/n)𝟙𝟙ᵀTextbook
Motivation
Many learning problems take an unordered collection as input: the points of a point cloud, the particles of a physical system, the atoms of a molecule, the nodes of a graph. Storing such data in memory fixes an order of the elements, but the order carries no information, and a model whose output changes when the rows of its input are shuffled is answering a different question from the one posed. Bronstein, Bruna, Cohen and Veličković, Geometric Deep Learning: Grids, Groups, Graphs, Geodesics, and Gauges (arXiv:2104.13478v2, 2021), §4.1, make this the first instance of their blueprint: the domain is a set (or a graph) of nodes, and the symmetry group is the symmetric group of all reorderings.
Two architectures used in practice follow this pattern. Deep Sets (Zaheer et al., NeurIPS 2017) and PointNets (Qi et al., CVPR 2017) process sets; message-passing graph neural networks process graphs. The design question in both cases is which layers respect the symmetry. For linear layers the question has a complete answer, and the answer determines the layer used by Deep Sets. For graphs, Maron, Ben-Hamu, Shamir and Lipman (2018) counted the linear layers acting on adjacency matrices.
Setting
A set of nodes carries node features , stacked as the rows of a matrix . A graph adds an adjacency matrix , eq. (8), whose nonzero entries are the edges.
A permutation acts through its permutation matrix by reordering rows: row of is row of . On adjacency matrices it acts by , reordering rows and columns together.
A function is permutation invariant if for every permutation matrix , and a matrix-valued function is permutation equivariant if . On graphs the book's definitions (9) and (10) are
for any permutation matrix .
The section's constructions are the sum readout (6); the shared node-wise linear map (7); the neighbourhood (11) and the multiset of neighbourhood features (12); and the local layer whose -th row is (13). For one feature channel the two linear generators are the identity and the average , every entry of which is the mean .
The book numbers no theorems, lemmas or definitions; every item of this mission is cited by section, equation number and printed page.
Formalization targets
Goal: linear equivariant maps on one channel
For and a linear map ,
This is the book's sentence on p. 33: any linear equivariant "can be written as a linear combination of two generators, the identity … and the average". Both directions are stated; the content is the "only if" direction.
Milestones
- The sum readout (6) is permutation invariant (p. 32).
- The node-wise linear map (7) is permutation equivariant, with rows (p. 33).
- The local layer (13) satisfies (10) whenever is invariant to the order of the neighbourhood (p. 34).
- Invariant and equivariant functions give identical outputs on isomorphic graphs (p. 35).
- On the complete graph , the aggregate is the same at every node (§5.4, p. 81).
Stronger statements
- Several channels (corrected). A linear is permutation equivariant iff .
- Adjacency matrices (Maron et al., corrected). For , the linear maps with form a space of dimension .
Significance
The result. The goal classifies the linear layers of a set network. Any equivariant linear layer on scalar node features mixes each node's own value with the mean of all values, and nothing else; with several channels the two scalars become two weight matrices, which is exactly the Deep Sets layer. Read the other way, the theorem says that the commutant of the permutation representation of on is two-dimensional (for ), the algebraic fact behind the decomposition into two irreducible pieces. The Maron et al. count does the same for layers acting on adjacency matrices and fixes the size of the parameter space of invariant graph networks. The milestones give the elementary invariance and equivariance facts on which the graph-network chapter rests, including the constant aggregate on a complete graph.
Formalizing it. All results here are classical and proved on paper. None is formalized on Prove2Me, and Mathlib has no statement about the commutant of the permutation representation or about equivariant layers on sets. The mission produces a vocabulary of permutation actions on feature and adjacency matrices, and the invariance and equivariance definitions (9)–(10), that later missions of this series (the WL comparison, the E(3)-equivariant layer) also use. Two of the book's printed claims need correction to be true, and the formal statements record the corrected versions.
Difficulty
The milestones are short reindexing arguments; the work is in choosing encodings that keep them faithful. The goal's "if" direction is a one-line computation. The "only if" direction is where an argument is needed: equivariance under all of must force the matrix of to have one value on the diagonal and one value off it. Equivariance under a smaller group does not suffice: commuting with the cyclic shift alone gives all circulant matrices, an -dimensional space, so the argument must use permutations beyond the shifts. The channel version adds a tensor-product layer to the same argument. The Maron et al. count is a dimension statement, so it needs both a spanning family and its independence, and the count is sensitive to : for the dimension is strictly smaller than 15, so any argument must use somewhere.
Formalization scope
Node features are Matrix (Fin n) (Fin d) ℝ (rows are nodes); a single channel is Fin n → ℝ. A permutation is σ : Equiv.Perm (Fin n), acting by X.submatrix σ.symm id (row of is row ) and on adjacency matrices by A.submatrix σ.symm σ.symm; this one convention is used in every statement. Adjacency matrices are real; the edge set is the set of nonzero entries, which for the book's 0/1 matrix (8) is its edge set and also admits weighted graphs. The function of (13) and the aggregator of §5.4 take a Multiset, so their invariance to the order of the neighbours holds by construction. The average is written with , and the goal and channel version assume . Equivariance quantifies over every permutation of , never a generating subset. The complete-graph item states the identical-aggregate clause of §5.4. The page's later claim of equal expressive power across inputs is false: with , , , and , changing changes while stays fixed, which no node-only map can reproduce.
A trivializing formalization is ruled out: "equivariant" is the commutation property itself, never defined as "of the form ".
Two printed claims are corrected. On p. 33 the two-generator form is stated for matrices with channels; it is false for (the map of (7) is a counterexample), so the goal is single-channel and the multi-channel statement uses weight matrices. On p. 34 the fifteen generators are said to be independent of ; the dimension is 1, 8 and 14 for , so that item assumes , and it is stated for adjacency-to-adjacency maps as the side-note on p. 33 specifies.
The development needs Mathlib's Equiv.Perm, Matrix.submatrix, Finset sums and reindexing (Equiv.sum_comp), Multiset.map, and Module.finrank. Proofs of the commutant statements, of the multi-channel version and of the dimension count are welcome; the permutation-action and invariance lemmas proved along the way are reusable by any later development on graph networks.
Selected references
- M. M. Bronstein, J. Bruna, T. Cohen, P. Veličković, Geometric Deep Learning: Grids, Groups, Graphs, Geodesics, and Gauges, arXiv preprint, 2021. https://arxiv.org/abs/2104.13478v2
- M. Zaheer, S. Kottur, S. Ravanbakhsh, B. Póczos, R. Salakhutdinov, A. Smola, Deep Sets, NeurIPS, 2017. https://arxiv.org/abs/1703.06114
- H. Maron, H. Ben-Hamu, N. Shamir, Y. Lipman, Invariant and Equivariant Graph Networks, ICLR, 2019. https://arxiv.org/abs/1812.09902
- C. R. Qi, H. Su, K. Mo, L. J. Guibas, PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation, CVPR, 2017. https://arxiv.org/abs/1612.00593