Change the Product, Keep the Parameters: Associative Algebra Layers for Transformers
Not on the map. Why?
Abstract (arXiv)
Fast matrix multiplication algorithms keep the product fixed and search for a cheaper way to evaluate it. We instead ask whether a Transformer's learned projections can use a different, cheaper product altogether. Building on an associative-algebra construction that replaces ordinary matrix multiplication with a sparser interaction table over the same weight blocks, we construct a family with quadratic arithmetic in the matrix dimension when the physical block size remains fixed, and derive finite-shape constraints for GPU execution. The construction is provably optimal for its bilinear rank by the Alder--Strassen bound and can be realized as row-typed rectangular projections compatible with causal masking and KV-cached decoding. We provide an empirical test of this approach by training two approximately 110M-parameter decoder-only Transformer LMs from the same recipe and 12.3B-token budget, differing only in their feed-forward layer: one uses ordinary dense matrix multiplication and the other uses the associative-algebra product. Across four prompt domains, the algebraic model achieves a 6.2--7.8% increase in end-to-end generation throughput, while obtaining lower scores on all three reported downstream metrics. We treat these results as a feasibility and trainability check for the proposed approach at small scale, leaving further investigation to future work.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 15 of its 24 references that are in the map. First posted on arXiv 26 Sep 2026.
Builds on
15 of 24 references.
References · 15
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2025Qwen3 Technical Reportcited by 8,820
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
- 2021Program Synthesis with Large Language Modelscited by 3,637
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
- 2023Instruction-Following Evaluation for Large Language Modelscited by 1,228
- 2015Tensorizing Neural Networkscited by 250
- 2022Monarch: Expressive Structured Matrices for Efficient and Accurate Trainingcited by 106
- 2021Beyond Fully-Connected Layers with Quaternions: Parameterization of Hypercomplex Multiplications with 1/n Parameterscited by 66
- 2024Compute Better Spent: Replacing Dense Layers with Structured Matricescited by 28
- 2024Scalable MatMul-free Language Modelingcited by 25
- 2020Algebraic Neural Networks: Stability to Deformationscited by 22
3 more
- 2020AlgebraNetscited by 7
- 2026Improving the matrix multiplication exponent with modern optimization and AlphaEvolvecited by 2
- 2024Fully tensorial approach to hypercomplex-valued neural networkscited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Universal Transformers for Circuit Computations: Perfect Length Generalization in Tiny Transformers
- 2022LLM.int8(): 8-bit Matrix Multiplication for Transformers at Scale
- 2024GPT-2 Through the Lens of Vector Symbolic Architectures
- 2026Shift-Accumulate Attention: Multiplier-Free Query--Key Products for Transformer Decoding ↗
- 2025EcoSpa: Efficient Transformer Training with Coupled Sparsity
- 2026Finite-Time Analysis of Gradient Descent for Shallow Transformers
- 2022HEAT: Hardware-Efficient Automatic Tensor Decomposition for Transformer Compression
- 2026BAPS: A Fine-Grained Low-Precision Scheme for Softmax in Attention via Block-Aware Precision reScaling