mlsanityData to 7 Oct 2026

Stacking as Accelerated Gradient Descent

arXiv 2403.04978 ↗First posted 8 Mar 2024cs.LG stat.MLcited by 5

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

  1. 2017Attention Is All You Needcited by 59,932
  2. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  3. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  4. 2020Language Models are Few-Shot Learnerscited by 32,199
  5. 2023GPT-4 Technical Reportcited by 18,132
  6. 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 6,979
  7. 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,304
  8. 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,120
  9. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
  10. 2020Longformer: The Long-Document Transformercited by 3,362
  11. 2024Mixtral of Expertscited by 1,785
  12. 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 896
12 more
  1. 2022PaLI: A Jointly-Scaled Multilingual Language-Image Modelcited by 762
  2. 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 748
  3. 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 413
  4. 2023Towards Generalist Biomedical AIcited by 286
  5. 2023Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-trainingcited by 260
  6. 2016Deep Learning without Poor Local Minimacited by 233
  7. 2016Identity Matters in Deep Learningcited by 107
  8. 2014Linear Coupling: An Ultimate Unification of Gradient and Mirror Descentcited by 58
  9. 2020On the Transformer Growth for Progressive BERT Trainingcited by 49
  10. 2022Staged Training for Transformer Language Modelscited by 49
  11. 2020Understanding Nesterov's Acceleration via Proximal Point Methodcited by 7
  12. 2017Potential-Function Proofs for First-Order Methodscited by 3

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 2; 2025: 2; 2026: 1.

5 papers.