Iterate averaging as regularization for stochastic gradient descent

2018·Arxiv

Abstract

Abstract

We propose and analyze a variant of the classic Polyak–Ruppert averaging scheme, broadly used in stochastic gradient methods. Rather than a uniform average of the iterates, we consider a weighted average, with weights decaying in a geometric fashion. In the context of linear least squares regression, we show that this averaging scheme has a the same regularizing effect, and indeed is asymptotically equivalent, to ridge regression. In particular, we derive finite-sample bounds for the proposed approach that match the best known results for regularized stochastic gradient methods.

1. Introduction

Stochastic gradient methods are ubiquitous in machine learning, where they are typically referred to as SGD (stochastic gradient descent1). Since these incremental methods use little computation per data point, they are naturally adapted to processing very large data sets or streams of data. Stochastic gradient methods have a long history, starting from the pioneering paper by Robbins and Monro (Robbins and Monro, 1951). For a more detailed discussion, we refer to the excellent review given by Nemirovski et al. (2009). In the present paper, we propose a variant of SGD based on a weighted average of the iterates. The idea of averaging iterates goes back to Polyak (1990) and Ruppert (1988), and indeed it is often referred to as Polyak–Ruppert averaging (see also Polyak and Juditsky, 1992). In this paper, we study SGD in the context of the linear least-squares regression problem, considering both finite- and infinite-dimensional settings. This latter case allows to derive results for nonparametric learning with kernel methods—we refer to the appendix in Rosasco and Villa (2015) for a detailed discussion on this subject. The study of SGD for least squares is classical in stochastic approximation (Kushner and Yin, 2003), where it is commonly known as the least-mean-squares (LMS) algorithm. In the context of machine learning theory, prediction-error guarantees for online algorithms can be derived through a regret analysis in a sequential-prediction setting and a so called online-to-batch conversion (Shalev-Shwartz, 2012; Hazan, 2016). For example, results in Vovk (2001); Azoury and Warmuth (2001) and Hazan et al. (2007) directly apply to least squares. Alternatively, one can directly analyze SGD in the stochastic setting, as done by Smale and Yao (2006); Ying and Pontil (2008); Tarres and Yao (2014), where the last iterate and decaying step-size are considered, and more recently by Rosasco and Villa (2015); Lin and Rosasco (2017) where multiple passes and mini-batching are considered. A recently popular approach is combining constant step-sizes with Polyak–Ruppert averaging, which was first shown to lead to strong finite-time prediction guarantees after a single pass on the data by Bach and Moulines (2013). This approach was first studied by Gy¨orfi and Walk (1996) and subsequent progress was made by D´efossez and Bach (2015); Dieuleveut et al. (2017); Jain et al. (2016, 2017); Lakshminarayanan and Szepesv´ari (2018).

In this paper we propose and analyze a novel form of weighted average, given by a sequence of weights decaying geometrically, so that the first iterates have more weight. Our main technical contribution is a characterization of the properties of this particular weighting scheme that we call geometric Polyak–Ruppert averaging. Our first result shows that SGD with geometric Polyak–Ruppert averaging is in expectation equivalent to considering SGD with a regularized loss function, and both sequences converge to the Tikhonovregularized solution of the expected least-squares problem. The regularization parameter is a tuning parameter defining the sequence of geometric weights. This result strongly suggests that geometric Polyak–Ruppert averaging can be used to control the bias-variance properties of the corresponding SGD estimator. Indeed, our main result quantifies this intuition deriving a finite-sample bound, matching previous results for regularized SGD, and leading to optimal rates (Tsybakov, 2008). While averaging is widely considered to have a stabilizing effect, to the best of our knowledge this is the first result characterizing the stability of an averaging scheme in terms of its regularization properties and corresponding prediction guarantees. Our findings can be contrasted to recent results on tail averaging (Jain et al., 2016) and provide some guidance on when and how different averaging strategies can be useful. On a high level, our results suggest that geometric averaging should be used when the data is poorly-conditioned and/or relatively small, and tail averaging should be used in the opposite case. Further, from a practical point of view, geometric Polyak–Ruppert averaging provides an efficient approach to perform model selection, since a regularization path (Friedman et al., 2001) is computed efficiently. Indeed, it is possible to compute a full pass of SGD once and store all the iterates, to then rapidly compute off-line the solutions corresponding to different geometric weights (or tail averages), hence different regularization levels. As the averaging operation is entirely non-serial, this method lends itself to trivially easy parallelization.

