A Universal Catalyst for First-Order Optimization
Hongzhou Lin, Julien Mairal, Zaid Harchaoui
Introduction
where is convex and has Lipschitz continuous derivatives with constant and is convex but may not be differentiable. The variable represents model parameters and the role of is to ensure that the estimated parameters fit some observed data. Specifically, is often a large sum of functions
Our goal is to accelerate gradient-based or first-order methods that are designed to solve (1), with a particular focus on large sums of functions (2). By “accelerating”, we mean generalizing a mechanism invented by Nesterov that improves the convergence rate of the gradient descent algorithm. More precisely, when , gradient descent steps produce iterates such that , where denotes the minimum value of . Furthermore, when the objective is strongly convex with constant , the rate of convergence becomes linear in . These rates were shown by Nesterov to be suboptimal for the class of first-order methods, and instead optimal rates— for the convex case and for the -strongly convex one—could be obtained by taking gradient steps at well-chosen points. Later, this acceleration technique was extended to deal with non-differentiable regularization functions .
For modern machine learning problems involving a large sum of functions, a recent effort has been devoted to developing fast incremental algorithms that can exploit the particular structure of (2). Unlike full gradient approaches which require computing and averaging gradients at every iteration, incremental techniques have a cost per-iteration that is independent of . The price to pay is the need to store a moderate amount of information regarding past iterates, but the benefit is significant in terms of computational complexity.
Our main achievement is a generic acceleration scheme that applies to a large class of optimization methods. By analogy with substances that increase chemical reaction rates, we call our approach a “catalyst”. A method may be accelerated if it has linear convergence rate for strongly convex problems. This is the case for full gradient and block coordinate descent methods , which already have well-known accelerated variants. More importantly, it also applies to incremental algorithms such as SAG , SAGA , Finito/MISO , SDCA , and SVRG . Whether or not these methods could be accelerated was an important open question. It was only known to be the case for dual coordinate ascent approaches such as SDCA or SDPC for strongly convex objectives. Our work provides a universal positive answer regardless of the strong convexity of the objective, which brings us to our second achievement.
Some approaches such as Finito/MISO, SDCA, or SVRG are only defined for strongly convex objectives. A classical trick to apply them to general convex functions is to add a small regularization . The drawback of this strategy is that it requires choosing in advance the parameter , which is related to the target accuracy. A consequence of our work is to automatically provide a direct support for non-strongly convex objectives, thus removing the need of selecting beforehand.
The approach Finito/MISO, which was proposed in and , is an incremental technique for solving smooth unconstrained -strongly convex problems when is larger than a constant (with in ). In addition to providing acceleration and support for non-strongly convex objectives, we also make the following specific contributions: we extend the method and its convergence proof to deal with the composite problem (1); we fix the method to remove the “big data condition” . The resulting algorithm can be interpreted as a variant of proximal SDCA with a different step size and a more practical optimality certificate—that is, checking the optimality condition does not require evaluating a dual objective. Our construction is indeed purely primal. Neither our proof of convergence nor the algorithm use duality, while SDCA is originally a dual ascent technique.
The catalyst acceleration can be interpreted as a variant of the proximal point algorithm , which is a central concept in convex optimization, underlying augmented Lagrangian approaches, and composite minimization schemes . The proximal point algorithm consists of solving (1) by minimizing a sequence of auxiliary problems involving a quadratic regularization term. In general, these auxiliary problems cannot be solved with perfect accuracy, and several notations of inexactness were proposed, including . The catalyst approach hinges upon (i) an acceleration technique for the proximal point algorithm originally introduced in the pioneer work ; (ii) a more practical inexactness criterion than those proposed in the past.Note that our inexact criterion was also studied, among others, in , but the analysis of led to the conjecture that this criterion was too weak to warrant acceleration. Our analysis refutes this conjecture. As a result, we are able to control the rate of convergence for approximately solving the auxiliary problems with an optimization method . In turn, we are also able to obtain the computational complexity of the global procedure for solving (1), which was not possible with previous analysis . When instantiated in different first-order optimization settings, our analysis yields systematic acceleration.
Beyond , several works have inspired this paper. In particular, accelerated SDCA is an instance of an inexact accelerated proximal point algorithm, even though this was not explicitly stated in . Their proof of convergence relies on different tools than ours. Specifically, we use the concept of estimate sequence from Nesterov , whereas the direct proof of , in the context of SDCA, does not extend to non-strongly convex objectives. Nevertheless, part of their analysis proves to be helpful to obtain our main results. Another useful methodological contribution was the convergence analysis of inexact proximal gradient methods of . Finally, similar ideas appear in the independent work . Their results overlap in part with ours, but both papers adopt different directions. Our analysis is for instance more general and provides support for non-strongly convex objectives. Another independent work with related results is , which introduce an accelerated method for the minimization of finite sums, which is not based on the proximal point algorithm.
The Catalyst Acceleration
We present here our generic acceleration scheme, which can operate on any first-order or gradient-based optimization algorithm with linear convergence rate for strongly convex objectives.
where denotes the minimum value of . The quantity controls the convergence rate: the larger is , the faster is convergence to . However, for a given algorithm , the quantity depends usually on the ratio , which is often called the condition number of .
Our approach can accelerate a wide range of first-order optimization algorithms, starting from classical gradient descent. It also applies to randomized algorithms such as SAG, SAGA, SDCA, SVRG and Finito/MISO, whose rates of convergence are given in expectation. Such methods should be contrasted with stochastic gradient methods , which minimize a different non-deterministic function. Acceleration of stochastic gradient methods is beyond the scope of this work.
We now highlight the mechanics of the catalyst algorithm, which is presented in Algorithm 1. It consists of replacing, at iteration , the original objective function by an auxiliary objective , close to up to a quadratic term:
where will be specified later and is obtained by an extrapolation step described in (6). Then, at iteration , the accelerated algorithm minimizes up to accuracy .
Substituting (4) to (1) has two consequences. On the one hand, minimizing (4) only provides an approximation of the solution of (1), unless ; on the other hand, the auxiliary objective enjoys a better condition number than the original objective , which makes it easier to minimize. For instance, when is the regular gradient descent algorithm with , has the rate of convergence (3) for minimizing with . However, owing to the additional quadratic term, can be minimized by with the rate (3) where . In practice, there exists an “optimal” choice for , which controls the time required by for solving the auxiliary problems (4), and the quality of approximation of by the functions . This choice will be driven by the convergence analysis in Sec. 3.1-3.3; see also Sec. C for special cases.
Similar to the classical gradient descent scheme of Nesterov , Algorithm 1 involves an extrapolation step (6). As a consequence, the solution of the auxiliary problem (5) at iteration is driven towards the extrapolated variable . As shown in , this step is in fact sufficient to reduce the number of iterations of Algorithm 1 to solve (1) when —that is, for running the exact accelerated proximal point algorithm.
Nevertheless, to control the total computational complexity of an accelerated algorithm , it is necessary to take into account the complexity of solving the auxiliary problems (5) using . This is where our approach differs from the classical proximal point algorithm of . Essentially, both algorithms are the same, but we use the weaker inexactness criterion , where the sequence is fixed beforehand, and only depends on the initial point. This subtle difference has important consequences: (i) in practice, this condition can often be checked by computing duality gaps; (ii) in theory, the methods we consider have linear convergence rates, which allows us to control the complexity of step (5), and then to provide the computational complexity of .
Convergence Analysis
In this section, we present the theoretical properties of Algorithm 1, for optimization methods with deterministic convergence rates of the form (3). When the rate is given as an expectation, a simple extension of our analysis described in Section 4 is needed. For space limitation reasons, we shall sketch the proof mechanics here, and defer the full proofs to Appendix B.
We first analyze the convergence rate of Algorithm 1 for solving problem 1, regardless of the complexity required to solve the subproblems (5). We start with the -strongly convex case.
Choose with and
Then, Algorithm 1 generates iterates such that
This theorem characterizes the linear convergence rate of Algorithm 1. It is worth noting that the choice of is left to the discretion of the user, but it can safely be set to in practice. The choice was made for convenience purposes since it leads to a simplified analysis, but larger values are also acceptable, both from theoretical and practical point of views. Following an advice from Nesterov[17, page 81] originally dedicated to his classical gradient descent algorithm, we may for instance recommend choosing such that .
The choice of the sequence is also subject to discussion since the quantity is unknown beforehand. Nevertheless, an upper bound may be used instead, which will only affects the corresponding constant in (7). Such upper bounds can typically be obtained by computing a duality gap at , or by using additional knowledge about the objective. For instance, when is non-negative, we may simply choose .
The proof of convergence uses the concept of estimate sequence invented by Nesterov , and introduces an extension to deal with the errors . To control the accumulation of errors, we borrow the methodology of for inexact proximal gradient algorithms. Our construction yields a convergence result that encompasses both strongly convex and non-strongly convex cases. Note that estimate sequences were also used in , but, as noted by , the proof of only applies when using an extrapolation step (6) that involves the true minimizer of (5), which is unknown in practice. To obtain a rigorous convergence result like (7), a different approach was needed.
Theorem 3.1 is important, but it does not provide yet the global computational complexity of the full algorithm, which includes the number of iterations performed by for approximately solving the auxiliary problems (5). The next proposition characterizes the complexity of this inner-loop.
Under the assumptions of Theorem 3.1, let us consider a method generating iterates for minimizing the function with linear convergence rate of the form
This proposition is generic since the assumption (8) is relatively standard for gradient-based methods . It may now be used to obtain the global rate of convergence of an accelerated algorithm. By calling the objective function value obtained after performing iterations of the method , the true convergence rate of the accelerated algorithm is
As a result, algorithm has a global linear rate of convergence with parameter
where typically depends on (the greater, the faster is ). Consequently, will be chosen to maximize the ratio . Note that for other algorithms that do not satisfy (8), additional analysis and possibly a different initialization may be necessary (see Appendix D for example).
2 Convergence Analysis for Convex but Non-Strongly Convex Objective Functions
We now state the convergence rate when the objective is not strongly convex, that is when .
When , choose and
Then, Algorithm 1 generates iterates such that
This theorem is the counter-part of Theorem 3.1 when . The choice of is left to the discretion of the user; it empirically seem to have very low influence on the global convergence speed, as long as it is chosen small enough (e.g., we use in practice). It shows that Algorithm 1 achieves the optimal rate of convergence of first-order methods, but it does not take into account the complexity of solving the subproblems (5). Therefore, we need the following proposition:
We can now draw up the global complexity of an accelerated algorithm when has a linear convergence rate (8) for -strongly convex objectives. To produce , is called at most times. Using the global iteration counter , we get
If is a first-order method, this rate is near-optimal, up to a logarithmic factor, when compared to the optimal rate , which may be the price to pay for using a generic acceleration scheme.
Acceleration in Practice
We show here how to accelerate existing algorithms and compare the convergence rates obtained before and after catalyst acceleration. For all the algorithms we consider, we study rates of convergence in terms of total number of iterations (in expectation, when necessary) to reach accuracy . We first show how to accelerate full gradient and randomized coordinate descent algorithms . Then, we discuss other approaches such as SAG , SAGA , or SVRG . Finally, we present a new proximal version of the incremental gradient approaches Finito/MISO , along with its accelerated version. Table 1 summarizes the acceleration obtained for the algorithms considered.
The convergence rate of an accelerated algorithm is driven by the parameter . In the strongly convex case, the best choice is the one that maximizes the ratio . As discussed in Appendix C, this rule also holds when (8) is given in expectation and in many cases where the constant is different than from (8). When , the choice of only affects the complexity by a multiplicative constant. A rule of thumb is to maximize the ratio (see Appendix C for more details).
After choosing , the global iteration-complexity is given by , where is an upper-bound on the number of iterations performed by per inner-loop, and is the upper-bound on the number of outer-loop iterations, following from Theorems 3.1-3.3. Note that for simplicity, we always consider that such that we may write simply as “” in the convergence rates.
1 Acceleration of Existing Algorithms
Table 1 presents convergence rates that are valid for proximal and non-proximal settings, since most methods we consider are able to deal with such non-differentiable penalties. The exception is SAG , for which proximal variants are not analyzed. The incremental method Finito/MISO has also been limited to non-proximal settings so far. In Section 4.2, we actually introduce the extension of MISO to composite minimization, and establish its theoretical convergence rates.
We now consider randomized incremental gradient methods, resp. SAG and SAGA . When , we focus on the “ill-conditioned” setting , where these methods have the complexity . Otherwise, their complexity becomes , which is independent of the condition number and seems theoretically optimal .
For these methods, the best choice for has the form , with for SAG, for SAGA. A similar formula, with a constant in place of , holds for SVRG; we omit it here for brevity. SDCA and Finito/MISO are actually related to incremental gradient methods, and the choice for has a similar form with .
2 Proximal MISO and its Acceleration
Finito/MISO was proposed in and for solving the problem (1) when and when is a sum of -strongly convex functions as in (2), which are also differentiable with -Lipschitz derivatives. The algorithm maintains a list of quadratic lower bounds—say at iteration —of the functions and randomly updates one of them at each iteration by using strong-convexity inequalities. The current iterate is then obtained by minimizing the lower-bound of the objective
Interestingly, since is a lower-bound of we also have , and thus the quantity can be used as an optimality certificate that upper-bounds . Furthermore, this certificate was shown to converge to zero with a rate similar to SAG/SDCA/SVRG/SAGA under the condition . In this section, we show how to remove this condition and how to provide support to non-differentiable functions whose proximal operator can be easily computed. We shall briefly sketch the main ideas, and we refer to Appendix D for a thorough presentation.
The first idea to deal with a nonsmooth regularizer is to change the definition of :
which was also proposed in without a convergence proof. Then, because the ’s are quadratic functions, the minimizer of can be obtained by computing the proximal operator of at a particular point. The second idea to remove the condition is to modify the update of the lower bounds . Assume that index is selected among at iteration , then
Whereas the original Finito/MISO uses , our new variant uses . The resulting algorithm turns out to be very close to variant “5” of proximal SDCA , which corresponds to using a different value for . The main difference between SDCA and MISO-Prox is that the latter does not use duality. It also provides a different (simpler) optimality certificate , which is guaranteed to converge linearly, as stated in the next theorem.
Let be obtained by MISO-Prox, then
Furthermore, we also have fast convergence of the certificate
The proof of convergence is given in Appendix D. Finally, we conclude this section by noting that MISO-Prox enjoys the catalyst acceleration, leading to the iteration-complexity presented in Table 1. Since the convergence rate (14) does not have exactly the same form as (8), Propositions 3.2 and 3.4 cannot be used and additional analysis, given in Appendix D, is needed. Practical forms of the algorithm are also presented there, along with discussions on how to initialize it.
Experiments
We use three datasets used in , namely real-sim, rcv1, and ocr, which are relatively large, with up to points for ocr and variables for rcv1. We consider three regimes: (no regularization), and , which leads significantly larger condition numbers than those used in other studies ( in ). We compare MISO, SAG, and SAGA with their default parameters, which are recommended by their theoretical analysis (step-sizes for SAG and for SAGA), and study several accelerated variants. The values of and and the sequences are those suggested in the previous sections, with in (10). Other implementation details are presented in Appendix E.
The restarting strategy for is key to achieve acceleration in practice. All of the methods we compare store gradients evaluated at previous iterates of the algorithm. We always use the gradients from the previous run of to initialize a new one. We detail in Appendix E the initialization for each method. Finally, we evaluated a heuristic that constrain to always perform at most iterations (one pass over the data); we call this variant AMISO2 for MISO whereas AMISO1 refers to the regular “vanilla” accelerated variant, and we also use this heuristic to accelerate SAG.
The results are reported in Table 1. We always obtain a huge speed-up for MISO, which suffers from numerical stability issues when the condition number is very large (for instance, for ocr). Here, not only does the catalyst algorithm accelerate MISO, but it also stabilizes it. Whereas MISO is slower than SAG and SAGA in this “small ” regime, AMISO2 is almost systematically the best performer. We are also able to accelerate SAG and SAGA in general, even though the improvement is less significant than for MISO. In particular, SAGA without acceleration proves to be the best method on ocr. One reason may be its ability to adapt to the unknown strong convexity parameter of the objective near the solution. When , we indeed obtain a regime where acceleration does not occur (see Sec. 4). Therefore, this experiment suggests that adaptivity to unknown strong convexity is of high interest for incremental optimization.
This work was supported by ANR (MACARON ANR-14-CE23-0003-01), MSR-Inria joint centre, CNRS-Mastodons program (Titan), and NYU Moore-Sloan Data Science Environment.
References
Appendix A Construction of the Approximate Estimate Sequence
The estimate sequence is a generic tool introduced by Nesterov for proving the convergence of accelerated gradient-based algorithms. We start by recalling the definition given in .
Estimate sequences are used for proving convergence rates thanks to the following lemma
If for some sequence we have
for an estimate sequence of , then
The rate of convergence of is thus directly related to the convergence rate of . Constructing estimate sequences is thus appealing, even though finding the most appropriate one is not trivial for the catalyst algorithm because of the approximate minimization of in (5). In a nutshell, the main steps of our convergence analysis are
define an “approximate” estimate sequence for corresponding to Algorithm 1—that is, finding a function that almost satisfies Definition A.1 up to the approximation errors made in (5) when minimizing , and control the way these errors sum up together.
extend Lemma A.2 to deal with the approximation errors to derive a generic convergence rate for the sequence .
This is also the strategy proposed by Güler in for his inexact accelerated proximal point algorithm, which essentially differs from ours in its stopping criterion. The estimate sequence we choose is also different and leads to a more rigorous convergence proof. Specifically, we prove in this section the following theorem:
where the ’s are defined in Algorithm 1. Then, the sequence satisfies
where is the minimum value of and
Then, the theorem will be used with the following lemma from to control the convergence rate of the sequence , whose definition follows the classical use of estimate sequences . This will provide us convergence rates both for the strongly convex and non-strongly convex cases.
If in the quantity defined in (17) satisfies , then the sequence from (15) satisfies
We may now move to the proof of the theorem.
The first step is to construct an estimate sequence is typically to find a sequence of lower bounds of . By calling the minimizer of , the following one is used in :
By strong convexity, , which is equivalent to
After developing the quadratic terms, we directly obtain (19). ∎
Unfortunately, the exact value is unknown in practice and the estimate sequence of yields in fact an algorithm where the definition of the anchor point involves the unknown quantity instead of the approximate solutions and as in (6), as also noted by others . To obtain a rigorous proof of convergence for Algorithm 1, it is thus necessary to refine the analysis of . To that effect, we construct below a sequence of functions that approximately satisfies the definition of estimate sequences. Essentially, we replace in (19) the quantity by to obtain an approximate lower bound, and control the error by using the condition . This leads us to the following construction:
;
where the value of , given in (17) will be explained later. Note that if one replaces by in the above construction, it is easy to show that would be exactly an estimate sequence for with the relation given in (15).
Before extending Lemma A.2 to deal with the approximate sequence and conclude the proof of the theorem, we need to characterize a few properties of the sequence . In particular, the functions are quadratic and admit a canonical form:
For all , can be written in the canonical form
where the sequences , , and are defined as follows
Differentiate twice the relations (23) gives us directly (20). Since minimizes , the optimality condition gives
and then we obtain (21). Finally, apply to (23), which yields
Using the expression of from (21), we have
It remains to plug this relation into (24), use once (20), and we obtain the formula (22) for . ∎
We may now start analyzing the errors to control how far is the sequence from an exact estimate sequence. For that, we need to understand the effect of replacing by in the lower bound (19). The following lemma will be useful for that purpose.
where is the minimum value of . Replacing by its definition (5) gives
We can now show that Algorithm 1 generates iterates that approximately satisfy the condition of Lemma A.2 from Nesterov .
Let be the estimate sequence constructed above. Then, Algorithm 1 generates iterates such that
where the sequence is defined by and
We proceed by induction. For , and . Assume now that . Then,
where the second inequality is due to (25). By Lemma A.6, we now have,
We now need to show that the choice of the sequences and will cancel all the terms involving . In other words, we want to show that
which will be sufficient to conclude that . The relation (27) can be obtained from the definition of in (6) and the form of given in (20). We have indeed from (6) that
Then, the quantity follows the same recursion as in (20). Moreover, we have
from the definition of in (17). We can then conclude by induction that for all and (27) is satisfied.
To prove (26), we assume that is chosen such that (26) is satisfied, and show that it is equivalent to defining as in (6). By lemma A.6,
As a result, using (26) by replacing by yields
and we obtain the original equivalent definition of (6). This concludes the proof. ∎
With this lemma in hand, we introduce the following proposition, which brings us almost to Theorem A.3, which we want to prove.
Let us consider the sequence defined in (15). Then, the sequence satisfies
where is a minimizer of and its minimum value.
By the definition of the function , we have
where the inequality comes from (25). Therefore, by using the definition of in Lemma A.8,
where the first equality uses the relation (28), the last inequality comes from the strong convexity relation , and the last equality uses the relation .
Dividing both sides by yields
To control the error term on the right and finish the proof of Theorem A.3, we are going to borrow some methodology used to analyze the convergence of inexact proximal gradient algorithms from , and use an extension of a lemma presented in to bound the value of . This lemma is presented below.
Assume that the nonnegative sequences and satisfy the following recursion for all :
where is an increasing sequence such that . Then,
The first part—that is, Eq. (31)—is exactly Lemma 1 from . The proof is in their appendix. Then, by calling the right-hand side of (31), we have that for all , . Furthermore is increasing and we have
and using the inequality , we have
We are now in shape to conclude the proof of Theorem A.3. We apply the previous lemma to (29):
Appendix B Proofs of the Main Theorems and Propositions
We simply use Theorem A.3 and specialize it to the choice of parameters. The initialization leads to a particularly simple form of the algorithm, where for all . Therefore, the sequence from Theorem A.3 is also simple. For all , we indeed have . To upper-bound the quantity from Theorem A.3, we now remark that and thus, by strong convexity of ,
Therefore, Theorem A.3 combined with the previous inequality gives us
Since is decreasing in $\sqrt{1-\rho}+\frac{\rho}{2}\geq\sqrt{1-\sqrt{q}}+\frac{\sqrt{q}}{2}$. Consequently,
B.2 Proof of Proposition 3.2
To control the number of calls of , we need to upper bound which is given by the following lemma:
Let and be generated by Algorithm 1. Remember that by definition of ,
which can be shown by using the respective definitions of and and manipulate the quadratic term resulting from the difference .
Plugging and in the previous relation yields
where the last inequality comes from the strong convexity inequality of
Moreover, from the inequality , we also have
Summing inequalities (33), (34) and (35) gives the desired result. ∎
Next, we need to upper-bound the term , which was also required in the convergence proof of the accelerated SDCA algorithm . We follow here their methodology.
Let us consider the iterates and produced by Algorithm 1, and define
which appears in Theorem 3.1 and which is such that . Then, for any ,
We follow here . By definition of , we have
where is defined in (6). The last inequality was due to the fact that . Indeed, the specific choice of in Theorem A.3 leads to for all . Note, however, that this relation is true regardless of the choice of :
where the last equality uses the relation from Algorithm 1. To conclude the lemma, we notice that by triangle inequality
We are now in shape to conclude the proof of Proposition 3.2.
By Proposition B.1 and lemma B.2, we have for all ,
Let be the sequence of using to solve with initialization . By assumption (8), we have
The number of iterations of to guarantee an accuracy of needs to satisfy
Let us denote the right-hand side. We remark that this upper bound holds for . We now consider the cases and .
When , . Note that , then . As a result,
When , we remark that . Then, by following similar steps as in the proof of Lemma B.2, we have
which is smaller than . Therefore, the previous steps from the case apply and . Thus, for any ,
B.3 Proof of Theorem 3.3.
We will again Theorem A.3 and specialize it to the choice of parameters. To apply it, the following Lemma will be useful to control the growth of .
Let be the sequence defined in (15) where is produced by Algorithm 1 with and . Then, we have the following bounds for all ,
Note that by definition of , we have for all ,
With the choice of , the quantity defined in (17) is equal to . By Lemma A.4, we have for all and thus for all (it is also easy to check numerically that this is also true for since ). We now have all we need to conclude the lemma:
With this lemma in hand, we may now proceed and apply Theorem A.3. We have remarked in the proof of the previous lemma that . Then,
where the last inequality uses Lemma B.3 to upper-bound the ratio . Moreover,
The last inequality uses .
B.4 Proof of Proposition 3.4
When , we remark that Proposition B.1 still holds but Lemma B.2 does not. The main difficulty is thus to find another way to control the quantity .
Since is bounded by Theorem 3.3, we may use the bounded level set assumptions to ensure that there exists such that for any where is a minimizer of . We can now follow similar steps as in the proof of Lemma B.2, and show that
Since , is strongly convex, then using the same argument as in the strongly convex case, the number of calls for is given by
The right hand side is upper-bounded by . Plugging this relation into (38) gives the desired result.
Appendix C Derivation of Global Convergence Rates
We give here a generic “template” for computing the optimal choice of to accelerate a given algorithm , and therefore compute the rate of convergence of the accelerated algorithm .
We assume here that is a randomized first-order optimization algorithm, i.e. the iterates generated by are a sequence of random variables; specialization to a deterministic algorithm is straightforward. Also, for the sake of simplicity, we shall use simple notations to denote the stopping time to reach accuracy . Definition and notation using filtrations, -algebras, etc. are unnecessary for our purpose here where the quantity of interest has a clear interpretation.
Assume that algorithm enjoys a linear rate of convergence, in expectation. There exists constants and such that the sequence of iterates for minimizing a strongly-convex objective satisfies
Define the random variable (stopping time) corresponding to the minimum number of iterations to guarantee an accuracy in the course of running
Then, an upper bound on the expectation is provided by the following lemma.
Let be an optimization method with the expected rate of convergence (39). Then,
where we have dropped the dependency in to simplify the notation.
We abbreviate by . Set
As simple calculation shows that for any , and then
Note that the previous lemma mirrors Eq. (36-37) in the proof of Prop. 3.1 in Appendix B. For all optimization methods of interest, the rate is independent of and varies with the parameter . We may now compute the iteration-complexity (in expectation) of the accelerated algorithm —that is, for a given , the expected total number of iterations performed by the method . Let us now fix . Calculating the iteration-complexity decomposes into three steps:
Find that maximizes the ratio for algorithm when is -strongly convex. In the non-strongly convex case, we suggest maximizing instead the ratio . Note that the choice of is less critical for non-strongly convex problems since it only affects multiplicative constants in the global convergence rate.
Compute the upper-bound of the number of outer iterations using Theorem 3.1 (for the strongly convex case), or Theorem 3.3 (for the non-strongly convex case), by replacing by the optimal value found in step 1.
Compute the upper-bound of the expected number of inner iterations
by replacing the appropriate quantities in Eq. 41 for algorithm ; for that purpose, the proofs of Propositions 3.2 of 3.4 my be used to upper-bound the ratio , or another dedicated analysis for may be required if the constant does not have the required form in (8).
Then, the iteration-complexity (in expectation) denoted Comp. is given by
Appendix D A Proximal MISO/Finito Algorithm
In this section, we present the algorithm MISO/Finito, and show how to extend it in two ways. First, we propose a proximal version to deal with composite optimization problems, and we analyze its rate of convergence. Second, we show how to remove a large sample condition , which was necessary for the convergence of the algorithm. The resulting algorithm is a variant of proximal SDCA with a different stepsize and a stopping criterion that does not use duality.
MISO/Finito was proposed in and for solving the following smooth unconstrained convex minimization problem
where each is differentiable with -Lipschitz continuous derivatives and -strongly convex. At iteration , the algorithm updates a list of lower bounds of the functions , by randomly picking up one index among and performing the following update
which is a lower bound of because of the -strong convexity of . Equivalently, one may perform the following updates
and all functions have the form
where is a constant. Then, MISO/Finito performs the following minimization to produce the iterate :
In many machine learning problems, it is worth remarking that each function has the specific form . In such cases, the vectors can be obtained by storing only scalars.Note that even though we call this algorithm MISO (or Finito), it was called MISO in , whereas “MISO” was originally referring to an incremental majorization-minimization procedure that uses upper bounds of the functions instead of lower bounds, which is appropriate for non-convex optimization problems. The main convergence result of is that the procedure above converges with a linear rate of convergence of the form (3), with (also refined in in ), when the large sample size constraint is satisfied.
Removing this condition and extending MISO to the composite optimization problem (1) is the purpose of the next section.
D.2 Proximal MISO
We now consider the composite optimization problem below,
where the functions are differentiable with -Lipschitz derivatives and -strongly convex. As in typical composite optimization problems, is convex but not necessarily differentiable. We assume that the proximal operator of can be computed easily. The algorithm needs to be initialized with some lower bounds for the functions :
Then, we remark that under the large sample size condition , we have and the update of the quantities in (45) is the same as in the original MISO/Finito algorithm. As we will see in the convergence analysis, the choice of ensures convergence of the algorithm even in the small sample size regime .
The algorithm MISO-Prox is almost identical to variant of proximal SDCA , which performs the same updates with instead of . It is however not clear that MISO-Prox actually performs dual ascent steps in the sense of SDCA since the proof of convergence of SDCA cannot be directly modified to use the stepsize of proximal MISO and furthermore, the convergence proof of MISO-Prox does not use the concept of duality. Another difference lies in the optimality certificate of the algorithms. Whereas Proximal-SDCA provides a certificate in terms of linear convergence of a duality gap based on Fenchel duality, Proximal-SDCA ensures linear convergence of a gap that relies on strong convexity but not on the Fenchel dual (at least explicitly).
Similar to the original MISO algorithm, Proximal MISO maintains a list of lower bounds of the functions , which are updated in the following fashion
Then, the following function is a lower bound of the objective :
and the update (45) can be shown to exactly minimize . As a lower bound of , we have that and thus
The quantity can then be interpreted as an optimality gap, and the analysis below will show that it converges linearly to zero. In practice, it also provides a convenient stopping criterion, which yields Algorithm 3.
To explain the stopping criterion in Algorithm 3, we remark that the functions are quadratic and can be written
where the ’s are some constants and . Equation (48) shows how to update recursively these constants , and finally
which justifies the stopping criterion. Since computing requires scanning all the data points, the criterion is only computed every iterations.
The convergence of MISO-Prox is guaranteed by Theorem 4.1 from the main part of paper. Before we prove this theorem, we note that this rate is slightly better than the one proven in MISO , which converges as . We start by recalling a classical lemma that provides useful inequalities. Its proof may be found in .
To start the proof, we need a sequence of upper and lower bounds involving the functions and . The first one is given in the next lemma
where the definition of is given in (46). The first inequality uses Lemma D.1, and the last one uses the inequality . From this inequality, we can obtain (50) by simply using . ∎
Next, we prove the following lemma to compare and .
Remember that the functions are quadratic and have the form (49), that is defined in (47), and that minimizes . Then, there exists a constant such that
Then, we are able to control the value of in the next lemma.
Using Lemma D.3 with and yields
Moreover is the minimum of which is -strongly convex. Thus,
Adding the two previous inequalities gives the first inequality below
and the last one comes from the basic inequality . ∎
We have now all the inequalities in hand to prove Theorem 4.1.
Choose that maximizes the above quadratic function, i.e.
Then, we start introducing expected values. By construction
After taking expectation, we obtain the relation
On the one hand, since , we have
This is true for any , as a result
On the other hand, since , then
which gives us the relation (14) of the theorem. We conclude giving the choice of . We choose it to maximize the rate of convergence, which turns to maximize . This is a quadratic function, which is maximized at . However, by definition . Therefore, the optimal choice of is given by
When , we have and .
When , we have and .
Therefore, , which concludes the first part of the theorem.
To prove the second part, we use (58) and (14), which gives
D.3 Accelerating MISO-Prox
The convergence rate of MISO (or also SDCA) requires a special handling since it does not satisfy exactly the condition (8) from Proposition 3.2. The rate of convergence is linear, but with a constant proportional to instead of for many classical gradient-based approaches.
To achieve acceleration, we show in this section how to obtain similar guarantees as Proposition 3.2 and 3.4—that is, how to solve efficiently the subproblems (5). This essentially requires the right initialization each time MISO-Prox is called. By initialization, we mean initializing the variables .
Assume that MISO-Prox is used to obtain from Algorithm 1 with , and that one wishes to use MISO-Prox again on to compute . Then, let us call the lower-bound of produced by MISO-Prox when computing such that
Note that we do not index these quantities with or for the sake of simplicity. The convergence of MISO-Prox may ensure that not only do we have , but in fact we have the stronger condition . Remember now that
and that is a lower-bound of . Then, we may set for all in
which is equivalent to initializing the new instance of MISO-Prox with
and by choosing appropriate quantities . Then, the following function is a lower bound of
and the new instance of MISO-Prox to minimize and compute will produce iterates, whose first point, which we call , minimizes . This leads to the relation
where we use the notation and as in Algorithm 2.
Then, it remains to show that the quantity is upper bounded in a similar fashion as in Propositions 3.2 and 3.4 to obtain a similar result for MISO-Prox and control the number of inner-iterations. This is indeed the case, as stated in the next lemma.
When initializing MISO-Prox as described above, we have
where the last inequality is using a simple relation . As a result,
We remark that this bound is half of the bound shown in (32). Hence, a similar argument gives the bound on the number of inner iterations. We may finally compute the iteration-complexity of accelerated MISO-Prox.
When is -strongly convex, the accelerated MISO-Prox algorithm achieves the accuracy with an expected number of iteration upper bounded by
When , there is no acceleration. The optimal value for is zero, and we may use Theorem 4.1 and Lemma C.1 to obtain the complexity
When , there is an acceleration, with . Let us compute the global complexity using the “template” presented in Appendix C. The number of outer iteration is given by
At each inner iteration, we initialize with the value described above, and we use Lemma D.5:
With Miso-Prox, with have , thus the expected number of inner iteration is given by Lemma C.1:
To conclude, the complexity of the accelerated algorithm is given by
Appendix E Implementation Details of Experiments
In the experimental section, we compare the performance with and without acceleration for three algorithms SAG, SAGA and MISO-Prox on -logistic regression problem. In this part, we clarify some details about the implementation of the experiments.
Firstly, we normalize the observed data before running the regression. Then we apply Catalyst using parameters according to the theoretical settings. Standard analysis of the logistic function shows that the Lipschitz gradient parameter is and strongly convex parameter when there is no regularization. Adding properly a term generates the strongly-convex regimes. Several parameters need to be fixed at the beginning stage. The parameter is set to its optimal value suggested by theory, which only depends on , and . More precisely, writes as , with for SAG, for SAGA and for MISO-Prox. The parameter is initialized as the positive solution of where . Furthermore, since the objective function is always positive, can be upper bounded by which allow us to set the in the strongly convex case and in the non-strongly convex case. Finally, we set the free parameter in the expression of as follows. We simply set in the strongly convex case and in the non strongly convex case.
To solve the subproblem at each iteration, the step-sizes parameter for SAG, SAGA and MISO are set to the values suggested by theory, which only depend on , and . All of the methods we compare store gradients evaluated at previous iterates of the algorithm. For MISO, the convergence analysis in Appendix D leads to the initialization that moves closer to and further away from . We found that using this initial point for SAGA was giving slightly better results than .