The Token Is a Group Element: On Lie-Algebra Attention over Matrix Lie Groups
Not on the map. Why?
Abstract (arXiv)
We place the attention token on the group: a token is an element gi of a matrix Lie group G -- a bare transformation, with no feature payload and no external action ρ(g) carrying it. To our knowledge this is the first attention construction whose tokens are bare matrix Lie group elements: their score is the closed-form algebra norm of the relative pose rather than a learned kernel, and it reaches the affine full-frame groups that every irrep- or surjective-exp-based method must exclude. We call it Lie-Algebra Attention. Once tokens are group elements, the rest follows with none of the usual representation-theoretic machinery. The relative geometry of a pair is canonical, gi-1 gj, so the pairwise invariant wij = log(gi-1 gj) is intrinsic rather than designed; equivariance under the diagonal G-action is tautological, and the cocycle condition holds automatically. The attention score is the negative squared algebra norm, sij = -|log(gi-1 gj)|λ2/τ: the canonical proximity kernel under a block-weighted Frobenius inner product, with no irreducible representations, spherical harmonics, Clebsch-Gordan products, or learned kernel. The construction applies to any matrix Lie group on a chosen logarithm chart containing the relative poses, including the non-compact non-abelian affine groups with scale and shear that no vector-token attention method reaches: neither the irrep tradition nor surjective-exp methods. Three sequence-completion experiments, on SE(2), SO(3), and Aff(2), bear this out: the closed-form score matches a learned MLP kernel on the same invariant and outperforms it on SE(2), using 50 to 80x fewer score parameters, while a vector-token baseline breaks invariance by five to twelve orders of magnitude.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 27 of its 29 references that are in the map. First posted on arXiv 18 Jun 2026.
Builds on
27 of 29 references.
References · 27
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2015Spatial Transformer Networkscited by 1,429
- 2016Group Equivariant Convolutional Networkscited by 1,165
- 2021E(n) Equivariant Graph Neural Networkscited by 965
- 2018Tensor field networks: Rotation- and translation-equivariant neural networks for 3D point cloudscited by 746
- 2020SE(3)-Transformers: 3D Roto-Translation Equivariant Attention Networkscited by 607
- 2022MACE: Higher Order Equivariant Message Passing Neural Networks for Fast and Accurate Force Fieldscited by 474
- 2019General E(2)-Equivariant Steerable CNNscited by 446
- 2022Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphscited by 264
- 2020Generalizing Convolutional Neural Networks for Equivariance to Lie Groups on Arbitrary Continuous Datacited by 248
- 2020LieTransformer: Equivariant self-attention for Lie Groupscited by 100
- 2023GTA: A Geometry-Aware Attention Mechanism for Multi-View Transformerscited by 79
15 more
- 2023Geometric Algebra Transformercited by 48
- 2023Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Spacecited by 41
- 2024ActionFlow: Equivariant, Accurate, and Efficient Policies with Spatially Symmetric Flow Matchingcited by 26
- 2024LieRE: Lie Rotational Positional Encodingscited by 18
- 2023Euclidean, Projective, Conformal: Choosing a Geometric Algebra for Equivariant Transformerscited by 13
- 2023Lie Group Decompositions for Equivariant Neural Networkscited by 12
- 2026EquiformerV3: Scaling Efficient, Expressive, and General SE(3)-Equivariant Graph Attention Transformerscited by 11
- 2025RiemannFormer: A Framework for Attention in Curved Spacescited by 8
- 2025Group Representational Position Encodingcited by 5
- 2025Equivariant Neural Networks for General Linear Symmetries on Lie Algebrascited by 4
- 2026Vanilla Group Equivariant Vision Transformer: Simple and Effectivecited by 3
- 2026Toward Manifest Relationality in Transformers via Symmetry Reductioncited by 2
- 2025Clebsch-Gordan Transformer: Fast and Global Equivariant Attentioncited by 1
- 2026Efficient Equivariant Transformer for Self-Driving Agent Modelingcited by 1
- 2026Exact Symmetry as Algebra: Group-Structured Tensors that Enforce Physical Selection Rulescited by 1
Built on by
Not cited yet.