The rest of the paper is organized as follows. In Section 2, we introduce the necessary background and present the geometric Polyak–Ruppert averaging scheme. In Section 3, we show the asymptotic equivalence between ridge regression and constant-stepsize SGD with geometric iterate averaging. Section 4 presents and dicuss our main results regarding the finite-time prediction error of the method. Section 5 describes the main steps in the proofs. The full proof is included in the Appendix. We conclude this section by introducing some basic notation used throughout the paper.

Notation. Let H be a separable Hilbert space with the inner product and norm . We let be the outer product of vector , which acts on as the rank-one operator . For a linear operator A acting on H, we let be its adjoint and the Frobenius norm. An operator A is positive semi-definite (PSD) if it satisfies for all and Hermitian if . We use to denote that an operator A is Hermitian and PSD (in short, HPSD). For HPSD operators A and B, we use to denote . For a HPSD operator A, we use to denote the unique HPSD operator satisfying . The identity operator on H is denoted as I. Besides standard asymptotic notation like or , we will sometimes use the cleaner but less standard notation to denote a = O (b). We will consider algorithms that interact with stochastic data in an sequential fashion. The sequence of random variables observed during the interaction induce a filtration . We will use the shorthand to denote expectations conditional on the history.

2. Preliminaries

We study the problem of linear regression under the square loss, more commonly known as linear least squares regression. The objective in this problem is to minimize the expected risk

where is a parameter vector, is a covariate and is a label, with (x, y) drawn from a fixed (but unknown) distribution D. Letting denote the covariance operator, the minimizer of the risk is given by

and satisfies . We assume to exist, since in general this might not be true in infinite dimensions. Also we abuse the notation in Eq. (2) since in general is not invertible and a pseudoinverse should be considered. This choice is made only to ease the notation.

We study algorithms that take as input a set of data points drawn identically and indepen- dently from D and output a weight vector w to approximately minimize (1). The quality of an approximate solution is measured by the the excess risk

To compute a solution from data, we consider the stochastic gradient method, a.k.a. SGD, that for least squares takes the form

where is a sequence of stepsizes (or learning rates), and is an initial point. Typically, a decaying stepsize sequence is chosen to ensure convergence, see Nemirovski et al. (2009) and references therein. A result relevant to our study is obtained by Bach and Moulines (2013) for a finite dimensional setting (H is of dimension d). Unlike previous results, here it is shown that convergence for a constant stepsize can be proved if Polyak–Ruppert (PR) averaging

is considered. This result was later strengthened in various respects by D´efossez and Bach (2015) and

Dieuleveut, Flammarion, and Bach (2017), notably by weakening the assumptions in Bach and Moulines (2013) and separating error terms related to “bias” and “variance”. We highlight one result from Dieuleveut et al. (2017) which considers the sequence

and under technical assumptions discussed later, proves the excess-risk bound

where is an upper bound on the variance of the label noise. The iteration (4) is only of theoretical interest since the covariance is not known in practice. However, the obtained bound is simpler to present allows easier comparison with our result. A bound slightly more complex than (5) can be obtained when is not known (Dieuleveut et al., 2017, Theorem 2).

In this paper, we propose a generalized version of Polyak–Ruppert averaging that we call geometric Polyak–Ruppert averaging. Specifically, the algorithm we study computes the standard SGD iterates as given by Equation (3) and outputs

after round n, where is a tuning parameter. That is, the output is a geometrically discounted (and appropriately normalized) average of the plain SGD iterates that puts a higher weight on initial iterates. It is easy to see that setting exactly recovers the standard form of Polyak–Ruppert averaging. Our main result essentially shows that the resulting estimate satisfies2

