Don't Jump Through Hoops and Remove Those Loops: SVRG and Katyusha are Better Without the Outer Loop

Dmitry Kovalev, Samuel Horvath, Peter Richtarik

Introduction

Empirical risk minimization (aka finite-sum) problems form the dominant paradigm for training supervised machine learning models such as ridge regression, support vector machines, logistic regression, and neural networks. In its most general form, a finite sum problem has the form

where nn refers to the number of training data points (e.g., videos, images, molecules), xx is the vector representation of a model using dd features, and fi(x)f_{i}(x) is the loss of model xx on data point ii.

Variance-reduced methods. One of the most remarkable algorithmic breakthroughs in recent years was the development of variance-reduced stochastic gradient algorithms for solving (1). These methods are significantly faster than SGD in theory and practice on convex and strongly convex problems, and faster in theory on several classes on nonconvex problems (unfortunately, these methods are no yet successful in training production-grade neural networks).

Two of the most notable and popular methods belonging to the family of variance-reduced methods are SVRG and its accelerated variant known as Katyusha . The latter method accelerates the former via the employment of a novel “negative momentum” idea. Both of these methods have a double loop design. At the beginning of the outer loop, a full pass over the training data is made to compute the gradient of ff at a reference point wkw^{k}, which is chosen as the freshest iterate (SVRG) or a weighted average of recent iterates (for Katyusha). This gradient is then used in the inner loop to adjust the stochastic gradient ∇fi(xk)\nabla f_{i}(x^{k}), where ii is sampled uniformly at random from [n]=def{1,2,…,n}[n]\overset{\text{def}}{=}\{1,2,\dots,n\}, and xkx^{k} is the current iterate, so as to reduce its variance. In particular, both SVRG and Katyusha perform the adjustment gk=∇fi(xk)−∇fi(wk)+∇f(wk).g^{k}=\nabla f_{i}(x^{k})-\nabla f_{i}(w^{k})+\nabla f(w^{k}). Note that, like ∇fi(xk)\nabla f_{i}(x^{k}), the new search direction gkg^{k} is an unbiased estimator of ∇f(xk)\nabla f(x^{k}). Indeed,

where the expectation is taken over random choice of i∈[n]i\in[n]. However, it turns out that as the methods progress, the variance of gkg^{k}, unlike that of ∇fi(xk)\nabla f_{i}(x^{k}), progressively decreases to zero. The total effect of this is significantly faster convergence.

Converegnce of SVRG and Katyusha for LL–smooth and μ\mu–strongly convex functions. For instance, consider the regime where fif_{i} is LL–smooth for each ii, and ff is μ\mu–strongly convex:

In this regime, the iteration complexity of SVRG is O((n+\nicefracLμ)log⁡\nicefrac1ϵ),{\cal O}\left(\left(n+\nicefrac{{L}}{{\mu}}\right)\log\nicefrac{{1}}{{\epsilon}}\right), which is a vast improvement on the linear rate of gradient descent (GD), which is O(\nicefracnLμlog⁡\nicefrac1ϵ){\cal O}\left(\nicefrac{{nL}}{{\mu}}\log\nicefrac{{1}}{{\epsilon}}\right), and on the sublinear rate of SGD, which is O(\nicefracLμ+\nicefracσ2μ2ϵ){\cal O}(\nicefrac{{L}}{{\mu}}+\nicefrac{{\sigma^{2}}}{{\mu^{2}\epsilon}}), where σ2=\nicefrac1n∑i∥∇fi(x∗)∥2\sigma^{2}=\nicefrac{{1}}{{n}}\sum_{i}\|\nabla f_{i}(x^{*})\|^{2} and x∗x^{*} is the (necessarily unique) minimizer of ff. On the other hand, Katyusha enjoys the accelerated rate O((n+\nicefracnLμ)log⁡\nicefrac1ϵ),{\cal O}((n+\sqrt{\nicefrac{{nL}}{{\mu}}})\log\nicefrac{{1}}{{\epsilon}}), which is superior to that of SVRG in the ill-conditioned regime where \nicefracLμ≥n\nicefrac{{L}}{{\mu}}\geq n. This rate has been shown to be optimal in a certain precise sense .

