Low-rank Gradient Approximation For Memory-Efficient On-device Training of Deep Neural Network
Not on the map. Why?
Abstract (arXiv)
Training machine learning models on mobile devices has the potential of improving both privacy and accuracy of the models. However, one of the major obstacles to achieving this goal is the memory limitation of mobile devices. Reducing training memory enables models with high-dimensional weight matrices, like automatic speech recognition (ASR) models, to be trained on-device. In this paper, we propose approximating the gradient matrices of deep neural networks using a low-rank parameterization as an avenue to save training memory. The low-rank gradient approximation enables more advanced, memory-intensive optimization techniques to be run on device. Our experimental results show that we can reduce the training memory by about 33.0% for Adam optimization. It uses comparable memory to momentum optimization and achieves a 4.5% relative lower word error rate on an ASR personalization task.
Key facts. Cited by 20 papers in the map (data through 7 Oct 2026). First posted on arXiv 24 Jan 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 3; 2022: 3; 2023: 3; 2024: 4; 2025: 5; 2026: 2.
20 papers.
Sort:
- 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 447
- 2021Large Scale Private Learning via Low-rank Reparametrizationcited by 87
- 2021Do Not Let Privacy Overbill Utility: Gradient Embedding Perturbation for Private Learningcited by 86
- 2024AdaRankGrad: Adaptive Gradient-Rank and Moments for Memory-Efficient LLMs Training and Fine-Tuningcited by 16
- 2025SUMO: Subspace-Aware Moment-Orthogonalization for Accelerating Memory-Efficient LLM Trainingcited by 16
- 2023Efficient Storage of Fine-Tuned Models via Low-Rank Approximation of Weight Residualscited by 12
- 2024Memory-Efficient Optimization with Factorized Hamiltonian Descentcited by 11
- 2023Surrogate Model Extension (SME): A Fast and Accurate Weight Update Attack on Federated Learningcited by 10
- 2021Improving Differentially Private SGD via Randomly Sparsified Gradientscited by 8
- 2022p-Meta: Towards On-device Deep Model Adaptationcited by 8
- 2022Federated Optimization of Smooth Loss Functionscited by 6
- 2025LORENZA: Enhancing Generalization in Low-Rank Gradient LLM Training via Efficient Zeroth-Order Adaptive SAMcited by 3
8 more
- 2025AdaPM: a Partial Momentum Algorithm for LLM Trainingcited by 3
- 2022Gradient Descent for Low-Rank Functionscited by 2
- 2026LoRIF: Low-Rank Influence Functions for Scalable Training Data Attributioncited by 2
- 2025On the Duality between Gradient Transformations and Adapterscited by 1
- 2023MobileASR: A resource-aware on-device learning framework for user voice personalization applications on mobile phonescited by 0
- 2024Weights Shuffling for Improving DPSGD in Transformer-based Modelscited by 0
- 2025Low-rank surrogate modeling and stochastic zero-order optimization for training of neural networks with black-box layerscited by 0
- 2026SDS-LoRA: Overcoming Anisotropic Gradient Scaling in Low-Rank Adaptationcited by 0
- 2026SDS-LoRA: Overcoming Anisotropic Gradient Scaling in Low-Rank Adaptation15 Jun 2026
- 2026LoRIF: Low-Rank Influence Functions for Scalable Training Data Attribution29 Jan 2026
- 2025AdaPM: a Partial Momentum Algorithm for LLM Training10 Oct 2025
- 2025Low-rank surrogate modeling and stochastic zero-order optimization for training of neural networks with black-box layers18 Sep 2025
- 2025SUMO: Subspace-Aware Moment-Orthogonalization for Accelerating Memory-Efficient LLM Training30 May 2025
- 2025LORENZA: Enhancing Generalization in Low-Rank Gradient LLM Training via Efficient Zeroth-Order Adaptive SAM26 Feb 2025
- 2025On the Duality between Gradient Transformations and Adapters19 Feb 2025
- 2024AdaRankGrad: Adaptive Gradient-Rank and Moments for Memory-Efficient LLMs Training and Fine-Tuning23 Oct 2024
- 2024Weights Shuffling for Improving DPSGD in Transformer-based Models22 Jul 2024
- 2024Memory-Efficient Optimization with Factorized Hamiltonian Descent14 Jun 2024
- 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projection6 Mar 2024
- 2023MobileASR: A resource-aware on-device learning framework for user voice personalization applications on mobile phones15 Jun 2023
8 more
- 2023Surrogate Model Extension (SME): A Fast and Accurate Weight Update Attack on Federated Learning31 May 2023
- 2023Efficient Storage of Fine-Tuned Models via Low-Rank Approximation of Weight Residuals28 May 2023
- 2022p-Meta: Towards On-device Deep Model Adaptation25 Jun 2022
- 2022Gradient Descent for Low-Rank Functions16 Jun 2022
- 2022Federated Optimization of Smooth Loss Functions6 Jan 2022
- 2021Improving Differentially Private SGD via Randomly Sparsified Gradients1 Dec 2021
- 2021Large Scale Private Learning via Low-rank Reparametrization17 Jun 2021
- 2021Do Not Let Privacy Overbill Utility: Gradient Embedding Perturbation for Private Learning25 Feb 2021
- 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 447
- 2021Large Scale Private Learning via Low-rank Reparametrizationcited by 87
- 2021Do Not Let Privacy Overbill Utility: Gradient Embedding Perturbation for Private Learningcited by 86
- 2024AdaRankGrad: Adaptive Gradient-Rank and Moments for Memory-Efficient LLMs Training and Fine-Tuningcited by 16
- 2025SUMO: Subspace-Aware Moment-Orthogonalization for Accelerating Memory-Efficient LLM Trainingcited by 16
- 2023Efficient Storage of Fine-Tuned Models via Low-Rank Approximation of Weight Residualscited by 12
- 2024Memory-Efficient Optimization with Factorized Hamiltonian Descentcited by 11
- 2023Surrogate Model Extension (SME): A Fast and Accurate Weight Update Attack on Federated Learningcited by 10
- 2021Improving Differentially Private SGD via Randomly Sparsified Gradientscited by 8
- 2022p-Meta: Towards On-device Deep Model Adaptationcited by 8
- 2022Federated Optimization of Smooth Loss Functionscited by 6
- 2025LORENZA: Enhancing Generalization in Low-Rank Gradient LLM Training via Efficient Zeroth-Order Adaptive SAMcited by 3
12 most cited of 20 All 20 (JSON)
- 2026SDS-LoRA: Overcoming Anisotropic Gradient Scaling in Low-Rank Adaptation15 Jun 2026
- 2026LoRIF: Low-Rank Influence Functions for Scalable Training Data Attribution29 Jan 2026
- 2025AdaPM: a Partial Momentum Algorithm for LLM Training10 Oct 2025
- 2025Low-rank surrogate modeling and stochastic zero-order optimization for training of neural networks with black-box layers18 Sep 2025
- 2025SUMO: Subspace-Aware Moment-Orthogonalization for Accelerating Memory-Efficient LLM Training30 May 2025
- 2025LORENZA: Enhancing Generalization in Low-Rank Gradient LLM Training via Efficient Zeroth-Order Adaptive SAM26 Feb 2025
- 2025On the Duality between Gradient Transformations and Adapters19 Feb 2025
- 2024AdaRankGrad: Adaptive Gradient-Rank and Moments for Memory-Efficient LLMs Training and Fine-Tuning23 Oct 2024
- 2024Weights Shuffling for Improving DPSGD in Transformer-based Models22 Jul 2024
- 2024Memory-Efficient Optimization with Factorized Hamiltonian Descent14 Jun 2024
- 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projection6 Mar 2024
- 2023MobileASR: A resource-aware on-device learning framework for user voice personalization applications on mobile phones15 Jun 2023
12 newest of 20 All 20 (JSON)