under the same assumptions as the ones made by Dieuleveut et al. (2017). Notably, this guarantee matches the bound of Equation (5), with the key difference being that the factor in the first term is replaced by . This observation suggests the (perhaps surprising) conclusion that geometric Polyak–Ruppert averaging has a regularization effect qualitatively similar to Tikhonov regularization. Before providing the proof of the main result stated above, we first show that this similarity is more than a coincidence. Specifically, we begin by showing in Section 3 that the limit of the weighted iterates is exactly the ridge regression solution on expectation.

3. Geometric iterate averaging realizes Tikhonov regularization on expectation

We begin by studying the relation between the averaged iterates of unregularized SGD and the iterates of regularized SGD on expectation. This setting will allow us to make minimal assumptions: We merely assume that so that the covariance operator satisfies for some B > 0. For ease of exposition, we assume in this section that . First, we notice the relation

between and , which can be iteratively applied to obtain

In contrast we also define the iterates of regularized SGD with stepsize and regularization parameter as

which can be similarly shown to satisfy

This latter definition can be seen as an empirical version of the iteration in Eq. (4). The following proposition reveals a profound connection between the limits of and the geometrically discounted average as (given that the stepsizes are small enough for the limits to exist).

Proposition 1 Let and be such that and , and assume that . Then, and both exist and satisfy

Proof The analysis crucially relies on defining the geometric random variable G with law P [G = k] = for all k = 1, 2, . . . and noticing that . We let stand for taking expectations with respect to the distribution of G. First we notice that the limit of can be written as

By our assumption on , we have , which implies that then the limit on the right-hand side exists and satisfies . Having established the existence of the limit, we rewrite

the regularized SGD iterates (7) as

This concludes the proof.

It is useful to recall that is the solution of the problem

that is Tikhonov regularization applied to the expected risk or, in other words, the population version of the ridge regression estimator. Then, the above result shows that SGD with geometric average (6) and the regularized iteration (7) both converge to this same solution. Besides this result, we can also show that the expected iterates and are also closely connected for finite t, without any assumption on the learning rates.

Proposition 2 Let and be such that and . Then,

This proposition is proved using the same ideas as Proposition 1; we include the proof in Appendix A for completeness.

4. Main result: Finite-time performance guarantees

While the previous section establishes a strong connection between the geometrically weighted SGD iterates with the iterates of regularized SGD on expectation, this connection is clearly not enough for the known performance guarantees to carry over to our algorithm. Specifically, the two iterative schemes propagate noise differently, thus the covariance of the resulting iterate sequences may be very different from each other. In this section, we prove our main result that shows that the prediction error of our algorithm also behaves similarly to that of SGD with Tikhonov regularization. For our analysis, we will make the same assumptions as Dieuleveut et al. (2017) and we will borrow several ideas from them, as well as most of their notation.

We now state the assumptions that we require for proving our main result. We first state an assumption on the fourth moment of the covariates.

Assumption 1 There exists R > 0 such that

This assumption implies that tr , and is satisfied, for example, when the covariates satisfy almost surely. We always assume a minimizer of the expected risk to exist and also make an assumption on the residual defined as the random variable

It is easy to show that , even though does not hold in general. We make the following assumption on the residual:

Assumption 2 There exists such that

This assumption is satisfied when and y are almost surely bounded, or when the model is well-specified and corrupted with bounded noise (i.e., when is independent of x and has variance bounded by ). Under the above assumptions, we prove the following bound on the prediction error—our main result:

Theorem 3 Suppose that Assumptions 1 and 2 hold and assume that and . Then, the iterates computed by the recursion given by Equations (3) and (6) satisfy

The (rather technical) proof of the theorem closely follows the proof of Theorem 2 of Dieuleveut et al. (2017). We describe the main components of the proof of our main result in Section 5. For didactic purposes, we also present a simplified version of our analysis where we assume full knowledge of in Appendix C.

4.1. Discussion

We next discuss various aspects and implications of our results.

