Subgroups of U(d) Induce Natural RNN and Transformer Architectures
Not on the map. Why?
Abstract (arXiv)
This paper presents a direct framework for sequence models with hidden states on closed subgroups of U(d). We use a minimal axiomatic setup and derive recurrent and transformer templates from a shared skeleton in which subgroup choice acts as a drop-in replacement for state space, tangent projection, and update map. We then specialize to O(d) and evaluate orthogonal-state RNN and transformer models on Tiny Shakespeare and Penn Treebank under parameter-matched settings. We also report a general linear-mixing extension in tangent space, which applies across subgroup choices and improves finite-budget performance in the current O(d) experiments.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 12 of its 14 references that are in the map. First posted on arXiv 20 Feb 2026.
Builds on
12 of 14 references.
References · 12
- 2017Attention Is All You Needcited by 59,932
- 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,085
- 2020SE(3)-Transformers: 3D Roto-Translation Equivariant Attention Networkscited by 602
- 2015Unitary Evolution Recurrent Neural Networkscited by 291
- 2019Cheap Orthogonal Constraints in Neural Networks: A Simple Parametrization of the Orthogonal and Unitary Groupcited by 133
- 2020LieTransformer: Equivariant self-attention for Lie Groupscited by 100
- 2016Full-Capacity Unitary Recurrent Neural Networkscited by 90
- 2017Orthogonal Recurrent Neural Networks with Scaled Cayley Transformcited by 82
- 2025DeltaProduct: Improving State-Tracking in Linear RNNs via Householder Productscited by 69
- 2024RotRNN: Modelling Long Sequences with Rotationscited by 6
- 2023Structure-Preserving Transformers for Sequences of SPD Matricescited by 4
- 2023Riemannian Self-Attention Mechanism for SPD Networkscited by 4
Built on by
Citing papers per year: 2026: 1.
1 paper.