Stochastic Dual Coordinate Ascent Methods for Regularized Loss Minimization 1: With L-Lipschitz Losses, SDCA Reaches Expected Duality Gap ε after ⌈n log(λn/(2L²))⌉ + n + 20L²/(λε) IterationsResearch Paper
Motivation
Regularized loss minimization over linear predictors is the training problem behind support vector machines, logistic regression, ridge regression and their many variants. Given examples , one minimizes an average of convex losses of the predictions plus a quadratic penalty. Dual coordinate ascent (DCA) solves the Fenchel dual of this problem by optimizing one dual variable at a time; it has been the workhorse of linear SVM solvers such as LIBLINEAR (Hsieh et al., ICML 2008). Before Shalev-Shwartz and Zhang's work, the available analyses of DCA either gave linear rates with constants depending on unknown problem quantities or did not control the duality gap, which is the quantity a practitioner can actually monitor as a stopping criterion.
Shalev-Shwartz and Zhang (2013) analysed stochastic dual coordinate ascent (SDCA), where the coordinate is picked uniformly at random, and proved explicit bounds on the expected duality gap. For Lipschitz losses (hinge loss, absolute loss) the bound is iterations, which matches stochastic gradient descent when the accuracy is low and improves on it when is small compared with . This mission formalizes that result, Theorem 1 of the paper, together with the lemmas of its proof.
Timeline. Hsieh, Chang, Lin, Keerthi and Sundararajan (2008) proposed dual coordinate descent for linear SVMs and proved linear convergence with an unspecified constant. Shalev-Shwartz and Zhang (arXiv 2012, JMLR 2013) gave the first explicit duality-gap rates for SDCA, for Lipschitz and for smooth losses. Proximal and accelerated variants followed (Shalev-Shwartz and Zhang, Math. Program. 2016).
Setting
Fix examples (Euclidean norm), convex losses and a regularization parameter . The primal problem (1) is to minimize
The convex conjugate of a scalar function is . The dual problem (2) is to maximize, over ,
Weak duality gives for all , so the duality gap bounds the primal sub-optimality of .
A loss is -Lipschitz if for all . The paper's standing assumptions are , and .
Procedure SDCA starts from . At iteration it picks a coordinate uniformly at random, chooses maximizing
and sets and . The averaging option outputs and . Let maximize and write for the dual sub-optimality.
Formalization targets
Goal: Theorem 1 (p. 5)
For convex -Lipschitz losses under the standing assumptions, if and
then
The constants and are the paper's.
Milestones
In the order the proof uses them: the one-coordinate increase (10); the identity ; Lemma 1 (expected dual increase at least times the gap minus ); Lemma 2 (, ); Lemma 3 ( outside ); Lemma 4 (); the recursion after (13); the rate (14), ; and the averaged duality-gap bound of p. 16.
Significance
Theorem 1 makes the duality gap of SDCA a certified stopping criterion with an explicit iteration count. Because the gap upper-bounds the primal sub-optimality, it also gives a primal guarantee for the averaged output . It covers non-smooth losses, the hinge loss in particular, where linear rates are not available. The analysis pattern, a lower bound on the expected dual increase by the duality gap (Lemma 1) followed by a recursion for the dual sub-optimality, was reused in the analyses of proximal, accelerated and mini-batch dual methods.
The result is proved in the paper; to the best of current knowledge it has no machine-checked proof. The mission produces a formal account of the primal–dual pair of regularized loss minimization with extended-real conjugates, a formal SDCA run with uniformly random coordinates, and the complete chain of lemmas. Lemma 1 is stated with a general strong-convexity modulus , as in the paper, so it serves the smooth-loss analysis as well.
Difficulty
The obvious first idea is to treat SDCA as stochastic coordinate ascent on a concave function and import a standard rate. That fails: is neither smooth nor strongly concave for Lipschitz losses, and its domain is a box, so the per-coordinate progress cannot be bounded by a gradient norm. The argument instead compares the exact coordinate maximizer with an explicit interpolation towards a sub-gradient, which needs the conjugate's value at a sub-gradient (Fenchel–Young equality) and its infinite values outside .
The rate (14) is not a plain geometric recursion. It needs a burn-in phase and then an induction with a step size that changes with , and the duality gap of the output has to be recovered from the dual progress by Jensen's inequality over the averaged iterates. Everything happens in expectation over the coordinate sequence, while the dual values have to be kept finite along every run.
Formalization scope
Lean conventions:
- is
EuclideanSpace ℝ (Fin d), examples and losses are indexed byFin n, and isinner ℝ w x. - The conjugate is an
ERealsupremum, and isEReal-valued. Real values of are used only where the same statement asserts that they are finite. - The coordinate step is a free map
Δ : (Fin n → ℝ) → Fin n → ℝconstrained by an arg-max predicate (IsSDCAStep). Any maximizer is admissible, and one exists for convex real losses. - The run is a function of the coordinate sequence
js : Fin T → Fin n, started at , with . - Expectations are uniform averages over coordinate sequences (
SAGA.Convex.expectIdx, a published definition). - is any maximizer of , and any minimizer of .
- The constant of the proof is pinned to its bound . Theorem 1's burn-in is exactly the value of obtained with and .
- Lemmas 1 and 4 are stated for a fixed dual-feasible state. The paper's history-averaged form follows from this.
A real-valued conjugate, which returns where the supremum is infinite, or an unconstrained step map would make the statements false or vacuous. The EReal conjugate and the arg-max predicate rule both out.
Infrastructure needed: Fenchel–Young for scalar convex functions, the existence of sub-gradients of convex real functions and their bound by the Lipschitz constant, Jensen's inequality for the convex and the concave , and the algebra of uniform averages over . The conjugate and sub-differential definitions and the duality-gap identity can be reused beyond this mission. Proofs of any milestone are welcome, as are lemmas on EReal sums that keep the finiteness bookkeeping short.
Selected references
- S. Shalev-Shwartz and T. Zhang, Stochastic Dual Coordinate Ascent Methods for Regularized Loss Minimization, arXiv:1209.1873v2, 2013; Journal of Machine Learning Research 14:567–599, 2013. https://arxiv.org/abs/1209.1873
- C.-J. Hsieh, K.-W. Chang, C.-J. Lin, S. S. Keerthi and S. Sundararajan, A Dual Coordinate Descent Method for Large-scale Linear SVM, ICML 2008. https://doi.org/10.1145/1390156.1390208
- S. Shalev-Shwartz and T. Zhang, Accelerated Proximal Stochastic Dual Coordinate Ascent for Regularized Loss Minimization, Mathematical Programming 155:105–145, 2016. https://doi.org/10.1007/s10107-014-0839-0