Comparison with Dieuleveut et al. (2017). Apart from constant factors3, our bound above precisely matches that of Theorem 1 of Dieuleveut et al. (2017), except for an additional term of order . This term, however, is not a mere artifact of our proof: in fact it captures a distinctive noise-propagating effect of geometric PR averaging scheme. Indeed, the regularization effect of our iterate averaging scheme is different from that of Tikhonov-regularized SGD in one significant way: while Tikhonov regularization increases the bias and strictly decreases the variance, our scheme may actually increase the variance for certain choices of . To see this, observe that setting a large puts a large weight on the initial iterates, so that the initial noise is amplified compared to noise in the later stages, and the concentration of the total noise becomes worse. We note however that this extra term does not qualify as a serious limitation, since the commonly recommended setting still preserves the optimal rates for both the bias and the variance up to constant factors.

Optimal excess risk bounds. The bound in Theorem 3 is essentially the same as the one derived in Dieuleveut et al. (2017)[Theorem 2]. Following their same reasoning, the bound can be optimized with respect to to derive the best parameters choice and explicit upper bounds on the corresponding excess risk. In the finite dimensional case, it is easy to derive a bound of order O(d/n), which is known to be optimal in a minmax sense (Tsybakov, 2008). In the infinite dimensional case, optimal minmax bound can again be easily derived, and also refined under further assumptions on and the covariance (De Vito et al., 2005; Caponnetto and De Vito, 2007). We omit this derivation.

When should we set ? We have two answers depending on the dimensionality of the underlying Hilbert space H. For infinite dimensional spaces, it is clearly necessary to set . In the finite-dimensional case, the advantage of our regularization scheme is less clear at first sight: while Tikhonov regularization strictly decreases the variance, this is not necessarily true for our scheme (as discussed above). A closer look reveals that, under some (rather interpretable) conditions, we can reduce the variance, as quantified by the following proposition.

Proof Letting be the eigenvalues of sorted in decreasing order, we have

Taking derivative of f with respect to gives

In particular, we have

so holds whenever tr. The proof is concluded by observing that implies the existence of a with the claimed property.

Intuitively, Proposition 4 suggests that the geometric PR averaging can definitely reduce the variance over standard PR averaging whenever the covariance matrix is poorly conditioned and/or the sample size is small. Notice however that the above argument only shows one example of a good choice of ; many other good choices may exist, but these are harder to characterize.

Geometric averaging vs. tail averaging. It is interesting to contrast our approach with the tail averaging scheme studied by Jain et al. (2016, 2017): instead of putting large weight on the initial iterates as our method does, Jain et al. suggest to average the last iterates of SGD for some . The effect of this operation is that theterm arising from Polyak–Ruppert averaging is replaced by a term of order , where is the smallest eigenvalue of . Clearly, this yields a significant asymptotic speedup, but gives no advantage when (noting that ). Contrasting this condition with our Proposition 4 leads to an interesting conclusion: for small values of n, geometric averaging has an edge over tail averaging and vice versa.

What is the computational advantage? The main practical advantage of our averaging scheme over Tikhonov regularization is a computational one: validating regularization parameters becomes trivially easy to parallelize. Indeed, one can perform a single pass of unregularized SGD over the data, store the iterates and average them with various schedules to evaluate different choices of . Through parallelization, this approach can achieve huge computational speedups over running regularized SGD from scratch. To see this, observe that the averaging operation is entirely non-serial: one can cut the (stored) SGD iterates into K contiguous batches and let each individual worker perform a geometrically discounted averaging with the same discount factor . The resulting averages are then combined by the master with appropriate weights. In contrast, regularized SGD is entirely serial, so validation cannot be parallelized.

Choosing the right averaging. Finally, we note that the same method as above can be used to choose the correct parameter for tail averaging. This suggests a simple and highly parallelizable scheme for choosing the right averaging, with the identity of the best scheme depending on the interaction between n and as discussed above.

