Papers for

data-intensive system designers

Papers whose findings have a practical use for this group, as judged from the abstract. Open a paper to read what it means in practice.

Adam optimization convergence slows with heavy noise and benefits from larger batches

On the Convergence of Vanilla Adam under Heavy-Tailed Noise and Index Sampling

Abstract: Understanding how stochastic gradient noise affects the convergence of Adam is an important problem in stochastic optimization, particularly when the noise exhibits heavy-tailed behavior. In this paper, we investigate the convergence of vanilla Adam in both the expected risk and the empirical risk minimization, focusing on the effects of heavy-tailed noise and its relationship to index sampling. For the expected risk minimization, we establish convergence guarantees under Hölder smoothness and bounded $\mathfrak{p}$-th moment assumptions, allowing stochastic gradient noise with infinite variance. Our analysis provides explicit upper bounds on the expected squared gradient norm and shows that the iteration-dependent convergence term decays more slowly as $\mathfrak{p}$ decreases. For the empirical risk minimization, we establish asymptotic and non-asymptotic convergence guarantees and demonstrate that heavy-tailed population noise can deteriorate convergence bounds through dataset-dependent constants as the dataset size increases. Furthermore, under Lipschitz smoothness, we formulate multistage Adam with practical learning rate schedules and demonstrate that increasing batch sizes can reduce index-sampling noise and improve convergence guarantees and SFO complexity. Numerical experiments support our theoretical findings on the effects of heavy-tailed noise and the computational benefits of increasing batch sizes.

Mon 2 NovMachine Learning
The gist
Training neural networks with the Adam optimizer can be tricky when the updates have heavy-tailed noise, which means the randomness in training is extreme and unpredictable. The authors studied how this kind of noise affects Adam's ability to find a good solution over time. They showed that when noise is more extreme, Adam's progress slows down and can depend heavily on how data samples are chosen. They also found that using bigger batches of data during training helps reduce this noisy effect and improves training stability.
Open → 2611.03201v1