Brauer O(2)-equivariant mixing layer
Implementation & benchmark of arXiv:2608.18328 — The Brauer category $\mathcal{B}(2)$ has principal graph $D_\infty$
Source paper: The Brauer category $\mathcal{B}(2)$ has principal graph $D_\infty$ arXiv:2608.18328 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Construct a neural mixing layer only from Brauer generators for the orthogonal group: identity, pairwise swaps, and pairwise contractions with the Euclidean metric. This gives an exactly O(2)-equivariant alternative to unconstrained tensor mixing, with trainable coefficients but fixed symmetry-preserving basis maps.
Formulas
Mathematical statement
The Brauer category has objects n representing tensor powers V^{\otimes n}, where V=\mathbb{R}^2 carries the standard O(2) action. A crossing is the swap S(u\otimes v)=v\otimes u, and a cup/cap is the Euclidean metric contraction C(u,v)=u^{\mathsf T}v and insertion \eta(1)=\sum_{a=1}^{2}e_a\otimes e_a. A closed bubble evaluates to δ=2. The extracted paper uses the relations E\rho(T)=E and \rho(T)^2=\mathrm{id}_2, where T is a crossing, \rho(T) is its representation on two tensor factors, and E is the contraction operator. These identities guarantee that the fixed diagram maps commute with the diagonal O(2) action. For selected Brauer diagrams d from p input vector slots to q output vector slots, \Phi=\sum_d heta_d\rho(d) is therefore equivariant.
Implementation notes
(1) Integrate this at the token- or node-mixing operation of an attention or graph block. Let each item contain vector features x_i∈R^2, optionally alongside scalar invariant channels. Replace a generic learned pairwise map with fixed Brauer basis operations: x_i, x_j, swaps of tensor slots, dot products x_i^T x_j, and metric-inserted vectors such as (x_i^T x_j)x_k. (2) Pseudocode: for every selected diagram d, compute z_d=rho(d)(x) using einsum contractions and permutations; compute y=sum_d theta_d*z_d+b; apply nonlinearities only to invariant scalar channels, or use scalar gates y=sigma(gate(s))*y for vector outputs; aggregate y over neighbors or tokens. (3) The mathematical part is the fixed contraction/permutation basis and the exact relations above. Estimate no group parameters. Verify equivariance by sampling random orthogonal matrices g and measuring err=||Phi(gx)-gPhi(x)||/(||gPhi(x)||+1e-8). (4) First cheap experiment: train a four-layer vector graph network on randomly rotated 2D point clouds, with a distance-prediction or point-cloud classification target. Compare a generic MLP edge function, a standard equivariant tensor-product baseline, and the Brauer layer at matched hidden width and parameter count. Measure accuracy on unseen rotations/reflections, parameter count, and equivariance error. Success is lower out-of-orientation error with comparable or better clean accuracy and exact numerical equivariance; failure is no generalization advantage over the generic model.
Verification
This idea has not been verified yet.
Verification happens in two stages: Stage 1 — a mechanism check on a toy system confirms the claimed mathematical phenomenon reproduces; Stage 2 — a benchmark implements the idea on a real (small) neural network task and compares it against a tuned baseline over 8 paired seeds with a permutation test.
Artifacts
Artifacts unavailable.