mlsanityData to 14 Oct 2026

Nexusformer: Nonlinear Attention Expansion for Stable and Inheritable Transformer Scaling

arXiv 2604.19147 ↗First posted 21 Apr 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Scaling Transformers typically necessitates training larger models from scratch, as standard architectures struggle to expand without discarding learned representations. We identify the primary bottleneck in the attention mechanism's linear projections, which strictly confine feature extraction to fixed-dimensional subspaces, limiting both expressivity and incremental capacity. To address this, we introduce Nexusformer, which replaces linear Q/K/V projections with a Nexus-Rank layer, a three-stage nonlinear mapping driven by dual activations in progressively higher dimensional spaces. This design overcomes the linearity constraint and enables lossless structured growth: new capacity can be injected along two axes via zero-initialized blocks that preserve pretrained knowledge. Experiments on language modeling and reasoning benchmarks demonstrate that Nexusformer matches Tokenformer's perplexity using up to 41.5% less training compute during progressive scaling (240M to 440M). Furthermore, our analysis of growth dynamics reveals that zero initialization induces a stable convergence trajectory, allowing us to derive a geometric scaling law that accurately predicts performance across expansion scales.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 35 of its 42 references that are in the map. First posted on arXiv 21 Apr 2026.

Builds on

35 of 42 references.

References · 35

  1. 2024The Llama 3 Herd of Modelscited by 14,715
  2. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
  3. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
  4. 2025Qwen3 Technical Reportcited by 8,820
  5. 2020Scaling Laws for Neural Language Modelscited by 6,586
  6. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
  7. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
  8. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
  9. 2023Mistral 7Bcited by 3,164
  10. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
  11. 2022Training Compute-Optimal Large Language Modelscited by 3,019
  12. 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
23 more
  1. 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,164
  2. 2025Gemma 3 Technical Reportcited by 1,824
  3. 2024Mixtral of Expertscited by 1,823
  4. 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,631
  5. 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,160
  6. 2024Phi-4 Technical Reportcited by 711
  7. 2025Kimi K2: Open Agentic Intelligencecited by 567
  8. 20242 OLMo 2 Furiouscited by 340
  9. 2015Net2Net: Accelerating Learning via Knowledge Transfercited by 250
  10. 2020Synthesizer: Rethinking Self-Attention in Transformer Modelscited by 205
  11. 2021An Attention Free Transformercited by 112
  12. 2025mHC: Manifold-Constrained Hyper-Connectionscited by 88
  13. 2023Learning to Grow Pretrained Models for Efficient Transformer Trainingcited by 83
  14. 2022ELLE: Efficient Lifelong Pre-training for Emerging Datacited by 70
  15. 2020Talking-Heads Attentioncited by 62
  16. 2025Every Step Evolves: Scaling Reinforcement Learning for Trillion-Scale Thinking Modelcited by 55
  17. 2021On the Expressive Power of Self-Attention Matricescited by 47
  18. 2024Stacking Your Transformers: A Closer Look at Model Growth for Efficient LLM Pre-Trainingcited by 43
  19. 2023On the Expressivity Role of LayerNorm in Transformers' Attentioncited by 33
  20. 2023Masked Structural Growth for 2x Faster Language Model Pre-trainingcited by 33
  21. 2023Composable Function-preserving Expansions for Transformer Architecturescited by 13
  22. 2024TokenFormer: Rethinking Transformer Scaling with Tokenized Model Parameterscited by 9
  23. 2025Efficient Construction of Model Family through Progressive Training Using Model Expansioncited by 8

Built on by

Not cited yet.