mlsanityData to 7 Oct 2026

Structured Multidimensional Representation Learning for Large Language Models

arXiv 2603.05727 ↗First posted 5 Mar 2026cs.CL cs.NA math.NAcited by 0

Not on the map. Why?

Abstract (arXiv)

Transformer architectures achieve state-of-the-art performance across a wide range of pattern recognition and natural language processing tasks, but their scaling is accompanied by substantial parameter growth and redundancy in the embedding dimension. In this work, we introduce a structured spectral factorization of the embedding space based on the L-product for third-order tensors. By reshaping token representations into spectral tensor slices and performing attention and feed-forward operations in the transform domain, we obtain a Tensor Transformer architecture that decomposes the encoder into p independent spectral sub-transformers while preserving standard Transformer semantics. We prove that the proposed L-Transformer is spectrally equivalent to p parallel Transformers operating on reduceddimensional embeddings, which yields approximately 1/p reduction (up to lower-order terms such as biases and normalization parameters) in encoder parameters under fixed total embedding size. When instantiated with a real-valued Discrete Cosine Transform (DCT), the method remains fully differentiable and compatible with existing training pipelines. Beyond compression, the spectral decomposition introduces an inductive bias over embedding frequencies, enabling slice-dependent frequency scaling that improves generalization. Experiments on IMDB and AG~News show that the proposed model can substantially reduce encoder parameters (up to 75% for p=4) while maintaining competitive accuracy. On IMDB, the tensorized encoder matches or improves upon the standard baseline under compression, whereas on AG~News at moderate width we observe a small accuracy decrease in exchange for a 4 times encoder reduction; at BERT-base width (d=768), performance returns to parity.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 22 of its 33 references that are in the map. First posted on arXiv 5 Mar 2026.

Builds on

22 of 33 references.

References · 22

  1. 2017Attention Is All You Needcited by 59,932
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  3. 2020Language Models are Few-Shot Learnerscited by 32,199
  4. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
  5. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
  6. 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
  7. 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,243
  8. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
  9. 2015Neural Machine Translation of Rare Words with Subword Unitscited by 2,898
  10. 2019ALBERT: A Lite BERT for Self-supervised Learning of Language Representationscited by 2,772
  11. 2015Character-level Convolutional Networks for Text Classificationcited by 2,293
  12. 2020Reformer: The Efficient Transformercited by 1,658
10 more
  1. 2020Rethinking Attention with Performerscited by 1,584
  2. 2020Linformer: Self-Attention with Linear Complexitycited by 1,462
  3. 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,153
  4. 2017Mixed Precision Trainingcited by 1,135
  5. 2019Are Sixteen Heads Really Better than One?cited by 803
  6. 2015Cyclical Learning Rates for Training Neural Networkscited by 627
  7. 2021FNet: Mixing Tokens with Fourier Transformscited by 343
  8. 2015Tensorizing Neural Networkscited by 246
  9. 2023The Truth is in There: Improving Reasoning in Language Models with Layer-Selective Rank Reductioncited by 120
  10. 2024TRAWL: Tensor Reduced and Approximated Weights for Large Language Modelscited by 5

Built on by

Not cited yet.