SAGA: A Fast Incremental Gradient Method With Support for Non-Strongly Convex Composite Objectives
Aaron DefazioFrancis BachSimon Lacoste-Julien
Introduces SAGA, an incremental gradient method that delivers faster convergence rates than SAG and SVRG while natively supporting composite regularizers on both strongly and non-strongly convex objectives.
Modern large-scale machine learning and data analysis tasks frequently involve optimizing functions defined over massive collections of data points. Standard approaches often struggle to balance computational speed and versatility: traditional gradient descent methods process all data points at each step, ensuring steady progress but incurring substantial computational expense, whereas conventional stochastic methods evaluate one data point at a time but introduce random noise that prevents them from converging quickly without carefully decaying step sizes. Recent variance-reduced incremental algorithms have sought to combine the speed of stochastic updates with the steady convergence of batch methods, yet existing options often suffer from restricted theoretical support, inability to handle non-smooth regularizers, or excessive tuning requirements.
The article introduces and analyzes SAGA, a novel incremental gradient optimization algorithm designed to deliver fast, reliable convergence across a broad spectrum of convex machine learning problems. The authors set out to rigorously prove its theoretical convergence rates—including settings with non-smooth regularizers and problems lacking strong convexity—and to demonstrate its practical computational performance against leading alternatives.
The research combines rigorous mathematical analysis with empirical benchmarking. The theoretical evaluation establishes convergence bounds by analyzing the expected step-by-step progress using a dedicated mathematical tracking function. To substantiate the theory, the authors implemented SAGA alongside competing algorithms—such as SAG, SVRG, SDCA, and Finito—and evaluated their optimization performance on standard benchmark datasets (MNIST, COVTYPE, IJCNN1, and MILLIONSONG) spanning binary classification and least-squares regression under both smooth and non-smooth regularization.
The evaluation yielded several central findings. First, SAGA matches or improves upon theoretical convergence rates of prior methods, establishing linear convergence for strongly convex problems while natively supporting non-smooth composite regularizers via standard proximal operations. Second, SAGA directly handles non-strongly convex objectives without requiring artificial regularization, achieving a standard convergence rate of order 1/k while automatically adapting to any underlying strong convexity. Third, by utilizing an unbiased gradient update direction, SAGA avoids the complex biases of earlier algorithms like SAG and eliminates the inner-loop tuning parameters required by methods such as SVRG. Fourth, in empirical benchmarks, SAGA demonstrates optimization efficiency comparable to top-performing methods like SDCA and Finito while requiring only a single tunable step-size parameter, avoiding the two- to three-fold higher gradient evaluation overhead incurred by SVRG.
These findings indicate that SAGA offers a versatile and mathematically sound optimization framework that reduces operational complexity. By requiring only a single parameter to be tuned and supporting diverse regularizers directly, SAGA reduces the engineering time and compute costs typically spent searching over algorithm hyperparameters. It also unifies several previously fragmented algorithmic approaches into a clear conceptual spectrum between biased storage-heavy methods and parameter-heavy recalculation methods.
For practical implementation, practitioners should deploy SAGA when solving large-scale linear classification and regression problems, particularly when non-smooth regularizers like L1 penalties are required. For models where gradients cannot be represented compactly as simple scalars, practitioners should weigh the trade-off between SAGA’s memory footprint and SVRG’s extra computational passes. When deploying SAGA on sparse data, teams should use just-in-time updating heuristics to maintain linear computational efficiency.
The primary practical constraint of SAGA is its memory requirement: the algorithm maintains a table of historical gradient information across all data points, which can become costly for high-dimensional models such as deep neural networks where compact gradient representations are unavailable. Overall, confidence in the reported results is high for convex and composite convex empirical risk minimization problems, supported by complete mathematical proofs and consistent multi-dataset empirical performance.
- Paper: Accelerating Stochastic Gradient Descent using Predictive Variance Reduction, Rie Johnson et al. (2013). This paper introduces the stochastic variance reduced gradient (SVRG) method, establishing foundational variance-reduction mechanics that SAGA directly analyzes, builds upon, and improves.
- Paper: Adaptive Subgradient Methods for Online Learning and Stochastic Optimization, John Duchi et al. (2011). This work establishes proximal updates and adaptive subgradient principles in stochastic optimization, providing critical theoretical groundwork for handling composite objective functions.
- Paper: HOGWILD!: A Lock-Free Approach to Parallelizing Stochastic Gradient Descent, Feng Niu et al. (2011). This paper analyzes stochastic gradient convergence and sparsity assumptions, providing essential foundational concepts for modern incremental first-order methods.
- Paper: Online Convex Programming and Generalized Infinitesimal Gradient Ascent, Martin A. Zinkevich (2003). This seminal paper introduces online convex programming and projected gradient methods, which form the mathematical basis for analyzing iterative convex optimization algorithms.
- Paper: Optimization Methods for Large-Scale Machine Learning, Léon Bottou et al. (2016). This survey provides a comprehensive synthesis of large-scale machine learning optimization, situating variance-reduced incremental gradient methods like SAGA alongside modern stochastic algorithms.
- Paper: An overview of gradient descent optimization algorithms, Sebastian Ruder (2016). This overview contextualizes first-order gradient methods and advanced optimizers across machine learning workflows, contrasting variance-reduction and adaptive learning rate strategies.
- Paper: CVXPY: A Python-Embedded Modeling Language for Convex Optimization, Steven Diamond et al. (2016). This work develops a high-level modeling framework for specifying and executing composite and non-strongly convex optimization problems suited for modern first-order solvers.
