Sparse Growing Transformer: Training-Time Sparse Depth Allocation via Progressive Attention Looping
Not on the map. Why?
Abstract (arXiv)
Existing approaches to increasing the effective depth of Transformers predominantly rely on parameter reuse, extending computation through recursive execution. Under this paradigm, the network structure remains static along the training timeline, and additional computational depth is uniformly assigned to entire blocks at the parameter level. This rigidity across training time and parameter space leads to substantial computational redundancy during training. In contrast, we argue that depth allocation during training should not be a static preset, but rather a progressively growing structural process. Our systematic analysis reveals a deep-to-shallow maturation trajectory across layers, where high-entropy attention heads play a crucial role in semantic integration. Motivated by this observation, we introduce the Sparse Growing Transformer (SGT). SGT is a training-time sparse depth allocation framework that progressively extends recurrence from deeper to shallower layers via targeted attention looping on informative heads. This mechanism induces structural sparsity by selectively increasing depth only for a small subset of parameters as training evolves. Extensive experiments across multiple parameter scales demonstrate that SGT consistently outperforms training-time static block-level looping baselines under comparable settings, while reducing the additional training FLOPs overhead from approximately 16--20% to only 1--3% relative to a standard Transformer backbone.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 56 of its 81 references that are in the map. First posted on arXiv 25 Mar 2026.
Builds on
56 of 81 references.
References · 56
- 2017Attention Is All You Needcited by 60,361
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2021Evaluating Large Language Models Trained on Codecited by 8,320
- 2020Measuring Massive Multitask Language Understandingcited by 7,963
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
- 2023Let's Verify Step by Stepcited by 3,789
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,754
38 more
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
- 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,955
- 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,164
- 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,822
- 2024RULER: What's the Real Context Size of Your Long-Context Language Models?cited by 1,346
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
- 2019BERT Rediscovers the Classical NLP Pipelinecited by 1,059
- 2022In-context Learning and Induction Headscited by 964
- 2017Crowdsourcing Multiple Choice Science Questionscited by 765
- 2023Distilling Step-by-Step! Outperforming Larger Language Models with Less Training Data and Smaller Model Sizescited by 737
- 2019Deep Equilibrium Modelscited by 615
- 2024OLMo: Accelerating the Science of Language Modelscited by 613
- 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 415
- 2024∞Bench: Extending Long Context Evaluation Beyond 100K Tokenscited by 367
- 2025MathArena: Evaluating LLMs on Uncontaminated Math Competitionscited by 297
- 2025Reasoning with Latent Thoughts: On the Power of Looped Transformerscited by 240
- 2022SGPT: GPT Sentence Embeddings for Semantic Searchcited by 204
- 2025Scaling Latent Reasoning via Looped Language Modelscited by 165
- 2023Stabilizing Transformer Training by Preventing Attention Entropy Collapsecited by 161
- 2025Mixture-of-Recursions: Learning Dynamic Recursive Depths for Adaptive Token-Level Computationcited by 133
- 2021JFB: Jacobian-Free Backpropagation for Implicit Networkscited by 86
- 2025Inner Thinking Transformer: Leveraging Dynamic Depth Scaling to Foster Adaptive Internal Thinkingcited by 57
- 2025A Little Depth Goes a Long Way: The Expressive Power of Log-Depth Transformerscited by 53
- 2024MoH: Multi-Head Attention as Mixture-of-Head Attentioncited by 44
- 2023SwitchHead: Accelerating Transformers with Mixture-of-Experts Attentioncited by 34
- 2025Semantic-Aware Logical Reasoning via a Semiotic Frameworkcited by 32
- 2025Think-While-Generating: On-the-Fly Reasoning for Personalized Long-Form Generationcited by 30
- 2024Attention Entropy is a Key Factor: An Analysis of Parallel Context Encoding with Full-attention-based Pre-trained Language Modelscited by 26
- 2026Logical Phase Transitions: Understanding Collapse in LLM Logical Reasoningcited by 25
- 2023Dual Path Modeling for Semantic Matching by Perceiving Subtle Conflictscited by 21
- 2024What Can Transformer Learn with Varying Depth? Case Studies on Sequence Learning Taskscited by 19
- 2024Pooling And Attention: What Are Effective Designs For LLM-Based Embedding Models?cited by 19
- 2024Token Prepending: A Training-Free Approach for Eliciting Better Sentence Embeddings from LLMscited by 18
- 2025Semantic Energy: Detecting LLM Hallucination Beyond Entropycited by 18
- 2025Semantic-Space Exploration and Exploitation in RLVR for LLM Reasoningcited by 17
- 2024Distributional Associations vs In-Context Reasoning: A Study of Feed-forward and Attention Layerscited by 12
- 2024Loop Neural Networks for Parameter Sharingcited by 11
- 2025COOPER: A Unified Model for Cooperative Perception and Reasoning in Spatial Intelligencecited by 9
50 most cited of 56 All 56 (JSON)
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026Allocating Recurrent Compute in Looped Language Models18 Aug 2026
- 2026Why Supervised Fine-Tuning Fails to Learn: A Systematic Study of Incomplete Learning in Large Language Models11 Apr 2026
- 2026Reason Only When Needed: Efficient Generative Reward Modeling via Model-Internal Uncertainty11 Apr 2026