LDAdam: Adaptive Optimization from Low-Dimensional Gradient Statistics
Not on the map. Why?
Abstract (arXiv)
We introduce LDAdam, a memory-efficient optimizer for training large models, that performs adaptive optimization steps within lower dimensional subspaces, while consistently exploring the full parameter space during training. This strategy keeps the optimizer's memory footprint to a fraction of the model size. LDAdam relies on a new projection-aware update rule for the optimizer states that allows for transitioning between subspaces, i.e., estimation of the statistics of the projected gradients. To mitigate the errors due to low-rank projection, LDAdam integrates a new generalized error feedback mechanism, which explicitly accounts for both gradient and optimizer state compression. We prove the convergence of LDAdam under standard assumptions, and show that LDAdam allows for accurate and efficient fine-tuning and pre-training of language models. Code is available at https://github.com/IST-DASLab/LDAdam
Key facts. Cited by 35 papers in the map (data through 7 Oct 2026). Builds on 46 of its 61 references that are in the map. First posted on arXiv 21 Oct 2024.
Builds on
46 of 61 references.
References · 46
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,854
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
- 2021Prefix-Tuning: Optimizing Continuous Prompts for Generationcited by 3,878
- 2020GLU Variants Improve Transformercited by 1,724
- 2016Understanding deep learning requires rethinking generalizationcited by 1,657
- 2019Root Mean Square Layer Normalizationcited by 1,425
34 more
- 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 896
- 2019On the Convergence of Adam and Beyondcited by 834
- 2024DoRA: Weight-Decomposed Low-Rank Adaptationcited by 740
- 2024Parameter-Efficient Fine-Tuning for Large Models: A Comprehensive Surveycited by 641
- 2017Deep Gradient Compression: Reducing the Communication Bandwidth for Distributed Trainingcited by 471
- 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 447
- 20218-bit Optimizers via Block-wise Quantizationcited by 395
- 2018Sparsified SGD with Memorycited by 373
- 2018Measuring the Intrinsic Dimension of Objective Landscapescited by 364
- 2020Parameter-Efficient Transfer Learning with Diff Pruningcited by 351
- 2023AdaLoRA: Adaptive Budget Allocation for Parameter-Efficient Fine-Tuningcited by 337
- 2019Error Feedback Fixes SignSGD and other Gradient Compression Schemescited by 306
- 2019PowerSGD: Practical Low-Rank Gradient Compression for Distributed Optimizationcited by 234
- 2018The Convergence of Sparsified Gradient Methodscited by 228
- 2023ReLoRA: High-Rank Training Through Low-Rank Updatescited by 200
- 2018On the Convergence of A Class of Adam-Type Algorithms for Non-Convex Optimizationcited by 173
- 2022Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Modelscited by 172
- 2020A Simple Convergence Proof of Adam and Adagradcited by 169
- 2023Full Parameter Fine-tuning for Large Language Models with Limited Resourcescited by 134
- 2024Adam-mini: Use Fewer Learning Rates To Gain Morecited by 134
- 2022Adam Can Converge Without Any Modification On Update Rulescited by 112
- 2023Convergence of Adam Under Relaxed Assumptionscited by 97
- 2018On the Convergence of Adaptive Gradient Methods for Nonconvex Optimizationcited by 91
- 2017Implicit Regularization in Deep Learningcited by 90
- 2024Chain of LoRA: Efficient Fine-tuning of Language Models via Residual Learningcited by 80
- 2013Accelerated Gradient Methods for Nonconvex Nonlinear and Stochastic Programmingcited by 74
- 2024RoSA: Accurate Parameter-Efficient Fine-Tuning via Robust Adaptationcited by 48
- 2024Q-GaLore: Quantized GaLore with INT4 Projection and Layer-Adaptive Low-Rank Gradientscited by 36
- 2023AdaLomo: Low-memory Optimization with Adaptive Learning Ratecited by 30
- 2024From Low Rank Gradient Subspace Stabilization to Low-Rank Weights: Observations, Theories, and Applicationscited by 24
- 2024On Convergence of Adam for Stochastic Optimization under Relaxed Assumptionscited by 21
- 2022On Distributed Adaptive Optimization with Gradient Compressioncited by 19
- 2024MicroAdam: Accurate Adaptive Optimization with Low Space Overhead and Provable Convergencecited by 19
- 2023Convergence of Adam for Non-convex Objectives: Relaxed Hyperparameters and Non-ergodic Casecited by 13
Built on by
Citing papers per year: 2024: 2; 2025: 19; 2026: 14.
35 papers.
Sort:
- 2024APOLLO: SGD-like Memory, AdamW-level Performancecited by 50
- 2025Purifying Shampoo: Investigating Shampoo's Heuristics by Decomposing its Preconditionercited by 21
- 2025PoLAR: Polar-Decomposed Low-Rank Adapter Representationcited by 19
- 2025SubTrack++ : Gradient Subspace Tracking for Scalable LLM Trainingcited by 14
- 2025GaLore 2: Large-Scale LLM Pre-Training by Gradient Low-Rank Projectioncited by 13
- 2025Greedy Low-Rank Gradient Compression for Distributed Learning with Convergence Guaranteescited by 9
- 2025On the Convergence Rate of LoRA Gradient Descentcited by 9
- 2024FRUGAL: Memory-Efficient Optimization by Reducing State Overhead for Scalable Trainingcited by 7
- 2025CR-Net: Scaling Parameter-Efficient Training with Cross-Layer Low-Rank Structurecited by 7
- 2025Scalable Parameter and Memory Efficient Pretraining for LLM: Recent Algorithmic Advances and Benchmarkingcited by 4
- 2025Low-rank Momentum Factorization for Memory Efficient Trainingcited by 4
- 2025LORENZA: Enhancing Generalization in Low-Rank Gradient LLM Training via Efficient Zeroth-Order Adaptive SAMcited by 3
23 more
- 2025Unified Scaling Laws for Compressed Representationscited by 3
- 2025Pay Attention to Small Weightscited by 3
- 2025FFT-based Dynamic Subspace Selection for Low-Rank Adaptive Optimization of Large Language Modelscited by 2
- 2025MT-DAO: Multi-Timescale Distributed Adaptive Optimizers with Local Updatescited by 2
- 2026Mitigating Staleness in Asynchronous Pipeline Parallelism via Basis Rotationcited by 2
- 2026MGUP: A Momentum-Gradient Alignment Update Policy for Stochastic Optimizationcited by 2
- 2025MLorc: Momentum Low-rank Compression for Memory Efficient Large Language Model Adaptationcited by 1
- 2025Geometrically Principled Randomized Optimization for Efficient LLM Trainingcited by 1
- 2025FOAM: Blocked State Folding for Memory-Efficient LLM Trainingcited by 1
- 2026LoRDO: Distributed Low-Rank Optimization with Infrequent Communicationcited by 1
- 2026Train Less, Infer Faster: Efficient Model Finetuning and Compression via Structured Sparsitycited by 1
- 2026OASIS: Online Activation Subspace Learning for Memory-Efficient Trainingcited by 1
- 2026Subspace Optimization for Efficient Federated Learning under Heterogeneous Datacited by 1
- 2025GWT: Scalable Optimizer State Compression for Large Language Model Trainingcited by 0
- 2025Bridging Training and Merging Through Momentum-Aware Optimizationcited by 0
- 2026Emergent Low-Rank Training Dynamics in MLPs with Smooth Activationscited by 0
- 2026One Algorithm, Two Goals: Dual Scoring for Parameter and Data Selection in LLM Fine-Tuningcited by 0
- 2026PowerStep: Memory-Efficient Adaptive Optimization via ℓp-Norm Steepest Descentcited by 0
- 2026Memory-Efficient LLM Training with Dynamic Sparsity: From Stability to Practical Scalingcited by 0
- 2026No Subspace to Track: Non-Identifiability and Optimizer State in Low-Rank Trainingcited by 0
- 2026Curvature-Conditioned Multiscale Momentum with Sphere Constraints for LLM Pretrainingcited by 0
- 2026Blog: Survey of Optimizerscited by 0
- 2026MoARa: Module-Aware Rank Allocation and Structure-Preserving Decomposition for Low-Rank LLM Pre-trainingcited by 0
- 2026MoARa: Module-Aware Rank Allocation and Structure-Preserving Decomposition for Low-Rank LLM Pre-training14 Sep 2026
- 2026Blog: Survey of Optimizers28 Aug 2026
- 2026Curvature-Conditioned Multiscale Momentum with Sphere Constraints for LLM Pretraining28 Aug 2026
- 2026No Subspace to Track: Non-Identifiability and Optimizer State in Low-Rank Training7 Jul 2026
- 2026MGUP: A Momentum-Gradient Alignment Update Policy for Stochastic Optimization16 Jun 2026
- 2026Memory-Efficient LLM Training with Dynamic Sparsity: From Stability to Practical Scaling30 May 2026
- 2026PowerStep: Memory-Efficient Adaptive Optimization via ℓp-Norm Steepest Descent11 May 2026
- 2026One Algorithm, Two Goals: Dual Scoring for Parameter and Data Selection in LLM Fine-Tuning7 May 2026
- 2026Subspace Optimization for Efficient Federated Learning under Heterogeneous Data28 Apr 2026
- 2026OASIS: Online Activation Subspace Learning for Memory-Efficient Training10 Apr 2026
- 2026Train Less, Infer Faster: Efficient Model Finetuning and Compression via Structured Sparsity9 Feb 2026
- 2026Emergent Low-Rank Training Dynamics in MLPs with Smooth Activations5 Feb 2026
23 more
- 2026LoRDO: Distributed Low-Rank Optimization with Infrequent Communication4 Feb 2026
- 2026Mitigating Staleness in Asynchronous Pipeline Parallelism via Basis Rotation3 Feb 2026
- 2025On the Convergence Rate of LoRA Gradient Descent20 Dec 2025
- 2025Bridging Training and Merging Through Momentum-Aware Optimization18 Dec 2025
- 2025FOAM: Blocked State Folding for Memory-Efficient LLM Training8 Dec 2025
- 2025MT-DAO: Multi-Timescale Distributed Adaptive Optimizers with Local Updates6 Oct 2025
- 2025Geometrically Principled Randomized Optimization for Efficient LLM Training2 Oct 2025
- 2025CR-Net: Scaling Parameter-Efficient Training with Cross-Layer Low-Rank Structure23 Sep 2025
- 2025Greedy Low-Rank Gradient Compression for Distributed Learning with Convergence Guarantees11 Jul 2025
- 2025Low-rank Momentum Factorization for Memory Efficient Training10 Jul 2025
- 2025Pay Attention to Small Weights26 Jun 2025
- 2025Purifying Shampoo: Investigating Shampoo's Heuristics by Decomposing its Preconditioner4 Jun 2025
- 2025PoLAR: Polar-Decomposed Low-Rank Adapter Representation3 Jun 2025
- 2025MLorc: Momentum Low-rank Compression for Memory Efficient Large Language Model Adaptation2 Jun 2025
- 2025Unified Scaling Laws for Compressed Representations2 Jun 2025
- 2025Scalable Parameter and Memory Efficient Pretraining for LLM: Recent Algorithmic Advances and Benchmarking28 May 2025
- 2025FFT-based Dynamic Subspace Selection for Low-Rank Adaptive Optimization of Large Language Models23 May 2025
- 2025GaLore 2: Large-Scale LLM Pre-Training by Gradient Low-Rank Projection29 Apr 2025
- 2025LORENZA: Enhancing Generalization in Low-Rank Gradient LLM Training via Efficient Zeroth-Order Adaptive SAM26 Feb 2025
- 2025SubTrack++ : Gradient Subspace Tracking for Scalable LLM Training3 Feb 2025
- 2025GWT: Scalable Optimizer State Compression for Large Language Model Training13 Jan 2025
- 2024APOLLO: SGD-like Memory, AdamW-level Performance6 Dec 2024
- 2024FRUGAL: Memory-Efficient Optimization by Reducing State Overhead for Scalable Training12 Nov 2024
- 2024APOLLO: SGD-like Memory, AdamW-level Performancecited by 50
- 2025Purifying Shampoo: Investigating Shampoo's Heuristics by Decomposing its Preconditionercited by 21
- 2025PoLAR: Polar-Decomposed Low-Rank Adapter Representationcited by 19
- 2025SubTrack++ : Gradient Subspace Tracking for Scalable LLM Trainingcited by 14
- 2025GaLore 2: Large-Scale LLM Pre-Training by Gradient Low-Rank Projectioncited by 13
- 2025Greedy Low-Rank Gradient Compression for Distributed Learning with Convergence Guaranteescited by 9
- 2025On the Convergence Rate of LoRA Gradient Descentcited by 9
- 2024FRUGAL: Memory-Efficient Optimization by Reducing State Overhead for Scalable Trainingcited by 7
- 2025CR-Net: Scaling Parameter-Efficient Training with Cross-Layer Low-Rank Structurecited by 7
- 2025Scalable Parameter and Memory Efficient Pretraining for LLM: Recent Algorithmic Advances and Benchmarkingcited by 4
- 2025Low-rank Momentum Factorization for Memory Efficient Trainingcited by 4
- 2025LORENZA: Enhancing Generalization in Low-Rank Gradient LLM Training via Efficient Zeroth-Order Adaptive SAMcited by 3
12 most cited of 35 All 35 (JSON)
- 2026MoARa: Module-Aware Rank Allocation and Structure-Preserving Decomposition for Low-Rank LLM Pre-training14 Sep 2026
- 2026Blog: Survey of Optimizers28 Aug 2026
- 2026Curvature-Conditioned Multiscale Momentum with Sphere Constraints for LLM Pretraining28 Aug 2026
- 2026No Subspace to Track: Non-Identifiability and Optimizer State in Low-Rank Training7 Jul 2026
- 2026MGUP: A Momentum-Gradient Alignment Update Policy for Stochastic Optimization16 Jun 2026
- 2026Memory-Efficient LLM Training with Dynamic Sparsity: From Stability to Practical Scaling30 May 2026
- 2026PowerStep: Memory-Efficient Adaptive Optimization via ℓp-Norm Steepest Descent11 May 2026
- 2026One Algorithm, Two Goals: Dual Scoring for Parameter and Data Selection in LLM Fine-Tuning7 May 2026
- 2026Subspace Optimization for Efficient Federated Learning under Heterogeneous Data28 Apr 2026
- 2026OASIS: Online Activation Subspace Learning for Memory-Efficient Training10 Apr 2026
- 2026Train Less, Infer Faster: Efficient Model Finetuning and Compression via Structured Sparsity9 Feb 2026
- 2026Emergent Low-Rank Training Dynamics in MLPs with Smooth Activations5 Feb 2026
12 newest of 35 All 35 (JSON)