Memory Savings at What Cost? A Study of Alternatives to Backpropagation
Not on the map. Why?
Abstract (arXiv)
Forward-mode automatic differentiation (FmAD) and zero-order (ZO) optimization are increasingly proposed as memory-efficient, backpropagation-free alternatives for large language model (LLM) fine-tuning, yet their benefits are typically evaluated only against standard backpropagation (BP), omitting memory-efficient variants such as activation checkpointing. We present a unified theoretical and empirical comparison of BP, checkpointed BP, FmAD, and ZO for LLM and vision-language model training, showing that while FmAD and ZO reduce activation memory, they trade memory for higher computational cost and longer wall-clock time to convergence, resulting in lower accuracy and slower training, especially under constrained perturbation budgets. Across models, BP with checkpointing outperforms FmAD and ZO variants, including variance-reduced methods, achieving up to 31.1% higher accuracy, 34.8% faster convergence, and 3.8x fewer computations at comparable memory usage, while also revealing instability-related failure modes in FmAD and ZO. Overall, our results correct a one-sided benchmarking narrative by showing that memory-efficient methods entail fundamentally different trade-offs, and that ignoring these distinctions has led to misleading conclusions about LLM optimization in prior work. Our source code is available at {https://github.com/Astuary/Gradient_Estimation_Methods}.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 42 of its 59 references that are in the map. First posted on arXiv 27 Jun 2025.
Builds on
42 of 59 references.
References · 42
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2014Microsoft COCO: Common Objects in Contextcited by 17,608
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2020Measuring Massive Multitask Language Understandingcited by 7,963
- 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,512
- 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,418
- 2016Making the V in VQA Matter: Elevating the Role of Image Understanding in Visual Question Answeringcited by 2,516
- 2015Character-level Convolutional Networks for Text Classificationcited by 2,330
- 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,183
- 2022GPTQ: Accurate Post-Training Quantization for Generative Pre-trained Transformerscited by 1,919
30 more
- 2019Towards VQA Models That Can Readcited by 1,718
- 2020Making Pre-trained Language Models Better Few-shot Learnerscited by 1,256
- 2016Optimization Methods for Large-Scale Machine Learningcited by 1,150
- 2015Automatic differentiation in machine learning: a surveycited by 855
- 2016Training Deep Nets with Sublinear Memory Costcited by 848
- 2023Fine-Tuning Language Models with Just Forward Passescited by 345
- 2017The Reversible Residual Network: Backpropagation Without Storing Activationscited by 217
- 2023Handbook of Convergence Theorems for (Stochastic) Gradient Methodscited by 163
- 2024Revisiting Zeroth-Order Optimization for Memory-Efficient LLM Fine-Tuning: A Benchmarkcited by 114
- 2018Zeroth-Order Stochastic Variance Reduction for Nonconvex Optimizationcited by 111
- 2016Memory-Efficient Backpropagation Through Timecited by 78
- 2019ZO-AdaMM: Zeroth-Order Adaptive Momentum Method for Black-Box Optimizationcited by 76
- 2023DeepZero: Scaling up Zeroth-Order Optimization for Deep Model Trainingcited by 76
- 2022Gradients without Backpropagationcited by 73
- 2017Stochastic Zeroth-order Optimization in High Dimensionscited by 63
- 2024Variance-reduced Zeroth-Order Methods for Fine-Tuning Language Modelscited by 54
- 2024Second-Order Fine-Tuning without Pain for LLMs:A Hessian Informed Zeroth-Order Optimizercited by 53
- 2018Zeroth-order Nonconvex Stochastic Optimization: Handling Constraints, High-Dimensionality and Saddle-Pointscited by 37
- 2023Can Forward Gradient Match Backpropagation?cited by 26
- 2023An Algorithm with Optimal Dimension-Dependence for Zero-Order Nonsmooth Nonconvex Stochastic Optimizationcited by 23
- 2023BAFFLE: A Baseline of Backpropagation-Free Federated Learningcited by 12
- 2023An Optimal Structured Zeroth-order Algorithm for Non-smooth Optimizationcited by 10
- 2024Thinking Forward: Memory-Efficient Federated Finetuning of Language Modelscited by 7
- 2024Beyond Backpropagation: Optimization with Multi-Tangent Forward Gradientscited by 6
- 2024Projected Forward Gradient-Guided Frank-Wolfe Algorithm via Variance Reductioncited by 5
- 2024Second-Order Forward-Mode Automatic Differentiation for Optimizationcited by 5
- 2023Interpreting learning in biological neural networks as zero-order optimization methodcited by 4
- 2023Randomized Forward Mode of Automatic Differentiation For Optimization Algorithmscited by 3
- 2023The Black-Box Optimization Problem: Zero-Order Accelerated Stochastic Method via Kernel Approximationcited by 2
- 2023A Zeroth-Order Variance-Reduced Method for Decentralized Stochastic Non-convex Optimizationcited by 1
Built on by
Citing papers per year: 2025: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026GRZO: Group-Relative Zeroth-Order Optimization for Large Language Model Fine-Tuning
- 2025StreamBP: Memory-Efficient Exact Backpropagation for Long Sequence Training of LLMs
- 2026LCSB: Layer-Cyclic Selective Backpropagation for Memory-Efficient On-Device LLM Fine-Tuning
- 2026Parameter Efficiency Is Not Memory Efficiency: Rethinking Fine-Tuning for On-Device LLM Adaptation
- 2026GNMR: Runtime Stability Control for Low-Precision Large Language Model Training
- 2025ZO2: Scalable Zeroth-Order Fine-Tuning for Extremely Large Language Models with Limited GPU Memory
- 2026FLOP-Efficient Training: Early Stopping Based on Test-Time Compute Awareness
- 2026ZO-Act: Efficient Zeroth-Order Fine-Tuning via One-Shot Activation-Informed Low-Rank Subspaces