Efficiency of Coordinate Descent Methods on Huge-Scale Optimization Problems 5: RCDM with Adaptive Lipschitz Estimates (RACDM) Has Expected Error at Most 8nR₁²(x₀)/(16 + 3k)Research Paper
Motivation
Random coordinate descent minimizes a smooth convex function by updating one coordinate at a time, chosen at random. Each iteration needs a single partial derivative instead of the full gradient, which is what makes the method usable on problems whose dimension is in the millions or billions ("huge-scale" problems). Nesterov's paper Efficiency of coordinate descent methods on huge-scale optimization problems (CORE Discussion Paper 2010/2; journal version SIAM J. Optim. 22 (2012) 341–362, doi:10.1137/100802001) gave the first global efficiency estimates for such methods and started a large literature on randomized block and coordinate methods.
The basic method RCDM of the paper takes, along coordinate , a step of length , where is a Lipschitz constant of the -th partial derivative along the -th coordinate. For simple functions these constants are known; for more complicated ones they are not, and a practical method has to estimate them on the fly. Section 6.1 of the paper analyses a backtracking strategy with restore for this purpose and proves that it keeps the efficiency of RCDM up to a constant factor. This mission formalizes that analysis, Theorem 7 of the paper.
Setting
Write and for the -th standard basis vector of , . The function is convex and differentiable, attains its minimum , and its partial derivatives are coordinate-wise Lipschitz with constants ((2.2) with one-dimensional blocks):
The weighted norm and its dual are and ((2.7) with ), and
measures the size of the initial level set, being the set of minimizers.
The random adaptive coordinate descent method RACDM (6.1) keeps estimates , initialised with given lower bounds . At iteration it
- draws uniformly from ;
- sets and, as long as , doubles and recomputes ;
- halves .
The loop test uses only the sign of one partial derivative and never a function value. With (the expectation over ) and the number of partial derivatives computed in iterations , Theorem 7 states three facts about this method.
Formalization targets
Goal: Theorem 7, item 2, (6.2)
This is the rate of RCDM, ((2.14)), with a larger constant.
Milestones
- (6.4) — started from an estimate , the doubling loop terminates and the accepted estimate satisfies .
- Theorem 7, item 1 — at the beginning of each iteration for every .
- One-step decrease (proof of Theorem 7, p. 19) — .
- Expected decrease (proof of Theorem 7, p. 19) — .
- Theorem 7, item 3, (6.3) — .
Significance
The result shows that the constants in RCDM need not be known: a method that starts from any lower bounds and adjusts them by doubling and halving converges at the same rate, and pays on average about two partial derivatives per iteration plus a logarithmic one-time overhead. This is what makes random coordinate descent applicable to functions whose coordinate smoothness is not available in closed form, and the doubling-with-restore pattern recurs in later adaptive and accelerated coordinate methods (the paper itself indicates the same technique for its accelerated method, (6.5)).
The theorem is proved in the paper; to our knowledge no machine-checked proof exists. The known-constant analogues for scalar coordinates are on Prove2Me in Bubeck's §6.4 development (the one-step and expected decreases are proved there; the rate is open). This mission adds a formal model of the adaptive method itself, including its doubling loop as a first-failure search, and the three statements of Theorem 7 on top of the published coordinate-descent definitions.
Difficulty
The analysis of RCDM rests on the one-step bound , which follows from the smoothness of along the coordinate line. For RACDM the step uses an estimate that may be smaller than , and then the same argument gives no decrease at all: a step that is too long can increase . The method never checks function values, so the decrease has to be extracted from the sign test alone, and it holds only because is convex along the line, not merely smooth. A second point is that the estimates are random and depend on the whole history, so the expected decrease must hold for every state the method can reach; this is what item 1 secures. The counting statement needs the exact number of loop iterations, so the loop has to be modelled as the least number of doublings after which the test fails, not as some number of doublings.
Formalization scope
The formalization works with scalar coordinates (, as the paper does in §6.1) on EuclideanSpace ℝ (Fin n), coordinates indexed . It reuses the published definition ConvexOptAlg_CoordDescent_Defs: IsCoordSmooth f g L is (2.2) with an explicit gradient map , wnorm L 1 and wnormDual L 1 are and , and rcdExpect L 0 k is the expectation over uniform draws as a finite weighted sum. The mission's own definition NesterovRCD.Adaptive.RACDM encodes the trial point, the number of doublings (a minimum over ), one iteration, the run along an explicit sequence of draws, and the count .
The following choices are made explicitly.
- Step 3 of (6.1) is printed "Set "; it is read as , as the paper's proof requires. The true constants never change and enter only hypotheses and bounds.
- is not computed: the goal takes any with for all in the level set and all minimizers . " nonempty" is a hypothesis; " bounded" follows from the existence of .
- Implicit hypotheses made explicit: ; and ; in the milestones, on the entry state.
- Convexity is assumed in the statements that need it (the decreases and the rate); (6.4) and items 1 and 3 hold without it and are stated without it.
- counts evaluations of at iteration with doublings, the count the paper's proof uses; counting as well would give .
A trivializing encoding is ruled out: the loop count is the least at which the sign test fails (an arbitrary admissible would change the method and make the count meaningless), the draws are uniform, and the rate is stated in the norm of the true constants , not of the estimates. Contributions welcome: proofs of the milestones, a one-dimensional co-coercivity lemma for convex smooth functions of one variable (reusable well beyond this mission), and the recursion argument from the expected decrease to the rate.
Selected references
- Yu. Nesterov, Efficiency of coordinate descent methods on huge-scale optimization problems, CORE Discussion Paper 2010/2, Université catholique de Louvain, 2010. Journal version: SIAM J. Optim. 22(2) (2012) 341–362. doi:10.1137/100802001
- Yu. Nesterov, Introductory Lectures on Convex Optimization: A Basic Course, Kluwer, 2004 (reference [6] of the paper; inequality (2.1.17)). doi:10.1007/978-1-4419-8853-9
- S. Bubeck, Convex Optimization: Algorithms and Complexity, Foundations and Trends in Machine Learning 8(3–4) (2015) 231–357, §6.4. arXiv:1405.4980