English

AdaFRUGAL: Adaptive Memory-Efficient Training with Dynamic Control

Machine Learning 2026-04-30 v2 Artificial Intelligence Computation and Language

Abstract

Training Large Language Models (LLMs) is highly memory-intensive due to optimizer state overhead. The FRUGAL framework mitigates this with gradient splitting, but its static hyperparameters -- the subspace ratio (ρ\rho) and update frequency (TT) -- require costly manual tuning, limiting adaptability. We present AdaFRUGAL, which automates this process by introducing two dynamic controls: (i) a linear decay for ρ\rho to progressively reduce memory, and (ii) a loss-aware schedule for TT to lower computational overhead. Experiments across large-scale pre-training (English C4, Vietnamese VietVault) and fine-tuning (GLUE) demonstrate that AdaFRUGAL achieves a compelling trade-off. It maintains competitive performance against AdamW and static FRUGAL while significantly reducing both GPU memory and training time, offering a more practical, autonomous solution for resource-constrained LLM training.

Keywords

Cite

@article{arxiv.2601.11568,
  title  = {AdaFRUGAL: Adaptive Memory-Efficient Training with Dynamic Control},
  author = {Quang-Hung Bui and Anh Son Ta},
  journal= {arXiv preprint arXiv:2601.11568},
  year   = {2026}
}

Comments

We have identified issues in the current version of the manuscript that may affect the validity of some results. We are withdrawing the paper to conduct further verification and improvements before resubmission

R2 v1 2026-07-01T09:08:05.271Z