In the past several years, an enormous effort of the machine learning and optimization communities was exerted into designing new efficient variance-reduced methods algorithms to tackle problem (1). These developments have brought about a renaissance in the field. The historically first provably variance-reduced method, the stochastic average gradient (SAG) method of , was awarded the Lagrange prize in continuous optimization in 2018. The SAG method was later modified to an unbiased variant called SAGA , achieving the same theoretical rates. Alternative variance-reduced method include MISO , FINITO , SDCA , dfSDCA , AdaSDCA , QUARTZ , SBFGS , SDNA , SARAH and S2GD , mS2GD , RBCN , JacSketch and SAGD . Accelerated variance-reduced method were developed in , , and .

Contributions

As explained in the introduction, a trade-mark structural feature of SVRG and its accelerated variant, Katyusha, is the presence of the outer loop in which a full pass over the data is made. However, the presence of this outer loop is the source of several issues. First, the methods are harder to analyze. Second, one needs to decide at which point the inner loop is terminated and the outer loop entered. For SVRG, the theoretically optimal inner loop size depends on both LL and μ\mu. However, μ\mu is not always known. Moreover, even when an estimate is available, as is the case in regularized problems with an explicit strongly convex regularizer, the estimate can often be very loose. Because of these issues, one often chooses the inner loop size in a suboptimal way, such as by setting it to nn or O(n){\cal O}(n).

Two loopless methods. In this paper we address the above issues by developing loopless variants of both SVRG and Katyusha; we refer to them as L-SVRG and L-Katyusha, respectively. In these methods, we dispose of the outer loop and replace its role by a biased coin-flip, to be performed in every step of the methods, used to trigger the computation of the gradient ∇f(wk)\nabla f(w^{k}) via a pass over the data. In particular, in each step, with (a small) probability p>0p>0 we perform a full pass over data and update the reference gradient ∇f(wk)\nabla f(w^{k}). With probability 1−p1-p we keep the previous reference gradient. This procedure can alternatively be interpreted as having an outer loop of a random length. However, the resulting methods are easier to write down, comprehend and analyze.

Fast rates are preserved. We show that L-SVRG and L-Katyusha enjoy the same fast theoretical rates as their loopy forefathers. Our proofs are different and the complexity results more insightful.

For L-SVRG with fixed stepsize η=\nicefrac16L\eta=\nicefrac{{1}}{{6L}} and probability p=\nicefrac1np=\nicefrac{{1}}{{n}}, we show (see Theorem 3.5) that for the Lyapunov function

we get E[Φk]≤ϵΦ0{{\rm E}}\left[\Phi^{k}\right]\leq\epsilon\Phi^{0} as long as k=O((n+\nicefracLμ)log⁡\nicefrac1ϵ).k={\cal O}\left(\left(n+\nicefrac{{L}}{{\mu}}\right)\log\nicefrac{{1}}{{\epsilon}}\right). In contrast, the classical SVRG result shows convergence of the expected functional suboptimality E[f(xk)−f(x∗)]{{\rm E}}\left[f(x^{k})-f(x^{*})\right] to zero at the same rate. Note that the classical result follows from our theorem by utilizing the inequality f(xk)−f(x∗)≤\nicefracL2∥xk−x∗∥2,f(x^{k})-f(x^{*})\leq\nicefrac{{L}}{{2}}\|x^{k}-x^{*}\|^{2}, which is a simple consequence of LL–smoothness. However, our result provides a deeper insight into the behavior of the method. In particular, it follows that the gradients ∇fi(wk)\nabla f_{i}(w^{k}) at the reference points wkw^{k} converge to the gradients at the optimum. This is a key intuition behind the workings of SVRG, one not revealed by the classical analysis. Hereby we close the gap in the theoretical understanding of the the SVRG convergence mechanism. Moreover, our theory predicts that as long as pp is chosen in the (possibly very large) interval

