AdLoCo claims faster distributed LLM training with fewer synchronization messages by adding adaptive batching, parallel model instances, and a gradient-accumulation fallback to DiLoCo.
Adaptive Sampling Strategies for Stochastic Optimization
1 Pith paper cite this work. Polarity classification is still indexing.
abstract
In this paper, we propose a stochastic optimization method that adaptively controls the sample size used in the computation of gradient approximations. Unlike other variance reduction techniques that either require additional storage or the regular computation of full gradients, the proposed method reduces variance by increasing the sample size as needed. The decision to increase the sample size is governed by an inner product test that ensures that search directions are descent directions with high probability. We show that the inner product test improves upon the well known norm test, and can be used as a basis for an algorithm that is globally convergent on nonconvex functions and enjoys a global linear rate of convergence on strongly convex functions. Numerical experiments on logistic regression problems illustrate the performance of the algorithm.
fields
cs.LG 1years
2025 1verdicts
UNVERDICTED 1representative citing papers
citing papers explorer
-
AdLoCo: adaptive batching significantly improves communications efficiency and convergence for Large Language Models
AdLoCo claims faster distributed LLM training with fewer synchronization messages by adding adaptive batching, parallel model instances, and a gradient-accumulation fallback to DiLoCo.