Memory-Efficient Backpropagation Through Time
Not on the map. Why?
Abstract (arXiv)
We propose a novel approach to reduce memory consumption of the backpropagation through time (BPTT) algorithm when training recurrent neural networks (RNNs). Our approach uses dynamic programming to balance a trade-off between caching of intermediate results and recomputation. The algorithm is capable of tightly fitting within almost any user-set memory budget while finding an optimal execution policy minimizing the computational cost. Computational devices have limited memory capacity and maximizing a computational performance given a fixed memory budget is a practical use-case. We provide asymptotic computational upper bounds for various regimes. The algorithm is particularly effective for long sequences. For sequences of length 1000, our algorithm saves 95% of memory usage while using only one third more time per iteration than the standard BPTT.
Key facts. Cited by 78 papers in the map (data through 7 Oct 2026). First posted on arXiv 10 Jun 2016.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2017: 1; 2018: 0; 2019: 0; 2020: 0; 2021: 11; 2022: 14; 2023: 18; 2024: 18; 2025: 10; 2026: 6.
78 papers.
Sort:
By area:
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
- 2023Directly Fine-Tuning Diffusion Models on Differentiable Rewardscited by 431
- 2022Flexible Diffusion Modeling of Long Videoscited by 294
- 2023Aligning Text-to-Image Diffusion Models with Reward Backpropagationcited by 284
- 2024Fine-Tuning of Continuous-Time Diffusion Models as Entropy-Regularized Controlcited by 136
- 2023End-to-End Diffusion Latent Optimization Improves Classifier Guidancecited by 107
- 2023InstructVideo: Instructing Video Diffusion Models with Human Feedbackcited by 85
- 2022Tutorial on amortized optimizationcited by 83
- 2021A Survey on Green Deep Learningcited by 52
- 2024Feedback Efficient Online Fine-Tuning of Diffusion Modelscited by 52
- 2023Compressed Context Memory For Online Language Model Interactioncited by 42
- 2024Tiny Machine Learning: Progress and Futurescited by 40
38 more
- 2021Efficient Differentiable Simulation of Articulated Bodiescited by 38
- 2022Enable Deep Learning on Mobile Devices: Methods, Systems, and Applicationscited by 32
- 2022Beyond backpropagation: bilevel optimization through implicit differentiation and equilibrium propagationcited by 26
- 2023Brain-inspired learning in artificial neural networks: a reviewcited by 26
- 2023FusionAI: Decentralized Training and Deploying LLMs with Massive Consumer-Level GPUscited by 20
- 2024On the Resurgence of Recurrent Models for Long Sequences -- Survey and Research Opportunities in the Transformer Eracited by 17
- 2024ORBIT: Oak Ridge Base Foundation Model for Earth System Predictabilitycited by 17
- 2024Adding Conditional Control to Diffusion Models with Reinforcement Learningcited by 16
- 2021PatrickStar: Parallel Training of Pre-trained Models via Chunk-based Memory Managementcited by 15
- 2023DisCo-CLIP: A Distributed Contrastive Loss for Memory Efficient CLIP Trainingcited by 15
- 2021HyperSPNs: Compact and Expressive Probabilistic Circuitscited by 14
- 2024FedMef: Towards Memory-efficient Federated Dynamic Pruningcited by 14
- 2024Banishing LLM Hallucinations Requires Rethinking Generalizationcited by 13
- 2025Neural ODE Transformers: Analyzing Internal Dynamics and Adaptive Fine-tuningcited by 12
- 2025Learning Regularization Functionals for Inverse Problems: A Comparative Studycited by 12
- 2021Enabling Binary Neural Network Training on the Edgecited by 11
- 2021Symplectic Adjoint Method for Exact Gradient of Neural ODE with Minimal Memorycited by 11
- 2021Efficient Scaling of Dynamic Graph Neural Networkscited by 11
- 2023Score-based Data Assimilation for a Two-Layer Quasi-Geostrophic Modelcited by 11
- 2022Recurrent Neural Networks for Dynamical Systems: Applications to Ordinary Differential Equations, Collective Motion, and Hydrological Modelingcited by 9
- 2022DropIT: Dropping Intermediate Tensors for Memory-Efficient DNN Trainingcited by 9
- 2024Learnable Sparsity for Vision Generative Modelscited by 9
- 2022p-Meta: Towards On-device Deep Model Adaptationcited by 8
- 2023Enabling Resource-efficient AIoT System with Cross-level Optimization: A surveycited by 8
- 2024FREE: Faster and Better Data-Free Meta-Learningcited by 8
- 2023Long-term Dependency for 3D Reconstruction of Freehand Ultrasound Without External Trackercited by 7
- 2024FusionLLM: A Decentralized LLM Training System on Geo-distributed GPUs with Adaptive Compressioncited by 7
- 2021Making EfficientNet More Efficient: Exploring Batch-Independent Normalization, Group Convolutions and Reduced Resolution Trainingcited by 6
- 2022RevBiFPN: The Fully Reversible Bidirectional Feature Pyramid Networkcited by 6
- 2023Unsupervised Discovery of Interpretable Directions in h-space of Pre-trained Diffusion Modelscited by 6
- 2024Projected Forward Gradient-Guided Frank-Wolfe Algorithm via Variance Reductioncited by 5
- 2024On-Device Training of Fully Quantized Deep Neural Networks on Cortex-M Microcontrollerscited by 5
- 2025Half-order Fine-Tuning for Diffusion Model: A Recursive Likelihood Ratio Optimizercited by 5
- 2025From Tiny Machine Learning to Tiny Deep Learning: A Surveycited by 5
- 2022Survey on Large Scale Neural Network Trainingcited by 4
- 2022DIVISION: Memory Efficient Training via Dual Activation Precisioncited by 4
- 2024Block Selective Reprogramming for On-device Training of Vision Transformerscited by 4
- 2024Resource Management for Low-latency Cooperative Fine-tuning of Foundation Models at the Network Edgecited by 4
50 most cited of 78 All 78 (JSON)
- 2026Dude, Where's My State? Execution Information Requirements for Stateful Agents26 Sep 2026
- 2026Large Distant Gradients Need Not Be Reliable: reliability-weighted credit assignment for long-horizon autoregressive forecasting11 Sep 2026
- 2026TERRA: A Hierarchical Parallel Training and Memory Orchestration Framework for High-Resolution AI-based Earth Modeling15 Aug 2026
- 2026Memory-Efficient Activation Checkpointing with Sliding Window and Hirschberg's Algorithm for 0/1 Knapsack Solving in PyTorch9 Aug 2026
- 2026WHALE: A Scalable Unified Model for Recommendation with Wukong-HSTU Architecture19 Jul 2026
- 2026Dynamical Systems Theory Behind a Hierarchical Reasoning Model24 Mar 2026
- 2025GLUE: Gradient-free Learning to Unify Experts27 Dec 2025
- 2025Learning Regularization Functionals for Inverse Problems: A Comparative Study2 Oct 2025
- 2025DaCe AD: Unifying High-Performance Automatic Differentiation for Machine Learning and Scientific Computing2 Sep 2025
- 2025Memory Savings at What Cost? A Study of Alternatives to Backpropagation27 Jun 2025
- 2025From Tiny Machine Learning to Tiny Deep Learning: A Survey21 Jun 2025
- 2025Text2Stereo: Repurposing Stable Diffusion for Stereo Generation with Consistency Rewards27 May 2025
38 more
- 2025Neural ODE Transformers: Analyzing Internal Dynamics and Adaptive Fine-tuning3 Mar 2025
- 2025GPU Memory Usage Optimization for Backward Propagation in Deep Network Training18 Feb 2025
- 2025Half-order Fine-Tuning for Diffusion Model: A Recursive Likelihood Ratio Optimizer2 Feb 2025
- 2025Memory-Efficient Fine-Tuning of Transformers via Token Selection31 Jan 2025
- 2024Optimal Gradient Checkpointing for Sparse and Recurrent Architectures using Off-Chip Memory16 Dec 2024
- 2024Learnable Sparsity for Vision Generative Models3 Dec 2024
- 2024Automatic Differentiation-based Full Waveform Inversion with Flexible Workflows30 Nov 2024
- 2024FusionLLM: A Decentralized LLM Training System on Geo-distributed GPUs with Adaptive Compression16 Oct 2024
- 2024On-Device Training of Fully Quantized Deep Neural Networks on Cortex-M Microcontrollers15 Jul 2024
- 2024Resource Management for Low-latency Cooperative Fine-tuning of Foundation Models at the Network Edge13 Jul 2024
- 2024Banishing LLM Hallucinations Requires Rethinking Generalization25 Jun 2024
- 2024Adding Conditional Control to Diffusion Models with Reinforcement Learning17 Jun 2024
- 2024FREE: Faster and Better Data-Free Meta-Learning2 May 2024
- 2024ORBIT: Oak Ridge Base Foundation Model for Earth System Predictability23 Apr 2024
- 2024Tiny Machine Learning: Progress and Futures28 Mar 2024
- 2024Block Selective Reprogramming for On-device Training of Vision Transformers25 Mar 2024
- 2024FedMef: Towards Memory-efficient Federated Dynamic Pruning21 Mar 2024
- 2024Projected Forward Gradient-Guided Frank-Wolfe Algorithm via Variance Reduction19 Mar 2024
- 2024Feedback Efficient Online Fine-Tuning of Diffusion Models26 Feb 2024
- 2024Fine-Tuning of Continuous-Time Diffusion Models as Entropy-Regularized Control23 Feb 2024
- 2024Moonwalk: Inverse-Forward Differentiation22 Feb 2024
- 2024On the Resurgence of Recurrent Models for Long Sequences -- Survey and Research Opportunities in the Transformer Era12 Feb 2024
- 2023InstructVideo: Instructing Video Diffusion Models with Human Feedback19 Dec 2023
- 2023Compressed Context Memory For Online Language Model Interaction6 Dec 2023
- 2023Coop: Memory is not a Commodity1 Nov 2023
- 2023FetusMapV2: Enhanced Fetal Pose Estimation in 3D Ultrasound30 Oct 2023
- 2023Long-term Dependency for 3D Reconstruction of Freehand Ultrasound Without External Tracker16 Oct 2023
- 2023Unsupervised Discovery of Interpretable Directions in h-space of Pre-trained Diffusion Models15 Oct 2023
- 2023Fast-ELECTRA for Efficient Pre-training11 Oct 2023
- 2023Aligning Text-to-Image Diffusion Models with Reward Backpropagation5 Oct 2023
- 2023OneAdapt: Fast Configuration Adaptation for Video Analytics Applications via Backpropagation3 Oct 2023
- 2023Score-based Data Assimilation for a Two-Layer Quasi-Geostrophic Model3 Oct 2023
- 2023Directly Fine-Tuning Diffusion Models on Differentiable Rewards29 Sep 2023
- 2023Enabling Resource-efficient AIoT System with Cross-level Optimization: A survey27 Sep 2023
- 2023FusionAI: Decentralized Training and Deploying LLMs with Massive Consumer-Level GPUs3 Sep 2023
- 2023Brain-inspired learning in artificial neural networks: a review18 May 2023
- 2023DisCo-CLIP: A Distributed Contrastive Loss for Memory Efficient CLIP Training17 Apr 2023
- 2023End-to-End Diffusion Latent Optimization Improves Classifier Guidance23 Mar 2023
50 newest of 78 All 78 (JSON)
- 2023Directly Fine-Tuning Diffusion Models on Differentiable Rewardscited by 431
- 2022Flexible Diffusion Modeling of Long Videoscited by 294
- 2023Aligning Text-to-Image Diffusion Models with Reward Backpropagationcited by 284
- 2024Fine-Tuning of Continuous-Time Diffusion Models as Entropy-Regularized Controlcited by 136
- 2023End-to-End Diffusion Latent Optimization Improves Classifier Guidancecited by 107
- 2023InstructVideo: Instructing Video Diffusion Models with Human Feedbackcited by 85
- 2024Feedback Efficient Online Fine-Tuning of Diffusion Modelscited by 52
- 2024Adding Conditional Control to Diffusion Models with Reinforcement Learningcited by 16
- 2024Learnable Sparsity for Vision Generative Modelscited by 9
- 2023Unsupervised Discovery of Interpretable Directions in h-space of Pre-trained Diffusion Modelscited by 6
- 2025Half-order Fine-Tuning for Diffusion Model: A Recursive Likelihood Ratio Optimizercited by 5
- 2025Text2Stereo: Repurposing Stable Diffusion for Stereo Generation with Consistency Rewardscited by 2
- 2025Text2Stereo: Repurposing Stable Diffusion for Stereo Generation with Consistency Rewards27 May 2025
- 2025Half-order Fine-Tuning for Diffusion Model: A Recursive Likelihood Ratio Optimizer2 Feb 2025
- 2024Learnable Sparsity for Vision Generative Models3 Dec 2024
- 2024Adding Conditional Control to Diffusion Models with Reinforcement Learning17 Jun 2024
- 2024Feedback Efficient Online Fine-Tuning of Diffusion Models26 Feb 2024
- 2024Fine-Tuning of Continuous-Time Diffusion Models as Entropy-Regularized Control23 Feb 2024
- 2023InstructVideo: Instructing Video Diffusion Models with Human Feedback19 Dec 2023
- 2023Unsupervised Discovery of Interpretable Directions in h-space of Pre-trained Diffusion Models15 Oct 2023
- 2023Aligning Text-to-Image Diffusion Models with Reward Backpropagation5 Oct 2023
- 2023Directly Fine-Tuning Diffusion Models on Differentiable Rewards29 Sep 2023
- 2023End-to-End Diffusion Latent Optimization Improves Classifier Guidance23 Mar 2023
- 2022Flexible Diffusion Modeling of Long Videos23 May 2022
- 2022Tutorial on amortized optimizationcited by 83
- 2021A Survey on Green Deep Learningcited by 52
- 2023Compressed Context Memory For Online Language Model Interactioncited by 42
- 2024Tiny Machine Learning: Progress and Futurescited by 40
- 2021Efficient Differentiable Simulation of Articulated Bodiescited by 38
- 2022Enable Deep Learning on Mobile Devices: Methods, Systems, and Applicationscited by 32
- 2022Beyond backpropagation: bilevel optimization through implicit differentiation and equilibrium propagationcited by 26
- 2023Brain-inspired learning in artificial neural networks: a reviewcited by 26
- 2023FusionAI: Decentralized Training and Deploying LLMs with Massive Consumer-Level GPUscited by 20
- 2024On the Resurgence of Recurrent Models for Long Sequences -- Survey and Research Opportunities in the Transformer Eracited by 17
- 2024ORBIT: Oak Ridge Base Foundation Model for Earth System Predictabilitycited by 17
- 2021PatrickStar: Parallel Training of Pre-trained Models via Chunk-based Memory Managementcited by 15
12 most cited of 62 All 62 (JSON)
- 2026Dude, Where's My State? Execution Information Requirements for Stateful Agents26 Sep 2026
- 2026Large Distant Gradients Need Not Be Reliable: reliability-weighted credit assignment for long-horizon autoregressive forecasting11 Sep 2026
- 2026TERRA: A Hierarchical Parallel Training and Memory Orchestration Framework for High-Resolution AI-based Earth Modeling15 Aug 2026
- 2026Memory-Efficient Activation Checkpointing with Sliding Window and Hirschberg's Algorithm for 0/1 Knapsack Solving in PyTorch9 Aug 2026
- 2026WHALE: A Scalable Unified Model for Recommendation with Wukong-HSTU Architecture19 Jul 2026
- 2025GLUE: Gradient-free Learning to Unify Experts27 Dec 2025
- 2025Learning Regularization Functionals for Inverse Problems: A Comparative Study2 Oct 2025
- 2025DaCe AD: Unifying High-Performance Automatic Differentiation for Machine Learning and Scientific Computing2 Sep 2025
- 2025Memory Savings at What Cost? A Study of Alternatives to Backpropagation27 Jun 2025
- 2025From Tiny Machine Learning to Tiny Deep Learning: A Survey21 Jun 2025
- 2025Neural ODE Transformers: Analyzing Internal Dynamics and Adaptive Fine-tuning3 Mar 2025
- 2025GPU Memory Usage Optimization for Backward Propagation in Deep Network Training18 Feb 2025
12 newest of 62 All 62 (JSON)