where c=Θ(1)c=\Theta(1), L-SVRG will enjoy the optimal complexity O((n+\nicefracLμ)log⁡\nicefrac1ϵ){\cal O}\left(\left(n+\nicefrac{{L}}{{\mu}}\right)\log\nicefrac{{1}}{{\epsilon}}\right). In the ill-conditioned regime \nicefracLμ≫n\nicefrac{{L}}{{\mu}}\gg n, for instance, we roughly have p∈[\nicefracμL,\nicefrac1n]p\in[\nicefrac{{\mu}}{{L}},\nicefrac{{1}}{{n}}]. This is in contrast with the (loopy/standard) SVRG method the outer loop of which needs to be of the size ≈\nicefracLμ\approx\nicefrac{{L}}{{\mu}}. To the best of our knowledge, SVRG does not enjoy this rate for an outer loop of size nn (or any value independent of μ\mu, which is often not known in practice), even though this is the setting most often used in practice. Several authors have tried to establish such a result, but without success. We thus answer an open problem since 2013, the inception of SVRG.

For L-Katyusha with stepsize η=θ2(1+θ2)θ1\eta=\frac{\theta_{2}}{(1+\theta_{2})\theta_{1}} we show convergence of the Lyapunov function

where Zk=L(1+ησ)2η∥zk−x∗∥2{\cal Z}^{k}=\tfrac{L(1+\eta\sigma)}{2\eta}\left\lVert z^{k}-x^{*}\right\rVert^{2}, Yk=1θ1(f(yk)−f(x∗)){\cal Y}^{k}=\tfrac{1}{\theta_{1}}(f(y^{k})-f(x^{*})), and Wk=θ2(1+θ1)pθ1(f(wk)−f(x∗)){\cal W}^{k}=\tfrac{\theta_{2}(1+\theta_{1})}{p\theta_{1}}(f(w^{k})-f(x^{*})), and where xk,ykx^{k},y^{k} and wkw^{k} are iterates produced by the method, with the parameters defined by σ=\nicefracμL\sigma=\nicefrac{{\mu}}{{L}}, θ1=min⁡{\nicefrac2σn3,\nicefrac12}\theta_{1}=\min\{\sqrt{\nicefrac{{2\sigma n}}{{3}}},\nicefrac{{1}}{{2}}\}, θ2=\nicefrac12\theta_{2}=\nicefrac{{1}}{{2}}, p=\nicefrac1np=\nicefrac{{1}}{{n}}. Our main result (Theorem 4.6) states that E[Ψk]≤ϵΨ0{{\rm E}}\left[\Psi^{k}\right]\leq\epsilon\Psi^{0} as long as k=O((n+\nicefracnLμ)log⁡\nicefrac1ϵ).k={\cal O}((n+\sqrt{\nicefrac{{nL}}{{\mu}}})\log\nicefrac{{1}}{{\epsilon}}).

Simplified analysis. Advantage of the loopless approach is that a single iteration analysis is sufficient to establish convergence. In contrast, one needs to perform elaborate aggregation across the inner loop to prove the convergence of the original loopy methods.

Superior empirical behaviour. We show through extensive numerical testing on both synthetic and real data that our loopless methods are superior to their loopy variants. We show through experiments that L-SVRG is very robust to the choice of pp from the optimal interval (6) predicted by our theory. Moreover, even the worst case for L-SVRG outperforms the best case for SVRG. This shows how further randomization can significantly speed up and stabilize the algorithm.

Notation. Throughout the whole paper we use conditional expectation E[X  ∣  xk,wk]{\rm E}\left[{\cal X}\;|\;x^{k},w^{k}\right] for L-SVRG and E[X  ∣  yk,zk,wk]{\rm E}\left[{\cal X}\;|\;y^{k},z^{k},w^{k}\right] for L-Katyusha, but for simplicity we will denote these expectations as E[X]{\rm E}\left[{\cal X}\right]. If E[X]{\rm E}\left[{\cal X}\right] refers to unconditional expectation, it is directly mentioned.

Loopless SVRG (L-SVRG)

In this section we describe in detail the Loopless SVRG method (L-SVRG), and its convergence.

The algorithm. The L-SVRG method, formalized as Algorithm 1, is inspired by the original SVRG method. We remove the outer loop present in SVRG and instead use a probabilistic update of the full gradient.This idea was independently explored in ; we have learned about this work after a first draft of our paper was finished. This update can be also seen in a way that outer loop size is generated by geometric distribution similar to .