Connections to early stopping. A close inspection of the proofs of our Propositions 1 and 2 reveals an interesting perspective on geometric iterate averaging: Thinking of the averaging operation as computing a probabilistic expectation, one can interpret the geometric average as a probabilistic early stopping method where the stopping time is geometrically distributed. Early stopping is a very well-studied regularization method for multipass stochastic gradient learning algorithms that has been observed and formally proved to have effects similar to Tikhonov regularization (Yao et al., 2007; Rosasco and Villa, 2015). So far, all4 published results that we are aware of merely point out the qualitative similarities between the performance bounds obtained for these two regularization methods, showing that using the stopping time should be chosen as . In contrast, our Propositions 1 and 2 show a much deeper connection: geometric random stopping with expected stopping time not only recovers the performance bounds, but exactly recovers the ridge-regression solution.

Open questions. It is natural to ask whether geometric iterate averaging has similar regularization effects in other stochastic approximation settings too. An possible direction for future work is studying the effects of our averaging scheme on accelerated and/or variance-reduced variants of SGD. Another promising direction is studying general linear stochastic approximation schemes (Lakshminarayanan and Szepesv´ari, 2018), and particularly Temporal-Difference learning algorithms for Reinforcement Learning that have so far resisted all attempts to regularize them (Sutton and Barto, 1998; Szepesv´ari, 2010; Farahmand, 2011).

5. The proof of Theorem 3

Our proof closely follows that of Dieuleveut et al. (2017, Theorem 1), with the key differences that

• we do not have to deal with an explicit “regularization-based” error term that gives rise to a term proportional to in their bound, and

• the factors for iterate averaging are replaced by for each round, where

As we will see, this change will propagate through the analysis and will eventually replace the and factors in the final bound by and , respectively.

In the interest of space, we only provide an outline of the proof here and defer the proofs of the key lemmas to Appendix B. Throughout the proof, we will suppose that the conditions of Theorem 3 hold. The lemma below shows that the factors involving are of the order claimed in the Theorem.

Lemma 5 For any , we have

and

The straightforward proof is given in Appendix B.1. Now we are ready to lay out the proof of Theorem 3. Let us start by introducing the notation

and recalling the definition . A simple recursive argument shows that

Thus, the averaged iterates satisfy

We first show a simple upper bound on the excess risk :

Lemma 6

The proof is included in Appendix B.2. In order to further upper bound the right-hand side in the bound stated in Lemma 6, we can combine the decomposition of in Equation (8) with the Minkowski inequality to get

The first term in the above decomposition can be thought of as the excess risk of a “noiseless” process (where ) and the second term as that of a “pure noise” process (where ). The rest of the analysis is devoted to bounding these two terms.

We begin with the conceptually simpler case of bounding , which can be done uniformly for all t. In particular, we have the following lemma:

Lemma 7 For any t, we have

The rather technical proof is presented in Appendix B.3. We now turn to bounding the excess risk of the “noiseless” process, :

The following lemma states a bound on .

Lemma 8

The extremely technical proof of this theorem is presented in Appendix B.4. The proof of Theorem 3 is concluded by plugging the bounds of Lemmas 7 and 8 into Equation (9) and using Lemma 6 to obtain

Now we can finish by using the bounds on and given in Lemma 5.

Acknowledgments

GN is supported by the UPFellows Fellowship (Marie Curie COFUND program n600387). LR is funded by the Air Force project FA9550-17-1-0390 (European Office of Aerospace Research and Development) and by the FIRB project RBFR12M3AC (Italian Ministry of Education, University and Research). The authors thank Francesco Orabona, Csaba Szepesv´ari and G´abor Lugosi for interesting discussions.

References

Katy S. Azoury and Manfred K. Warmuth. Relative loss bounds for on-line density estimation with the exponential family of distributions. Machine Learning Journal, 43(3):211–246, 2001.

Francis Bach and Eric Moulines. Non-strongly-convex smooth stochastic approximation with convergence rate O(1/n). In Advances in Neural Information Processing Systems 26, pages 773–781, 2013.

Andrea Caponnetto and Ernesto De Vito. Optimal rates for the regularized least-squares algorithm. Foundations of Computational Mathematics, 7(3):331–368, 2007.

