Non-strongly-convex smooth stochastic approximation with convergence rate O(1/n)
Francis Bach, Eric Moulines
Introduction
Large-scale machine learning problems are becoming ubiquitous in many areas of science and engineering. Faced with large amounts of data, practitioners typically prefer algorithms that process each observation only once, or a few times. Stochastic approximation algorithms such as stochastic gradient descent (SGD) and its variants, although introduced more than 60 years ago , still remain the most widely used and studied method in this context (see, e.g., ).
We partially understand the properties of that affect the problem difficulty. Strong convexity (i.e., when is twice differentiable, a uniform strictly positive lower-bound on Hessians of ) is a key property. Indeed, after observations and with the proper step-sizes, averaged SGD achieves the rate of in the strongly-convex case , while it achieves only in the non-strongly-convex case , with matching lower-bounds .
The main issue with strong convexity is that typical machine learning problems are high dimensional and have correlated variables so that the strong convexity constant is zero or very close to zero, and in any case smaller than . This then makes the non-strongly convex methods better. In this paper, we aim at obtaining algorithms that may deal with arbitrarily small strong-convexity constants, but still achieve a rate of .
Smoothness plays a central role in the context of deterministic optimization. The known convergence rates for smooth optimization are better than for non-smooth optimization (e.g., see ). However, for stochastic optimization the use of smoothness only leads to improvements on constants (e.g., see ) but not on the rate itself, which remains for non-strongly-convex problems.
We show that for the square loss and for the logistic loss, we may use the smoothness of the loss and obtain algorithms that have a convergence rate of without any strong convexity assumptions. More precisely, for least-squares regression, we show in Section 2 that averaged stochastic gradient descent with constant step-size achieves the desired rate. For logistic regression this is achieved by a novel stochastic gradient algorithm that (a) constructs successive local quadratic approximations of the loss functions, while (b) preserving the same running time complexity as stochastic gradient descent (see Section 3). For these algorithms, we provide a non-asymptotic analysis of their generalization error (in expectation, and also in high probability for least-squares), and run extensive experiments on standard machine learning benchmarks showing in Section 4 that they often outperform existing approaches.
Constant-step-size least-mean-square algorithm
In this section, we consider stochastic approximation for least-squares regression, where SGD is often referred to as the least-mean-square (LMS) algorithm. The novelty of our convergence result is the use of the constant step-size with averaging, leading to rate without strong convexity.
is a -dimensional Euclidean space, with .
The observations are independent and identically distributed.
We study the stochastic gradient (a.k.a. least mean square) recursion defined as
started from . We also consider the averaged iterates .
Assume (A1-6). For any constant step-size , we have
Proof technique.
We adapt and extend a proof technique from which is based on non-asymptotic expansions in powers of . We also use a result from which studied the recursion in Eq. (1), with replaced by its expectation . See the appendix for details.
Optimality of bounds.
Our bound in Eq. (2) leads to a rate of , which is known to be optimal for least-squares regression (i.e., under reasonable assumptions, no algorithm, even more complex than averaged SGD can have a better dependence in ) . The term is also unimprovable.
Initial conditions.
If is small, then the initial condition is forgotten more slowly. Note that with additional strong convexity assumptions, the initial condition would be forgotten faster (exponentially fast without averaging), which is one of the traditional uses of constant-step-size LMS .
Specificity of constant step-sizes.
2 Convergence in higher orders
We are now going to consider an extra assumption in order to bound the -th moment of the excess risk and then get a high-probability bound. Let be a real number greater than .
There exists , and such that, for all , a.s., and
The last condition in Eq. (4) says that the kurtosis of the projection of the covariates on any direction is bounded. Note that computing the constant happens to be equivalent to the optimization problem solved by the FastICA algorithm , which thus provides an estimate of . In Table 1, we provide such an estimate for the non-sparse datasets which we have used in experiments, while we consider only directions along the axes for high-dimensional sparse datasets. For these datasets where a given variable is equal to zero except for a few observations, is typically quite large. Adapting and analyzing normalized LMS techniques to this set-up is likely to improve the theoretical robustness of the algorithm (but note that results in expectation from Theorem 1 do not use ). The next theorem provides a bound for the -th moment of the excess risk.
Assume (A1-7). For any real , and for a step-size , we have:
Note that to control the -th order moment, a smaller step-size is needed, which scales as . We can now provide a high-probability bound; the tails decay polynomially as and the smaller the step-size , the lighter the tails.
For any step-size such that , any ,
Beyond least-squares: M-estimation
Although this algorithm has shown to be the most efficient in practice (see Section 4) we currently have no proof of convergence. Given that the behavior of the algorithms does not change much when the support point is updated less frequently than each iteration, there may be some connections to two-time-scale algorithms (see, e.g., ). In Section 4, we also consider several other strategies based on doubling tricks.
Interestingly, for non-quadratic functions, our algorithm imposes a new bias (by replacing the true gradient by an approximation which is only valid close to ) in order to reach faster convergence (due to the linearity of the underlying gradients).
Relationship with one-step-estimators.
One-step estimators (see, e.g., ) typically take any estimator with -convergence rate, and make a full Newton step to obtain an efficient estimator (i.e., one that achieves the Cramer-Rao lower bound). Although our novel algorithm is largely inspired by one-step estimators, our situation is slightly different since our first estimator has only convergence rate and is estimated on different observations.
1 Self-concordance and logistic regression
is a -dimensional Euclidean space, with .
The observations are independent and identically distributed.
We denote by a global minimizer of , which we thus assume to exist, and we denote by the Hessian operator at a global optimum .
We assume that there exists , and such that , and
We get an convergence rate without assuming strong convexity, even locally, thus improving on results from where the the rate is proportional to . The proof relies on self-concordance properties and the sharp analysis of the Newton step (see appendix).
Experiments
We consider normally distributed inputs, with covariance matrix that has random eigenvectors and eigenvalues , . The outputs are generated from a linear function with homoscedastic noise with unit signal to noise-ratio. We consider and the least-mean-square algorithm with several settings of the step size , constant or proportional to . Here denotes the average radius of the data, i.e., . In the left plot of Figure 1, we show the results, averaged over 10 replications.
Without averaging, the algorithm with constant step-size does not converge pointwise (it oscillates), and its average excess risk decays as a linear function of (indeed, the gap between each values of the constant step-size is close to , which corresponds to a linear function in ).
With averaging, the algorithm with constant step-size does converge at rate , and for all values of the constant , the rate is actually the same. Moreover (although it is not shown in the plots), the standard deviation is much lower.
With decaying step-size and without averaging, the convergence rate is , and improves to with averaging.
Logistic regression.
We consider the same input data as for least-squares, but now generates outputs from the logistic probabilistic model. We compare several algorithms and display the results in Figure 1 (middle and right plots).
On the middle plot, we consider SGD. Without averaging, the algorithm with constant step-size does not converge and its average excess risk reaches a constant value which is a linear function of (indeed, the gap between each values of the constant step-size is close to ). With averaging, the algorithm does converge, but as opposed to least-squares, to a point which is not the optimal solution, with an error proportional to (the gap between curves is twice as large).
On the right plot, we consider various variations of our Newton-approximation scheme. The “2-step” algorithm is the one for which our convergence rate holds ( being the total number of examples, we perform steps of averaged SGD, then steps of LMS). Not surprisingly, it is not the best in practice (in particular at , when starting the constant-size LMS, the performance worsens temporarily). It is classical to use doubling tricks to remedy this problem while preserving convergence rates , this is done in “2-step-dbl.”, which avoids the previous erratic behavior.
We have also considered getting rid of the first stage where plain averaged stochastic gradient is used to obtain a support point for the quadratic approximation. We now consider only Newton-steps but change only these support points. We consider updating the support point at every iteration, i.e., the recursion from Eq. (8), while we also consider updating it every dyadic point (“dbl.-approx”). The last two algorithms perform very similarly and achieve the early. In all experiments on real data, we have considered the simplest variant (which corresponds to Eq. (8)).
2 Standard benchmarks
We have considered 6 benchmark datasets which are often used in comparing large-scale optimization methods. The datasets are described in Table 1 and vary in values of , and sparsity levels. These are all finite binary classification datasets with outputs in . For least-squares and logistic regression, we have followed the following experimental protocol: (1) remove all outliers (i.e., sample points whose norm is greater than 5 times the average norm), (2) divide the dataset in two equal parts, one for training, one for testing, (3) sample within the training dataset with replacement, for 100 times the number of observations in the training set (this corresponds to effective passes; in all plots, a black dashed line marks the first effective pass), (4) compute averaged cost on training and testing data (based on 10 replications). All the costs are shown in log-scale, normalized to that the first iteration leads to .
All algorithms that we consider (ours and others) have a step-size, and typically a theoretical value that ensures convergence. We consider two settings: (1) one when this theoretical value is used, (2) one with the best testing error after one effective pass through the data (testing powers of times the theoretical step-size).
Here, we only consider covertype, alpha, sido and news, as well as test errors. For all training errors and the two other datasets (quantum, rcv1), see the appendix.
We compare three algorithms: averaged SGD with constant step-size, averaged SGD with step-size decaying as , and the stochastic averaged gradient (SAG) method which is dedicated to finite training data sets , which has shown state-of-the-art performance in this set-upThe original algorithm from is considering only strongly convex problems, we have used the step-size of , which achieves fast convergence rates in all situations (see \urlhttp://research.microsoft.com/en-us/um/cambridge/events/mls2013/downloads/stochastic_gradient.pdf).. We show the results in the two left plots of Figure 2 and Figure 3.
Averaged SGD with decaying step-size equal to is slowest (except for sido). In particular, when the best constant is used (right columns), the performance typically starts to increase significantly. With that step size, even after 100 passes, there is no sign of overfitting, even for the high-dimensional sparse datasets.
SAG and constant-step-size averaged SGD exhibit the best behavior, for the theoretical step-sizes and the best constants, with a significant advantage for constant-step-size SGD. The non-sparse datasets do not lead to overfitting, even close to the global optimum of the (unregularized) training objectives, while the sparse datasets do exhibit some overfitting after more than 10 passes.
Logistic regression.
We also compare two additional algorithms: our Newton-based technique and “Adagrad” , which is a stochastic gradient method with a form a diagonal scalingSince a bound on is not available, we have used step-sizes proportional to . that allows to reduce the convergence rate (which is still in theory proportional to ). We show results in the two right plots of Figure 2 and Figure 3.
Averaged SGD with decaying step-size proportional to has the same behavior than for least-squares (step-size harder to tune, always inferior performance except for sido).
SAG, constant-step-size SGD and the novel Newton technique tend to behave similarly (good with theoretical step-size, always among the best methods). They differ notably in some aspects: (1) SAG converges quicker for the training errors (shown in the appendix) while it is a bit slower for the testing error, (2) in some instances, constant-step-size averaged SGD does underfit (covertype, alpha, news), which is consistent with the lack of convergence to the global optimum mentioned earlier, (3) the novel Newton approximation is consistently better.
On the non-sparse datasets, Adagrad performs similarly to the Newton-type method (often better in early iterations and worse later), except for the alpha dataset where the step-size is harder to tune (the best step-size tends to have early iterations that make the cost go up significantly). On sparse datasets like rcv1, the performance is essentially the same as Newton. On the sido data set, Adagrad (with fixed steps size, left column) achieves a good testing loss quickly then levels off, for reasons we cannot explain. On the news dataset, it is inferior without parameter-tuning and a bit better with. Adagrad uses a diagonal rescaling; it could be combined with our technique, early experiments show that it improves results but that it is more sensitive to the choice of step-size.
Overall, even with and very large (where our bounds are vacuous), the performance of our algorithm still achieves the state of the art, while being more robust to the selection of the step-size: finer quantities likes degrees of freedom should be able to quantify more accurately the quality of the new algorithms.
Conclusion
In this paper, we have presented two stochastic approximation algorithms that can achieve rates of for logistic and least-squares regression, without strong-convexity assumptions. Our analysis reinforces the key role of averaging in obtaining fast rates, in particular with large step-sizes. Our work can naturally be extended in several ways: (a) an analysis of the algorithm that updates the support point of the quadratic approximation at every iteration, (b) proximal extensions (easy to implement, but potentially harder to analyze); (c) adaptive ways to find the constant-step-size; (d) step-sizes that depend on the iterates to increase robustness, like in normalized LMS , and (e) non-parametric analysis to improve our theoretical results for large values of .
This work was partially supported by the European Research Council (SIERRA Project 239993). The authors would like to thank Simon Lacoste-Julien and Mark Schmidt for discussions related to this work.
In the appendix we provide proofs of all three theorems, as well as additional experimental results (all training objectives, and two additional datasets quantum and rcv1).
Appendix A Proof of Theorem 1
We first denote by the deviation to . Since we consider quadratic functions, it satisfies a simplified recursion:
We also consider the averaged iterate. We have and .
The crux of the proof is to consider the same recursion as Eq. (12), but replacing by its expectation (which is related to fixed design analysis in linear regression). This is of course only an approximation, and thus one has to study the remainder term; it happens to satisfy a similar recursion, on which we can apply the same technique, and so on. This proof technique is taken from . Here we push it to arbitrary orders with explicit constants for averaged constant-step-size stochastic gradient descent.
We denote by the -algebra generated by . Both and are -measurable.
A.1 Two main lemmas
The proof relies on two lemmas, one that provides a weak result essentially equivalent (but more specific and simpler because the step-size is constant) to non-strongly-convex results from , and one that replaces by its expectation in Eq. (12), which may then be seen as a non-asymptotic counterpart to the similar set-tup in .
Proof. We follow the proof technique of (which relies only on smoothness) and get:
This leads to the desired result, because, by convexity, .
Proof. The proof relies on the fact that cost functions are quadratic and our recursions are thus linear, allowing to obtain in closed form. The sequence satisfies a linear recursion, from which we get, for all :
which leads to the first result using classical martingale second moment expansions (which amount to considering , independent, so that the variance of the sum is the sum of variances). Moreover, using the identity \sum_{k=0}^{n-1}(I-\gamma H)^{k}=\big{(}I-(I-\gamma H)^{n}\big{)}\big{(}\gamma H\big{)}^{-1}, we get:
We then get, using standard martingale square moment inequalities (which here also amount to considering , independent, so that the variance of the sum is the sum of variances):
because for all , (see Lemma 3 in Section A.6), and the second term is the sum of terms which are all less than .
Note that we may replace the term by , which is only interesting when is small.
A.2 Proof principle
The proof relies on an expansion of and as polynomials in due to . This expansion is done separately for the noise process (i.e., when assuming ) and for the noise-free process that depends only on the initial conditions (i.e., when assuming that ). The bounds may then be added.
Indeed, we have , with and , and thus \bar{\eta}_{n}=\frac{1}{n+1}\sum_{i=0}^{n}\bigg{[}M_{1}^{i}\eta_{0}+\gamma\sum_{k=1}^{i}M_{k+1}^{i}\xi_{k}\bigg{]}=\frac{1}{n+1}\sum_{i=0}^{n}M_{1}^{i}\eta_{0}+\gamma\sum_{k=1}^{n}\bigg{(}\sum_{i=k}^{n}M_{k+1}^{i}\bigg{)}\xi_{k}, leading to
for any for which it is defined: the left term depends only on initial conditions and the right term depends only on the noise process (note the similarity with bias-variance decompositions).
A.3 Initial conditions
In this section, we assume that is uniformly equal to zero, and that .
We thus have and thus
By taking expectations (first given , then unconditionally), we get:
from which we obtain, by summing from to and using convexity (note that Lemma 1 could be used directly as well):
Here, it would be interesting to explore conditions under which the initial conditions may be forgotten at a rate , as obtained by in the strongly convex case.
A.4 Noise process
In this section, we assume that and (which implies ). Following , we recursively define the sequences for (and their averaged counterparts ):
The sequence is defined as and for , .
The sequence is defined from as and, for all :
We now show that the sequence then satisfies the following recursion, for any (which is of the same type than ):
In order to prove Eq. (16) by recursion, we have, for ,
Bound on covariance operators.
We now show that we also have a bound on the covariance operator of , for any and :
In order to prove Eq. (17) by recursion, we get for :
In order to go from to , we have, using Lemma 2 and the fact that and are independent:
Putting things together.
We may apply Lemma 1 to the sequence \big{(}\eta_{n}-\sum_{i=0}^{r}\eta_{n}^{i}\big{)}, to get
We may now apply Lemma 2 to Eq. (15), to get, with a noise process which is such that
We thus get, using Minkowski’s inequality (i.e., triangle inequality for the norms ):
This implies that for any , we obtain, by letting tend to :
A.5 Final bound
We can now take results from Appendices A.3 and A.4, to get
A.6 Proof of Lemma 3
In this section, we state and prove a simple lemma.
For any and , .
Proof. Since , we have, . Moreover, , and by integrating between and , we get . By multiplying the two previous inequalities, we get the desired result.
Appendix B Proof of Theorem 2
We use the same notations than the proof of Theorem 1, and the same proof principle: (a) splitting the contributions of the initial conditions and the noise, (b) providing a direct argument for the initial condition, and (c) performing an expansion for the noise contribution.
B.1 Contribution of initial conditions
When the noise is assumed to be zero, we have almost surely, and thus, since , almost surely, and
and . We may now apply the Burkholder-Rosenthal-Pinelis inequality in Eq. (B), to get:
We have used above that (a) and that (b) \big{\|}A_{n}\big{\|}_{p/2}\leqslant\big{\|}A_{n}\big{\|}_{p}. This leads to
Finally, we obtain, for any
i.e., by a change of variable , for any , we get
By using monotonicity of norms, we get, for any :
Note that the constants in the bound above could be improved by using a proof by recursion.
B.2 Contribution of the noise
We follow the same proof technique than for Theorem 1 and consider the expansion based on the sequences , for . We need (a) bounds on , (b) a recursion on the magnitude (in norm) of and (c) a control of the error made in the expansions.
We start by a lemma similar to Lemma 2 but for all moments. This will show a bound for the sequence .
Proof. We have, from the proof of Lemma 2:
with \beta_{j}=\big{(}{I-(I-\gamma H)^{n-j}}\big{)}\big{(}{\gamma H}\big{)}^{-1}H^{1/2}\xi_{j}. We have
using Lemma 3 in Section A.6, and assumption (A7).
Using Burkholder-Rosenthal-Pinelis inequality in Eq. (B), we then obtain
Following the same proof technique as above, we have
from which we get, for any positive semidefinite operator such that , using BRP’s inequality:
We introduce the following quantity to control the deviations of :
We have from Eq. (20), .
Since , for all , we have the closed form expression
and we may use BRP’s inequality in Eq. (B) to get, for any such that :
This implies that A_{r}\leqslant A_{0}\big{(}\sqrt{p\gamma R^{2}\kappa}+p\gamma R^{2}+p(\gamma R^{2})^{1-1/p}\big{)}^{r}, which in turn implies, from Eq. (20),
The condition on will come from the requirement that .
leading to, using BRP’s inequality in Eq. (B), similar arguments than in the previous bounds, \big{(}{I-(I-\gamma H)^{n-j}}\big{)}^{2}\big{(}{\gamma H}\big{)}^{-2}H\preccurlyeq H^{-1}\gamma and \big{(}{I-(I-\gamma H)^{n-j}}\big{)}^{2}\big{(}{\gamma H}\big{)}^{-2}H\preccurlyeq\frac{n}{\gamma}I:
We may then impose a restriction on , i.e., with . We then have
With , we obtain a bound of above.
From Eq. (16) and the fact that almost surely, we get:
Putting things together.
We get by combining Lemma 4 with Eq. (22) and Eq. (23) and then letting tends to infinity,
B.3 Final bound
For , we obtain, from the last equations of Section B.1 and Section B.2,
Moreover, when , we have:
B.4 Proof of Corollary 1
We have from the previous proposition, for :
with , and and .
This leads to, using Markov’s inequality:
Thus the large deviations decay as power of , with a power that decays as . If is small, the deviations are lighter.
In order to get the desired result, we simply take .
Appendix C Proof of Theorem 3
In terms of convergence rates, will be -optimal, while will be -optimal for the quadratic problem because of previous results on averaged LMS. A classical property is that a single Newton step squares the error. Therefore, the full Newton step should have an error which is the square of the one of , i.e., . Overall, since approaches the full Newton step with rate , this makes a bound of .
In Section C.1, we provide a general deterministic result on the Newton step, while in Section C.2, we combine with two stochastic approximation results, making the informal reasoning above more precise.
In this section, we study the effect of an approximate Newton step. We consider , the Newton iterate , and an approximation of . In the next proposition, we provide a bound on , under different conditions, whether is close to optimal for , and/or is close to optimal for the quadratic approximation around (i.e., close to ). Eq. (25) corresponds to the least-favorable situations where both errors are small, while Eq. (26) and Eq. (27) consider cases where is sufficiently good. See proof in Section E.3. These three cases are necessary for the probabilistic control.
Assume (B3-4), and such that , and . Then, if ,
If , then
Moreover, if and , then
Note that in the favorable situation in Eq. (26), we get error of the form . It essentially suffices now to show that in our set-up, in a probabilistic sense to be determined, and , while controlling the unfavorable situations.
C.2 Stochastic analysis
We consider the following two-step algorithm:
Starting from any initialization , run iterations of averaged stochastic gradient descent to get ,
Run from steps of LMS on the quadratic approximation around , to get , which is an approximation of the Newton step .
We denote by the -field generated by the first observations (the ones used to define ). We have, by separating all events, i.e., using :
We now need to control , i.e., the error made by the LMS algorithm started from .
We consider the quadratic approximation around , and write is as an expectation, i.e.,
almost surely.
We may thus apply the previous results, i.e., Theorem 1, to obtain with the LMS algorithm a such that, with :
which are such that and if .
The last inequalities may be checked graphically.
We now need to use bounds on the behavior of the first steps of regular averaged stochastic gradient descent.
Fine results on averaged stochastic gradient descent.
In order to get error bounds on , we run steps of averaged stochastic gradient descent with constant-step size . We need the following bounds from [21, Appendix E and Prop. 1]:
However, we need a finer control of the deviations in order to bound quantities of the form . In Section D, extending results from , we show in Prop. 3 that if , then .
Putting things together.
From Eq. (30), we then get, if :
Using the notation , we obtain:
The condition is implied by .
Appendix D Higher-order bounds for stochastic gradient descent
In this section, we provide high-order bounds for averaged stochastic gradient for logistic regression. The first proposition gives a finer result than , with a simpler proof, while the second proposition is new.
Proof. Following , we have the recursion:
Thus if B=\bigg{\|}\sup_{k\in\{0,\dots,n\}}A_{k}\bigg{\|}_{p}, we have
By solving this quadratic inequality, we get:
The previous statement leads to the desired result if . For , we may bound it by the value at , and a direct calculation shows that the bound is still correct.
Proof. Using that almost surely, we obtain that almost surely .
Moreover, from the previous proposition, we have for ,
For , we get:
This leads to the bound valid for all :
Appendix E Properties of self-concordance functions
In this section, we review various properties of self-concordant functions, that will prove useful in proving Theorem 3. All these properties rely on bounding the third-order derivatives by second-order derivatives. More precisely, from assumptions (B3-4), we have for any , where denotes the -th order differential of :
In this section, we derive global non-asymptotic Taylor expansions for self-concordant functions, which show that they behave similarly to like quadratic functions.
The following proposition shows that having a small excess risk implies that the weighted distance to optimum is small. Note that for quadratic functions, these two quantities are equal and that throughout this section, we always consider norms weighted by the matrix (Hessian at optimum).
Assume (B3-4). Then, for any :
Proof. Let \varphi:t\mapsto f\big{[}\theta_{\ast}+t(\theta-\theta_{\ast})\big{]}. Denoting , we have:
from which we obtain . Following , by integrating twice (and noting that and ), we get
The identity is equivalent to , for . It then suffices to show that . This can be shown by proving the monotonicity of , and we leave this exercise to the reader.
The next proposition shows that Hessians between two points which are close in weighted distance are close to each other, for the order between positive semi-definite matrices.
Assume (B3-4). Then, for any :
Proof. Let and . We have:
with . Thus . This implies, for , that
which implies the desired results. denotes the operator norm (largest singular value).
The following proposition gives an approximation result bounding the first order expansion of gradients by the first order expansion of function values.
Assume (B3-4). Then, for any and :
Proof. Let . We have and . This leads to
The following proposition considers a global Taylor expansion of function values. Note that when tends to zero, we obtain exactly the second-order Taylor expansion. For more details, see . This is followed by a corrolary that upper bounds excess risk by distance to optimum (this is thus the other direction than Prop. 4).
Assume (B3-4). Then, for any and :
Proof. Let \varphi(t)=f\big{[}\theta_{1}+t(\theta_{2}-\theta_{1})\big{]}-f(\theta_{1})-t\langle f^{\prime}(\theta_{1}),\theta_{2}-\theta_{1}\rangle. We have \varphi^{\prime}(t)=\langle f^{\prime}\big{[}\theta_{1}+t(\theta_{2}-\theta_{1})\big{]},\theta_{2}-\theta_{1}\rangle-\langle f^{\prime}(\theta_{1}),\theta_{2}-\theta_{1}\rangle and \varphi^{\prime\prime}(t)=\langle\theta_{2}-\theta_{1},f^{\prime\prime}\big{[}\theta_{1}+t(\theta_{2}-\theta_{1})\big{]}(\theta_{2}-\theta_{1})\rangle. Moreover, , leading to . Integrating twice between 0 and 1 leads to the desired result.
Assume (B3-4), and and . Then
where .
Proof. Applying Prop. 7 to and , we get the desired result.
The following proposition looks at a similar type of bounds than Prop. 7; it is weaker when and are close (it does not converge to the second-order Taylor expansion), but stronger for large values (it does not grow exponentially fast).
Assume (B3-4), and . Then
E.2 Analysis of Newton step
Self-concordance has been traditionally used in the analysis of Newton’s method (see ). In this section, we adapt classical results to our specific notion of self-concordance (see also ). A key quantity is the so-called “Newton decrement” at a certain point , equal to , which governs the convergence behavior of Newton methods (this is the quantity which is originally shown to be quadratically convergent). In this paper, we consider a slightly different version where the Hessian is chosen to be the one at , i.e., .
The following proposition shows how a full Newton step improves the Newton decrement (by taking a square).
Assume (B3-4), and and . Then
where and .
Proof. When applying the two previous propositions to the Newton step , we get:
We then optimize with respect to to obtain the desired result.
The following proposition shows how the Newton decrement is upper bounded by a function of the excess risk.
Assume (B3-4), and , then,
with and .
Proof. We may bound the Newton decrement as follows:
The following proposition provides a bound on a quantity which is not the Newton decrement. Indeed, this is (up to the difference in the Hessians), the norm of the Newton step. This will be key in the following proofs.
Assume (B3-4), and , then,
with .
The next proposition shows that having a small Newton decrement implies that the weighted distance to optimum is small.
Assume (B3-4). If we have , with , then
Proof. For any such that , and , we have, following the same reasoning than for Prop. 7:
with and
This implies that if , for , . Thus,
Note that the quantity appears twice in the result above.
E.3 Proof of Prop. 1
In this section, we prove Prop. 1 using tools from self-concordance analysis. These tools are described in the previous Sections E.1 and E.2. In order to understand the proof, it is preferable to read these sections first.
We use the notation . We then get from Prop. 4.
Moreover, we have, also from Prop. 8, , and using Prop. 5, we get
We may now use Prop. 10 and use the bound:
Combining with Eq. (46) and Eq. (47), we get
Proof of Eq. (26) and Eq. (27).
For these two inequalities, the starting point is the same. Using Eq. (48) (i.e., the Newton decrement at ), we first show that the distances and are bounded. Using (Prop. 5):
Now, we can bound the Newton decrement at , using Prop. 9:
Thus, using Prop. 12, if , then
where .
We now have two separate paths to obtain Eq. (26) and Eq. (27).
If we assume that is bounded, i.e., with and , then, one can check computationally that we obtain and thus below:
If we only assume bounded, then we have (from the beginning of the proof):
because we may use the earlier reasoning with . This is Eq. (26).
Appendix F Additional experiments
In Table 2, we describe the datasets we have used in experiments and where they were downloaded from.
In Figure 4, we provide similar results than in Section 4, for two additional datasets, quantum and rcv1, while in Figure 5, Figure 6 and Figure 7, we provide training objectives for all methods. We can make the following observations:
For non-sparse datasets, SAG manages to get the smallest training error, confirming the results of .
For the high-dimensional sparse datasets, constant step-size SGD is performing best (note that as shown in Section 3, it is not converging to the optimal value in general, this happens notably for the alpha dataset).