Note that the reference point wkw^{k} (at which a full gradient is computed) is updated in each iteration with probability pp to the current iterate xkx^{k}, and is left unchanged with probability 1−p1-p. Alternatively, the probability pp can be seen as a parameter that controls the expected time before next full pass over data. To be more precise, the expected time before next full pass over data is \nicefrac1p\nicefrac{{1}}{{p}}. Intuitively, we wish to keep pp small so that full passes over data are computed rarely enough. As we shall see next, the simple choice p=\nicefrac1np=\nicefrac{{1}}{{n}} leads to complexity identical to that of original SVRG.

Convergence theory. A key role in the analysis is played by the gradient learning quantity

and the Lyapunov function Φk=def∥xk−x∗∥2+Dk.\Phi^{k}\overset{\text{def}}{=}\left\lVert x^{k}-x^{*}\right\rVert^{2}+{\cal D}^{k}. The analysis involves four lemmas, followed by the main theorem. We wish to mention the lemmas as they highlight the way in which the argument works. All lemmas combined, together with the main theorem, can be proved on a single page, which underlines the simplicity of our approach.

Our first lemma upper bounds the expected squared distance of xk+1x^{k+1} from x∗x^{*} in terms of the same distance but for xkx^{k}, function suboptimality, and variance of gkg^{k}.

In our next lemma, we further bound the variance of gkg^{k} in terms of function suboptimality and Dk{\cal D}^{k}.

Finally, we bound E[Dk+1]{\rm E}\left[{\cal D}^{k+1}\right] in terms of Dk{\cal D}^{k} and function suboptimality.

Putting the above three lemmas together naturally leads to the following result involving Lyapunov function (5).

Let the step size η≤\nicefrac16L\eta\leq\nicefrac{{1}}{{6L}}. Then for all k≥0k\geq 0 the following inequality holds:

In order to obtain a recursion involving the Lyapunov function on the right-hand side of (12)

Let η=\nicefrac16L\eta=\nicefrac{{1}}{{6L}}, p=\nicefrac1np=\nicefrac{{1}}{{n}}. Then E[Φk]≤εΦ0{\rm E}\left[\Phi^{k}\right]\leq\varepsilon\Phi^{0} as long as k≥O((n+\nicefracLμ)log⁡\nicefrac1ε).k\geq{\cal O}\left(\left(n+\nicefrac{{L}}{{\mu}}\right)\log\nicefrac{{1}}{{\varepsilon}}\right).

As the corollary of Lemma 3.4 we have E[Φk]≤max⁡{1−ημ,1−\nicefracp2}Φk−1.{\rm E}\left[\Phi^{k}\right]\leq\max\left\{1-\eta\mu,1-\nicefrac{{p}}{{2}}\right\}\Phi^{k-1}. Setting η=\nicefrac16L\eta=\nicefrac{{1}}{{6L}}, p=\nicefrac1np=\nicefrac{{1}}{{n}} and unrolling conditional probability one obtains E[Φk]≤max⁡{1−\nicefracμ6L,1−\nicefrac12n}kΦ0,{\rm E}\left[\Phi^{k}\right]\leq\max\left\{1-\nicefrac{{\mu}}{{6L}},1-\nicefrac{{1}}{{2n}}\right\}^{k}\Phi^{0}, which concludes the proof. ∎

Note that the step size does not depend on the strong convexity parameter μ\mu and yet the resulting complexity adapts to it.