Ernesto De Vito, Andrea Caponnetto, and Lorenzo Rosasco. Model selection for regularized least-squares algorithm in learning theory. Foundations of Computational Mathematics, 5(1):59–85, 2005.

Alexandre D´efossez and Francis Bach. Constant step size least-mean-square: Bias-variance trade-offs and optimal sampling distributions. In International Conference on Artificial Intelligence and Statistics (AISTATS),, 2015.

Aymeric Dieuleveut, Nicolas Flammarion, and Francis Bach. Harder, better, faster, stronger convergence rates for least-squares regression. Journal of Machine Learning Research, 18(101):1–51, 2017.

Amir-massoud Farahmand. Regularization in reinforcement learning. PhD thesis, University of Alberta, 2011.

Henry E. Fleming. Equivalence of regularization and truncated iteration in the solution of ill-posed image reconstruction problems. Linear Algebra and its applications, 130:133–150, 1990.

Jerome Friedman, Trevor Hastie, and Robert Tibshirani. The elements of statistical learning, volume 1. Springer series in statistics New York, 2001.

L´aszl´o Gy¨orfi and Harro Walk. On the averaged stochastic approximation for linear regression. SIAM Journal on Control and Optimization, 34(1):31–61, 1996.

Elad Hazan. Introduction to online convex optimization. Foundations and Trends in Optimization, 2(3-4): 157–325, 2016.

Elad Hazan, Amit Agarwal, and Satyen Kale. Logarithmic regret algorithms for online convex optimization. Machine Learning, 69:169–192, 2007.

Prateek Jain, Sham M. Kakade, Rahul Kidambi, Praneeth Netrapalli, and Aaron Sidford. Parallelizing stochastic approximation through mini-batching and tail-averaging. arXiv preprint arXiv:1610.03774, 2016.

Prateek Jain, Sham M. Kakade, Rahul Kidambi, Praneeth Netrapalli, Venkata Krishna Pillutla, and Aaron Sidford. A markov chain theory approach to characterizing the minimax optimality of stochastic gradient descent (for least squares). arXiv preprint arXiv:1710.09430, 2017.

Harold Kushner and G George Yin. Stochastic approximation and recursive algorithms and applications, volume 35. Springer Science & Business Media, 2003.

Chandrashekar Lakshminarayanan and Csaba Szepesv´ari. Linear stochastic approximation: Constant step- size and iterate averaging. To appear in proceedings of AI & Statistics, 2018.

Junhong Lin and Lorenzo Rosasco. Optimal rates for multi-pass stochastic gradient methods. Journal of Machine Learning Research, 18(97):1–47, 2017.

Arkadi Nemirovski, Anatoli Juditsky, Guanghui Lan, and Alexander Shapiro. Robust stochastic approxima- tion approach to stochastic programming. SIAM Journal on optimization, 19(4):1574–1609, 2009.

Boris Polyak. New stochastic approximation type procedures. Autom. i Telemekh., 7:98107, 1990, 7(7): 98–107, 1990.

Boris T. Polyak and Anatoli B. Juditsky. Acceleration of stochastic approximation by averaging. SIAM Journal on Control and Optimization, 30(4):838–855, 1992.

Herbert Robbins and Sutton Monro. A stochastic approximation method. The annals of mathematical statistics, pages 400–407, 1951.

Lorenzo Rosasco and Silvia Villa. Learning with incremental iterative regularization. In Proceedings of the 28th International Conference on Neural Information Processing Systems-Volume 1, pages 1630–1638. MIT Press, 2015.

David Ruppert. Efficient estimations from a slowly convergent robbins–monro process. Technical report, Cornell University Operations Research and Industrial Engineering, 1988.

Shai Shalev-Shwartz. Online learning and online convex optimization. Foundations and Trends in Machine Learning, 4(2):107–194, 2012.

Steve Smale and Yuan Yao. Online learning algorithms. Foundations of computational mathematics, 6(2): 145–170, 2006.

Richard S Sutton and Andrew G Barto. Reinforcement learning: An introduction, volume 1. 1998.

Csaba Szepesv´ari. Algorithms for reinforcement learning. Synthesis lectures on artificial intelligence and machine learning, 4(1):1–103, 2010.

