Stacking as Accelerated Gradient Descent
Not on the map. Why?
Abstract (arXiv)
Stacking, a heuristic technique for training deep residual networks by progressively increasing the number of layers and initializing new layers by copying parameters from older layers, has proven quite successful in improving the efficiency of training deep neural networks. In this paper, we propose a theoretical explanation for the efficacy of stacking: viz., stacking implements a form of Nesterov's accelerated gradient descent. The theory also covers simpler models such as the additive ensembles constructed in boosting methods, and provides an explanation for a similar widely-used practical heuristic for initializing the new classifier in each round of boosting. We also prove that for certain deep linear residual networks, stacking does provide accelerated training, via a new potential function analysis of the Nesterov's accelerated gradient method which allows errors in updates. We conduct proof-of-concept experiments to validate our theory as well.
Key facts. Cited by 5 papers in the map (data through 7 Oct 2026). Builds on 24 of its 40 references that are in the map. First posted on arXiv 8 Mar 2024.
Builds on
24 of 40 references.
References · 24
- 2017Attention Is All You Needcited by 59,932
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2023GPT-4 Technical Reportcited by 18,132
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 6,979
- 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,304
- 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,120
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
- 2020Longformer: The Long-Document Transformercited by 3,362
- 2024Mixtral of Expertscited by 1,785
- 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 896
12 more
- 2022PaLI: A Jointly-Scaled Multilingual Language-Image Modelcited by 762
- 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 748
- 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 413
- 2023Towards Generalist Biomedical AIcited by 286
- 2023Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-trainingcited by 260
- 2016Deep Learning without Poor Local Minimacited by 233
- 2016Identity Matters in Deep Learningcited by 107
- 2014Linear Coupling: An Ultimate Unification of Gradient and Mirror Descentcited by 58
- 2020On the Transformer Growth for Progressive BERT Trainingcited by 49
- 2022Staged Training for Transformer Language Modelscited by 49
- 2020Understanding Nesterov's Acceleration via Proximal Point Methodcited by 7
- 2017Potential-Function Proofs for First-Order Methodscited by 3
Built on by
Citing papers per year: 2024: 2; 2025: 2; 2026: 1.
5 papers.
- 2024When Attention Collapses: How Degenerate Layers in LLMs Enable Smaller, Stronger Modelscited by 12
- 2024Model-GLUE: Democratized LLM Scaling for A Large Model Zoo in the Wildcited by 9
- 2025All is Not Lost: LLM Recovery without Checkpointscited by 1
- 2025Scaling depth capacity via zero/one-layer model expansioncited by 1
- 2026Expert Upcycling: Shifting the Compute-Efficient Frontier of Mixture-of-Expertscited by 0
- 2026Expert Upcycling: Shifting the Compute-Efficient Frontier of Mixture-of-Experts21 Apr 2026
- 2025Scaling depth capacity via zero/one-layer model expansion7 Nov 2025
- 2025All is Not Lost: LLM Recovery without Checkpoints18 Jun 2025
- 2024Model-GLUE: Democratized LLM Scaling for A Large Model Zoo in the Wild7 Oct 2024
- 2024When Attention Collapses: How Degenerate Layers in LLMs Enable Smaller, Stronger Models12 Apr 2024