Discussion. Examining (12), we can see that contraction of the Lyapunov function is max⁡{1−ημ,1−\nicefracp2}\max\{1-\eta\mu,1-\nicefrac{{p}}{{2}}\}. Due to the limitation of η≤\nicefrac16L\eta\leq\nicefrac{{1}}{{6L}}, the first term is at least 1−\nicefracη6μ1-\nicefrac{{\eta}}{{6\mu}}, thus the complexity cannot better than O(\nicefracLμlog⁡\nicefrac1ε){\cal O}\left(\nicefrac{{L}}{{\mu}}\log\nicefrac{{1}}{{\varepsilon}}\right). In terms of total complexity (number of stochastic gradient calls), L-SVRG calls the stochastic gradient oracle in expectation O(1+pn){\cal O}(1+pn) times times in each iteration. Combining these two complexities together, one gets the total complexity O((\nicefrac1p+n+\nicefracLμ+\nicefracLpnμ)log⁡\nicefrac1ε).{\cal O}\left(\left(\nicefrac{{1}}{{p}}+n+\nicefrac{{L}}{{\mu}}+\nicefrac{{Lpn}}{{\mu}}\right)\log\nicefrac{{1}}{{\varepsilon}}\right). Note that any choice of p∈[min⁡{\nicefraccn,\nicefraccμL},max⁡{\nicefraccn,\nicefraccμL}],p\in\left[\min\left\{\nicefrac{{c}}{{n}},\nicefrac{{c\mu}}{{L}}\right\},\max\left\{\nicefrac{{c}}{{n}},\nicefrac{{c\mu}}{{L}}\right\}\right], where c=Θ(1)c=\Theta(1), leads to the optimal total complexity O((n+\nicefracLμ)log⁡\nicefrac1ε){\cal O}\left(\left(n+\nicefrac{{L}}{{\mu}}\right)\log\nicefrac{{1}}{{\varepsilon}}\right). This fills the gap in SVRG theory, where the outer loop length (in our case \nicefrac1p\nicefrac{{1}}{{p}} in expectation) needs to be proportional to \nicefracLμ\nicefrac{{L}}{{\mu}}. Moreover, analysis for L-SVRG is much simpler and provides more insights.

Loopless Katyusha (L-Katyusha)

In this section we describe in detail the Loopless Katyusha method (L-Katyusha), and its convergence properties.

The algorithm. The L-Katyusha method, formalized as Algorithm 2, is inspired by the original Katyusha method. We use the same technique as for Algorithm 1, where we remove the outer loop present in Katyusha and instead use a probabilistic update of the full gradient.

The exact analogy applies to the reference point wkw^{k} (at which a full gradient is computed) as for L-SVRG. Instead of updating this point in a deterministic way every mm iteration, we use the probabilistic update with parameter pp, when we update wk+1w^{k+1} to the current iterate yky^{k} with this probability and is left unchanged with probability 1−p1-p. As we shall see next, the same choice p=\nicefrac1np=\nicefrac{{1}}{{n}} as for L-SVRG leads to complexity identical to that of original Katyusha.

Convergence theory. In comparison to L-SVRG, we don’t use gradient mapping as the key component of our analysis. Instead, we prove convergence of functional values in yk,wky^{k},w^{k} and point-wise convergence of zkz^{k}. This is summarized in the following Lyapunov function:

where Zk=L(1+ησ)2η∥zk−x∗∥2{\cal Z}^{k}=\tfrac{L(1+\eta\sigma)}{2\eta}\left\lVert z^{k}-x^{*}\right\rVert^{2}, Yk=1θ1(f(yk)−f(x∗)){\cal Y}^{k}=\tfrac{1}{\theta_{1}}(f(y^{k})-f(x^{*})), Wk=θ2(1+θ1)pθ1(f(wk)−f(x∗)){\cal W}^{k}=\tfrac{\theta_{2}(1+\theta_{1})}{p\theta_{1}}(f(w^{k})-f(x^{*})). Note that even if xkx^{k} is not in this function, its point-wise convergence is directly implied by the convergence of Ψk\Psi^{k} due to the definition of xkx^{k} in Algorithm 2 and LL-smoothness of ff.

The analysis involves five lemmas, followed by the convergence summarized in the main theorem. The lemmas highlight important steps of our analysis. The simplicity of our approach is still preserved: all lemmas and the main theorem can be proved on not more than two pages.

Our first lemma upper bounds the variance of the gradient estimator gkg^{k}, which eventually goes to zero as our algorithm progresses.

Next two lemmas are more technical, but essential for proving the convergence.

Finally, we use the update of Algorithm 2 to decompose Wk+1{\cal W}^{k+1} in terms of Wk{\cal W}^{k} and Yk{\cal Y}^{k}, which is one of the main components that allow for simpler analysis than the one of original Katyusha.

Putting all lemmas together, we obtain the following contraction of the Lyapunov function (7).

Let θ1,θ2>0\theta_{1},\theta_{2}>0, θ1+θ2≤1\theta_{1}+\theta_{2}\leq 1, σ=μL\sigma=\frac{\mu}{L} and η=θ2(1+θ2)θ1\eta=\frac{\theta_{2}}{(1+\theta_{2})\theta_{1}}, then we have

