English

A Convergence Analysis of Adaptive Optimizers under Floating-point Quantization

Machine Learning 2026-03-03 v2 Artificial Intelligence Machine Learning

Abstract

The rapid scaling of large language models (LLMs) has made low-precision training essential for reducing memory, improving efficiency, and enabling larger models and datasets. Existing convergence theories for adaptive optimizers, however, assume all components are exact and neglect hardware-aware quantization, leaving open the question of why low-precision training remains effective. We introduce the first theoretical framework for analyzing the convergence of adaptive optimizers, including Adam and Muon, under floating-point quantization of gradients, weights, and optimizer states (e.g., moment estimates). Within this framework, we derive convergence rates on smooth non-convex objectives under standard stochastic gradient assumptions, explicitly characterizing how quantization errors from different components affect convergence. We show that both algorithms retain rates close to their full-precision counterparts provided mantissa length scales only logarithmically with the number of iterations. Our analysis further reveals that Adam is highly sensitive to weights and second-moment quantization due to its reliance on β21\beta_2 \to 1, while Muon requires weaker error control and is thus potentially more robust. These results narrow the gap between empirical success and theoretical understanding of low-precision training methods. Numerical experiments on synthetic and real-world data corroborate our theory.

Keywords

Cite

@article{arxiv.2510.21314,
  title  = {A Convergence Analysis of Adaptive Optimizers under Floating-point Quantization},
  author = {Xuan Tang and Jichu Li and Difan Zou},
  journal= {arXiv preprint arXiv:2510.21314},
  year   = {2026}
}

Comments

68 pages, 13 figures, ICLR 2026

R2 v1 2026-07-01T07:03:41.700Z