stat.MLFeb 13, 2026

AdaGrad-Diff: a Corrected Version of the Adaptive Gradient Algorithm for Composite Optimization

Authors: Matia Bojovic, Saverio Salzo, Massimiliano Pontil

Organizations: Computational Statistics and Machine Learning, Istituto Italiano di Tecnologia, Genoa, Italy · Department of Mathematics, University of Genoa, Genoa, Italy · DIAG, Sapienza University of Rome, Rome, Italy · Department of Computer Science, University College London, London, UK

Abstract

Vanilla gradient methods are often highly sensitive to the choice of stepsize, which typically requires manual tuning. Adaptive methods alleviate this issue, with AdaGrad being a prominent example. In this work, we first show that, in composite convex optimization, AdaGrad may fail to exploit the Hölder-smoothness of the differentiable term, as its stepsizes may decrease too rapidly, leading to convergence rates arbitrarily close to the nonsmooth O(n−1/2)\mathcal O(n^{-1/2}) rate. Motivated by this limitation, we study AdaGrad-Diff, a block-wise extension of the UniSgd method considered by \cite{rodomanov2024universality}, which accumulates successive gradient differences rather than gradients themselves. In the deterministic Lipschitz-smooth setting, we prove summability of squared gradient differences without assuming bounded iterates, yielding weak convergence to a minimizer in Hilbert spaces. Moreover, we show that, under νν-Hölder-smoothness, stepsizes decay no faster than n−(1−ν)/2n^{-(1-ν)/2}, thereby avoiding AdaGrad's failure mechanism and recovering the standard O(n−(1+ν)/2)\mathcal O(n^{-(1+ν)/2}) rate. The same approach recovers known stochastic convergence rates under a unified condition covering bounded variance and interpolation. Numerical experiments demonstrate that the proposed method is more robust than AdaGrad in several practically relevant settings.

Figures & tables

Explore similar work

CardsList
  1. Revisiting AdaGrad in Stochastic Convex Optimization: Last Iterates, High Probability, and Lower Bounds

    Sep 26, 2026Weiming Ou, Xiao WangConvergence AnalysisAdaptive Gradient Methods

  2. Adaptive Federated Optimization Across Smooth and Nonsmooth Regimes

    Oct 29, 2026Zimeng Wang, Ali Dadras, Alp YurtseverFederated LearningAdaptive Gradient Methods