In order to obtain a recursion involving the Lyapunov function on the right-hand side of (18)

Let θ1=min⁡{\nicefrac2σn3,\nicefrac12}\theta_{1}=\min\{\sqrt{\nicefrac{{2\sigma n}}{{3}}},\nicefrac{{1}}{{2}}\}, θ2=\nicefrac12\theta_{2}=\nicefrac{{1}}{{2}}, p=\nicefrac1np=\nicefrac{{1}}{{n}}. Then E[Ψk]≤εΨ0{\rm E}\left[\Psi^{k}\right]\leq\varepsilon\Psi^{0} after the following number of iterations: k=O((n+\nicefracnLμ)log⁡\nicefrac1ε).k={\cal O}((n+\sqrt{\nicefrac{{nL}}{{\mu}}})\log\nicefrac{{1}}{{\varepsilon}}).

From Lemma 4.5 we get E[Ψk+1]≤max⁡{\nicefrac1(1+ησ),1−θ1(1−θ2),1−\nicefracpθ1(1+θ1)}Ψk.{\rm E}\left[\Psi^{k+1}\right]\leq\max\left\{\nicefrac{{1}}{{(1+\eta\sigma)}},1-\theta_{1}(1-\theta_{2}),1-\nicefrac{{p\theta_{1}}}{{(1+\theta_{1})}}\right\}\Psi^{k}. Setting p=\nicefrac1np=\nicefrac{{1}}{{n}}, θ1=min⁡{\nicefrac2σn3,\nicefrac12}\theta_{1}=\min\{\sqrt{\nicefrac{{2\sigma n}}{{3}}},\nicefrac{{1}}{{2}}\}, θ2=\nicefrac12\theta_{2}=\nicefrac{{1}}{{2}}, and unrolling conditional probability one obtains E[Ψk+1]≤(1−θ)E[Ψk]{\rm E}\left[\Psi^{k+1}\right]\leq(1-\theta){\rm E}\left[\Psi^{k}\right], where θ=min⁡{\nicefracσ6θ1,\nicefracθ12n}.\theta=\min\left\{\nicefrac{{\sigma}}{{6\theta_{1}}},\nicefrac{{\theta_{1}}}{{2n}}\right\}. Choosing σ=\nicefracμL\sigma=\nicefrac{{\mu}}{{L}} concludes the proof. ∎

Discussion. One can show by analyzing (18) that for ill-conditioned problems (n<\nicefracLμn<\nicefrac{{L}}{{\mu}}), the iteration complexity is O(\nicefracLμplog⁡\nicefrac1ε){\cal O}(\sqrt{\nicefrac{{L}}{{\mu p}}}\log\nicefrac{{1}}{{\varepsilon}}). Algorithm 2 calls stochastic gradient oracle O(1+pn){\cal O}(1+pn) times per iteration in expectation. Thus, the total complexity is O((1+pn)\nicefracLμplog⁡\nicefrac1ε){\cal O}((1+pn)\sqrt{\nicefrac{{L}}{{\mu p}}}\log\nicefrac{{1}}{{\varepsilon}}). One can see that p=Θ(\nicefrac1n)p=\Theta\left(\nicefrac{{1}}{{n}}\right) leads to optimal rate.

Numerical Experiments

We compare our methods L-SVRG and L-Katyusha with their original version. It is well-known that whenever practical, SAGA is a bit faster than SVRG. While a comparison to SAGA seems natural as it also does not have a double loop structure, we position our loopless methods for applications where the high memory requirements of SAGA prevent it to be applied. Thus, we do not compare to SAGA.

Plots are constructed in such a way that the yy-axis displays ∥xk−x⋆∥2\left\lVert x^{k}-x^{\star}\right\rVert^{2} for L-SVRG and ∥yk−x⋆∥2\left\lVert y^{k}-x^{\star}\right\rVert^{2} for L-Katyusha, where x⋆x^{\star} were obtained by running gradient descent for a large number of epochs. The xx-axis displays the number of epochs (full gradient evaluations). That is, nn computations of ∇fi(x)\nabla f_{i}(x) equals one epoch.

