ICML2023

Brauer's Group Equivariant Neural Networks

Edward Pearce-Crump

19 citations

Abstract

We provide a full characterisation of all of the possible group equivariant neural networks whose layers are some tensor power of Rn\mathbb{R}^{n} for three symmetry groups that are missing from the machine learning literature: O(n)O(n), the orthogonal group; SO(n)SO(n), the special orthogonal group; and Sp(n)Sp(n), the symplectic group. In particular, we find a spanning set of matrices for the learnable, linear, equivariant layer functions between such tensor power spaces in the standard basis of Rn\mathbb{R}^{n} when the group is O(n)O(n) or SO(n)SO(n), and in the symplectic basis of Rn\mathbb{R}^{n} when the group is Sp(n)Sp(n).