Pierre Tarres and Yuan Yao. Online learning as stochastic approximation of regularization paths: Optimality and almost-sure convergence. IEEE Transactions on Information Theory, 60(9):5716–5735, 2014.

Alexandre B. Tsybakov. Introduction to Nonparametric Estimation. Springer Publishing Company, Incorporated, 1st edition, 2008. ISBN 0387790519, 9780387790510.

Vladimir Vovk. Competitive on-line statistics. International Statistical Review, 69:213–248, 2001.

Yuan Yao, Lorenzo Rosasco, and Andrea Caponnetto. On early stopping in gradient descent learning. Constructive Approximation, 26(2):289–315, 2007.

Yiming Ying and Massimiliano Pontil. Online gradient descent learning algorithms. Foundations of Computational Mathematics, 8(5):561–596, 2008.

Appendix A. The proof of Proposition 2

The proof is similar to that of Proposition 1, although a little more cluttered due to the normalization constants involved in the definition of . The analysis in this case relies on defining the truncated geometric random variable G with law

for all k = 0, 1, 2, . . ., t. Using the notation , we have . Again, letting stand for taking expectations with respect to the distribution of G, we rewrite the regularized SGD iterates as

where we used and the definition of in the last step. This concludes the proof.

Appendix B. Tools for proving Theorem 3

B.1. The proof of Lemma 5

Regarding the first statement, we have

where the first inequality uses that holds for all and the second one uses that holds for all x > 0. The second statement is proven as

where the first inequality again uses that holds for all and the second one uses that holds for all x > 0.

B.2. The proof of Lemma 6

We start by noticing that

To handle the second term, we first notice that for any t and k > t, we have

where we used that holds for k > t and . Using this insight, we obtain

Noticing that the last term matches the first term on the right-hand side of Equation (10), the proof is concluded.

B.3. The proof of Lemma 7

The proof of this lemma crucially relies on the following inequality:

Proof We have

Reordering, we obtain

The result follows from noticing that .

where the last inequality uses our assumption on the noise that . Now, using Lemma 9, we obtain

tr

where the last step uses that . This concludes the proof.

B.4. The proof of Lemma 8

We begin by noticing that

holds since the sum only has positive elements. Following Dieuleveut et al. (2017) again, we define the operator T acting on an arbitrary Hermitian operator A as

and also introduce the operator S defined as

so that . We note that S and T are Hermitian and positive definite (the latter being true by our assumption about ). Finally, we define I as the identity operator acting on Hermitian matrices. With this notation, we can write

Thus, defining , we have

Also introducing the operators and as the left- and right-multiplication operators with , respectively, we get after reordering that

Using the fact that and its inverse are Hermitian, we can show

Furthermore, by again following the arguments5 of Dieuleveut et al. (2017, pp. 28), we can also show

Since SW is positive, this leads to the bound

so it remains to bound tr [SW]. On this front, we have

by our assumption on the covariates. Also, by Equation (11), we have

by crucially using the assumption . Plugging into Equation (12) and using again proves the lemma.

Appendix C. Analysis under additive noise

In this section, we consider the “additive-noise” model of Dieuleveut et al. (2017). This model assumes that the learner has access to the gradient estimator , that is, the gradient is subject to the noise vector which doesn’t depend on the parameter vector w. This assumption is admittedly very strong, and we mainly include our analysis for this case for didactic purposes. Indeed, the analysis for this case is significantly simpler than in the setting considered in the main body of our paper.

In this setting, stochastic gradient descent takes the form

which is the unregularized counterpart of the iteration already introduced in Section 2 as Equation (4). Again, we will study the geometric average

For the analysis, we recall the definition and study the evolution of :

We prove the following performance guarantee about this algorithm:

Proposition 10 Suppose that for some and assume that and . Then, the iterates computed by the recursion given by Equations (13) and (14) satisfy

Now, exploiting the assumption that the noise is i.i.d. and zero-mean, we get

The proof is concluded by appealing to Lemma 5.

designed for accessibility and to further open science