Superior practical behaviour of the loopless approach. Here we show that L-SVRG and L-Katyusha perform better in experiments than their loopy variants. In terms of theoretical iteration complexity, both the loopy and the loopless methods are the same. However, as we can see from Figure 1, the improvement of the loopless approach can be significant. One can see that for these datasets, L-SVRG is always better than SVRG, and can be faster by several orders of magnitude! Looking at Figure 2, we see that the performance of L-Katyusha is at least as good as that of Katyusha, and can be significantly faster in some cases. All parameters of the methods were chosen as suggested by theory. For L-SVRG and L-Katyusha they are chosen based on Theorems 3.5 and 4.6, respectively. For SVRG and Katyusha we also choose the parameters based on the theory, as described in the original papers. The initial point x0x^{0} is chosen to be the origin.

Different choices of probability/ outer loop size. We now compare several choices of the probability pp of updating the full gradient for SVRG and several outer loop sizes mm for SVRG. Since our analysis guarantees the optimal rate for any choice of pp between \nicefrac1n\nicefrac{{1}}{{n}} and \nicefracμL\nicefrac{{\mu}}{{L}} for well condition problems, we decided to perform experiments for pp within this range. More precisely, we choose 55 values of pp, uniformly distributed in logarithmic scale across this interval, and thus our choices are nn,κn3\sqrt{\kappa n^{3}}, κn\sqrt{\kappa n}, κ3n\sqrt{\kappa^{3}n}, and κ\kappa, where κ=\nicefracLμ\kappa=\nicefrac{{L}}{{\mu}}, denoted in the figures by 1,2,3,4,51,2,3,4,5, respectively. Since the expected “outer loop” length (length for which reference point stays the same) is \nicefrac1p\nicefrac{{1}}{{p}}, for SVRG we choose m=\nicefrac1pm=\nicefrac{{1}}{{p}}. Looking at Figure 3, one can see that L-SVRG is very robust to the choice of pp from the “optimal interval” predicted by our theory. Moreover, even the worst case for L-SVRG outperforms the best case for SVRG.

All methods together. Finally, we provide all algorithms together in one plot for different datasets with different regularizer weight, thus with different condition numbers, displayed in Figure 4. As for the previous experiments, loopless methods are not worse and sometimes significantly better.

References

Supplementary Material: SVRG and Katyusha are Better Without the Outer Loop

The next lemma is a consequence of Jensen’s inequality applied to x↦∥x∥2x\mapsto\|x\|^{2}.

Appendix B Proofs for Algorithm 1 (L-SVRG)

In all proofs below, we will for simplicity write f∗=deff(x∗)f^{*}\overset{\text{def}}{=}f(x^{*}).

Definition of xk+1x^{k+1} and unbiasness of gkg^{k} guarantee that

B.2 Proof of Lemma 3.2

B.3 Proof of Lemma 3.3

B.4 Proof of Lemma 3.4

Now we use the fact that η≤16L\eta\leq\frac{1}{6L} and obtain the desired inequality:

Appendix C Proofs for Algorithm 2 (L-Katyusha)

To upper bound the variance of gkg^{k} we first uses its definition

C.2 Proof of Lemma 4.2

We start with the definition of zk+1z^{k+1}

which implies ηLgk=ησ(xk−zk+1)+(zk−zk+1),\frac{\eta}{L}g^{k}=\eta\sigma(x^{k}-z^{k+1})+(z^{k}-z^{k+1}), which further implies that

C.3 Proof of Lemma 4.3

where the last inequality uses the Young’s inequality in the form of <a,b>≥−∥a∥22β−β∥b∥22\left<a,b\right>\geq-\frac{\left\lVert a\right\rVert^{2}}{2\beta}-\frac{\beta\left\lVert b\right\rVert^{2}}{2} for β=ηθ1L(1−ηθ1)\beta=\frac{\eta\theta_{1}}{L(1-\eta\theta_{1})}, which concludes the proof.

C.4 Proof of Lemma 4.4

From the definition of wk+1w^{k+1} in Algorithm 2 we have

The rest of proof follows from the definition of Wk{\cal W}^{k} (17).

C.5 Proof of Lemma 4.5

Combining all the previous lemmas together, we obtain

where in the second inequality we use also convexity of f(x)f(x).

Using definition of Wk{\cal W}^{k} we get