Low-depth gradient measurements can improve convergence in variational hybrid quantum-classical algorithms
Aram Harrow, John Napp
Introduction
where is the Hermitian operator which generates pulse . This is the form of variational state most commonly encountered in the literature on variational algorithms, and is also motivated theoretically [MRBAG16, YRS+17, BJ18]. Note that in the presence of noise, the parameterized family of states which may be prepared will actually consist of mixed states. We only consider the noiseless case in this paper. It may also be the case that there are more pulses than independent parameters. For instance, one may impose a constraint like . We assume for simplicity that the parameters are independent, but comment on how our results could be easily extended to this case.
It is assumed that the quantum device is controlled by a classical “outer loop”, and the quantum device is used only for preparing simple quantum states and making simple measurements. The classical outer loop uses this measurement information to perform a classical optimization of some function over the feasible set , where the objective function is induced by some Hermitian objective observable , via the relation . Algorithms of this family have been proposed in the context of quantum simulation (e.g. variational quantum eigensolvers [PMS+14, WHT15]), combinatorial optimization (e.g. QAOA [FGG14]), and machine learning (e.g. quantum classifiers [FN18, MNKF18, SK18, SBSW18, HCT+18]).
As a simple example, in a simulation context, could be some physical Hamiltonian for which we want to approximately obtain the ground state energy. If the true ground state (or a state close to the true ground state) belongs to the parameterized family for , approximately minimizing yields an approximation to the ground state energy. The typical way a variational algorithm would obtain information about with little quantum resources is by expanding where are tensor products of Pauli operators (recall that any operator may be expanded in this way), or are some other observables which may be easily measured. In this paper, we assume the are products of Pauli operators. Assuming it is possible to easily measure these operators, one can estimate by estimating each separately and combining the results according to the coefficients . Of course, due to the randomness of the measurement outcomes, many preparations of and measurements may be required to obtain a good estimate of .
This effect occurs generally in variational hybrid algorithms: the randomness of the quantum measurement outcomes translates into the classical outer loop only having stochastic access to the objective function . That is, at point in parameter space, it cannot directly observe the function value , but rather some random variable whose expectation value is . Numerical simulations that overlook this fact may be misleading. For instance, letting denote the optimization error, some problems that admit convergence rates given noiseless access to the objective function values admit convergence rates in the stochastic setting [Bub15].
But even worse for prospects of optimization, the resulting classical stochastic optimization problem will generally be complicated and nonconvex, and hence may be intractable. However, one can hope for heuristics which find a reasonable approximate solution. Furthermore, if the algorithm is in a convex vicinity of an optimum, or local optimum, algorithms like stochastic gradient descent which are known to converge for convex problems may converge to the local optimum despite the problem being globally nonconvex. For some proposed applications of variational algorithms one desires a very precise solution, so it is likely that much time will be spent converging in the vicinity of an optimum and this situation may be especially relevant. In this paper, we focus on this latter scenario of convergence within a convex region containing a local optimum, either assuming or proving that this case applies.
In the usual formulation of variational algorithms, the classical outer loop is assumed to take some number of quantum measurements at a point in parameter space in order to approximate the objective function value at that point, and then perform an optimization based on these values. However, one can imagine more complicated algorithms which, instead of taking measurements to estimate at point , take measurements corresponding to some other property of the optimization problem. Indeed, a natural alternative choice is to take measurements corresponding to , the gradient of the objective function at point . Such a strategy was proposed in the context of combinatorial optimization [GS17], quantum chemistry [RBM+18], and machine learning [MNKF18, SBSW18, FN18]. Similar ideas were also proposed in the context of implementing Hamiltonian evolution in a low-depth variational setting [LB17]. Low-depth procedures for directly measurement gradients in variational algorithms typically require only marginally greater quantum circuit depth than that required for measuring the objective function. Some such methods are reviewed and extended in [BIS+18, SBG+18].
However, it was not clear that such gradient-measurement based strategies could confer any advantage over objective-measurement based strategies. For one, note that it is possible to obtain an estimate of using only estimates of the objective function . To see how, note that for small ,
2 Summary of results
In order to rigorously prove a lower bound on the number of quantum measurements required for zeroth-order variational optimization, it is convenient to introduce a black-box formalism. To motivate the introduction of a black-box formalism, note that any variational algorithm can be simulated by a purely classical algorithm that makes zero quantum measurements. Of course, the time complexity of the classical simulation may be exponential in the problem size.
One encounters a similar problem in the classical setting of trying to quantify the complexity of optimization. The objective function to be minimized could be extremely complicated and difficult to study analytically (for instance, it could correspond to the output of some complicated algorithm) or otherwise inaccessible. A black-box model was therefore developed for the study of convex optimization [NY83] and remains popular in current research in the field. In this setting, the function to be optimized is encoded in an oracle, and we define the query complexity of an algorithm for optimizing the function to be the number of calls made to the oracle. The algorithm may be promised that the objective function has certain properties, but is not given an exact description of the function. This black-box formalism provides a natural and general setting for proving bounds in convex optimization.
It should be noted that, when variational algorithms are used in practice, commuting terms in the Pauli decomposition of can be measured on a single trial state. To simplify the analysis, our black-box model does not take advantage of this possible speedup. Hence, the query cost in our model corresponds to the number of products of Pauli operators measured, rather than the number of state preparations required, which is a related quantity but could be lower. (We comment on how taking this into account would affect our bounds for the toy model we study in Table 2.)
We consider a restriction of the variational problem to a convex region of parameter space on which the objective function is assumed to be convex. We import known results from the stochastic optimization literature to obtain convergence rates for various optimization strategies and for various assumptions about , such as strong convexity. We derive upper bounds on the query cost for strategies that use analytic gradient measurements in conjunction with stochastic gradient descent or stochastic mirror descent with an setup. The upper bounds are functions of the dimension of parameter space , precision , geometry of the feasible set, and coefficients in the Pauli expansions of the objective observable and pulse generators.
Stochastic mirror decent (see Appendix A.2, or e.g. [NJLS09, JN11, Bub15] for more detailed reviews), which we will abbreviate as SMD, can be thought of as a generalization of SGD to non-Euclidean spaces. As motivation for why SMD might be relevant in our setting, note that the 1-norm of a parameter vector has a very natural interpretation. Namely, since is essentially the unitary evolution generated by for time , may be interpreted as the total amount of time that the starting state is evolved for to reach the trial state , and is associated with the amount of time for which the associated pulse sequences differ. On the other hand, SGD is appropriate for Euclidean geometriesa. Taking our norm to be instead of and using a suitable version of SMD yields nearly quadratically better scaling with respect to the dimension of parameter space in some settings, as compared with SGD. In other settings, SGD outperforms SMD. We record upper bounds based on both strategies. It is an open problem to understand what sort of values the parameters in the upper bounds typically take in practice, and relatedly, whether a Euclidean geometry or some other geometry is more appropriate.
For comparison, we also record a rigorous upper bound for zeroth-order strategies by applying two of the best known zeroth-order upper bounds [FKM05, AFH+11] from the stochastic optimization literature. Our results on general upper bounds are displayed in Table 1. It should be noted that these zeroth-order upper bounds are the best rigorous zeroth-order upper bounds that we are aware of, but it is likely that other derivative-free algorithms would significantly outperform these bounds in practice in many instances. For instance, methods based on trust regions and surrogate models [CSV09] often perform very well in practice, despite not necessarily having strong theoretical guarantees.
This point is an example of a more general limitation that applies to all of the upper bounds we report, relating to the difference between theoretical and empirical results. For one, these are upper bounds for convergence that apply when the algorithm has a trusted region which it knows contains a local optimum, and which the objective function is convex over. Furthermore, while strong convexity of the objective function in the domain can greatly improve the convergence, the rigorous upper bounds that exploit this property apply when the algorithm has a good estimate of this strong convexity parameter. In other language, the upper bounds apply in a promise setting, where the algorithm is promised that a certain convex subset of parameter space contains the optimum, the objective function is convex in this domain, and any other relevant parameters take certain values. However, this may very well not be the case in practice. Furthermore, these are worst-case upper bounds, and may be outperformed in practice.
All of these challenges are well-known in machine learning tasks such as deep learning, in which one may want to converge to a local optimum of some complicated nonconvex optimization problem. In practice, one can do hyperparameter optimization, which would involve yet another classical outer loop varying over the parameters used to define the optimization algorithm itself. Parameters could also be set adaptively (e.g. [KB14]). Another possibility is that the algorithm could try to construct a surrogate model for the objective function which is valid in some region, and estimate relevant parameters from the surrogate model. Unfortunately, such methods are often not backed up by strong theoretical results, even when the practical performance is very good. While there may be a sizable gap between theory and practice, we hope that the theoretical convergence upper bounds will nonetheless be useful for guiding practical implementations and expectations.
Finally, we point out that these upper bounds do not explicitly depend on the number of terms in the Pauli expansion of the objective observable or pulse generators, but rather on sums of coefficients in the expansion. This is a desirable feature for applications such as quantum chemistry, where it is often the case that the electronic structure Hamiltonian is written as a sum of a very large number of terms, many of which have very small norm.
After establishing an oracular setting and recording some upper bounds, for a given parameter we define a certain class of simple 1-local objective observables on qubits which we use to demonstrate a separation in query complexity between variational algorithms which only make zeroth-order queries to the oracle and those which make first-order queries. The optima of the observables in are -close to each other in objective function value, in the sense that for all , if is an optimum (i.e. ground state) of , then , where is the smallest eigenvalue of .
We show that for any precision parameter , any zeroth-order variational algorithm which optimizes any observable to precision and only queries the oracle with states in the vicinity of the optimum of must make at least queries. Here, the “vicinity of the optimum” is essentially the set of states that are -close to optimal in objective function value. On the other hand, we show that after making a good choice of variational ansatz, an SGD algorithm that performs analytic gradient measurements to get gradient estimates optimizes any objective observable in the family to expected precision with only queries of states in the vicinity of the optimum.
Suppose is a variational algorithm that only makes zeroth-order queries to , only queries states in the vicinity of the optima of , and for any that is realized, outputs a description of a state whose expected objective function value is -close to . Then must make queries.
There exists a variational algorithm that only makes first-order queries to , only queries states in the vicinity of the optima of , and for any , makes queries and outputs a description of a state whose expected objective function value is -close to . An algorithm that achieves this rate is a simple stochastic gradient descent strategy.
Suppose is a variational algorithm that may make queries of any order to , may query the oracle with any state, and for any outputs a description of a state whose expected objective function value is -close to the optimum. Then makes at least queries.
We summarize the above results in Table 2, where for comparison we also include a rigorous zeroth-order upper bound based on the results of [FKM05] and [AFH+11].
3 Related work
In this paper, we are primarily interested in the low-depth setting. The methods we consider for measuring the gradient yield an unbiased, but possibly very noisy estimate of the gradient in low depth. An alternative approach for measuring the gradient in variational algorithms was recently proposed in [GAW17], which builds on Jordan’s gradient measurement algorithm [Jor05]. Their algorithm offers significantly better performance for obtaining precise estimates of the gradient, but also requires significantly more quantum resources, with coherence time requirements increasing with the desired precision.
A lower bound for a class of derivative-free stochastic convex optimization problems was shown in [JNR12]. Our separation result in Section 5 is similar in spirit to their result, and the proof strategies share similarities. However, our setting of variational hybrid algorithms is very different from theirs, preventing their result from being ported to variational quantum algorithms. In particular, we are interested specifically in stochastic optimization problems induced by a quantum observable and variational ansatz. Furthermore, in our setting the classical outer loop’s optimization problem is not fixed; different variational ansätze induce different optimization problems. Our lower bound for zeroth-order algorithms takes this extra freedom into account, applying for any choice of variational ansatz (and also allowing the algorithm to change ansatz over the course of the optimization). Our proof strategy for the zeroth-order lower bound also borrows some techniques from [AWBR09], which showed lower bounds for classes of first-order stochastic optimization problems. In turn, these techniques are inspired by methods in statistical minimax and learning theory.
4 Organization
In Section 2, we state the conventions we adhere to and record known results from stochastic convex optimization that we later use. In Section 3, we introduce a black-box setting for variational algorithms and define the oracle encoding an objective observable . In Section 4, we present general upper bounds on the query cost of optimization for different algorithms and different assumptions on the objective function. In Section 5, we define a parameterized class of variational optimization problems on qubits and use this class of problems to prove a query complexity separation between zeroth-order and first-order optimization strategies. We conclude and mention some open questions in Section 6. Appendix A contains some relevant background on first-order stochastic convex optimization algorithms.
Preliminaries
We will assume throughout that variational states are parameterized according to an ansatz of the form
Given a parameterization and a feasible set , the classical objective function to be minimized is induced by some Hermitian operator , which we refer to as the objective observable. In particular, the classical objective function is given by
In the context of variational algorithms, it is assumed that the quantum device is capable of measuring some subset of quantum observables. We assume that the set of observables which may be measured is the set of all Pauli operators.
We collect notation and parameters in Table 3.
2 Requisite results about stochastic convex optimization
We will obtain upper bounds for variational algorithms in convex regions by combining well known classical convergence results with sampling strategies for estimating the gradient. Here, we record the classical optimization results we will need. Background on stochastic gradient descent and stochastic mirror descent may be found in Appendix A (see e.g. [NJLS09, JN11, Bub15] for more thorough reviews). First, we define strong convexity.
For , the real-valued function is -strongly convex with respect to norm on some convex domain if ,
Note that a twice-differentiable function is -strongly convex with respect to the -norm if all of the eigenvalues of the Hessian matrix at each point in the domain are at least . More generally, is -strongly convex at w.r.t. an arbitrary norm if for all , where is the Hessian of at . In contrast, is convex if the Hessians are merely positive semidefinite. Intuitively, if is strongly convex, then it is lower bounded by a quadratic function. Strong convexity can often be used to accelerate optimization [Bub15].
In this section, we record known upper bounds for optimizing convex functions given access to noisy, unbiased gradient information. For the first two results below, we follow the presentation of the review on algorithms for convex optimization [Bub15].
Assume is contained in a Euclidean ball of radius and is convex on . Then projected SGD with fixed step size satisfies
where is the starting point, and the algorithm visits points .
Assume is -strongly convex on with respect to . Then SGD with step size at iteration satisfies
where is the starting point, and the algorithm visits points .
Assume is contained in a 1-ball of radius , and is convex on . Then stochastic mirror descent with an appropriate setup and and step size , satisfies
where is the starting point, and the algorithm queries points .
Assume is -strongly convex on with respect to norm . Then a certain SMD-like algorithm, running for iterations, outputs a (random) vector such that
2.2 Upper bounds for stochastic zeroth-order (derivative-free) optimization
Compared to stochastic first-order optimization, less is known about rigorous upper bounds for stochastic zeroth-order optimization. The few rigorous upper bounds for zeroth-order optimization that are known [FKM05, AD10, AFH+11, JNR12, Sha13] are usually weaker than their first-order counterparts. The upper bounds we use in this paper are from [FKM05] and [AFH+11], in which the authors prove rigorous upper bounds for stochastic zeroth-order convex optimization in which the expected error in objective function value converges to zero like and , respectively, where is the number of iterations. We record their results below, adapted for our purposes. Note that in these papers, the authors state the results in terms of online optimization. However, it is straightforward to convert these into results for stochastic optimization. Similar adaptations of these same results are also noted in [JNR12] and [Sha13].
There also exists an algorithm [AFH+11] that makes queries and outputs such that
Note that this implies that queries are needed to optimize to expected precision . Other rigorous upper bounds for derivative-free stochastic convex optimization exist [AD10, JNR12, Sha13], which apply in settings which are less applicable to this paper, or to specific families of functions.
Black-box formulation
One of our main goals is to prove a rigorous lower bound on the number of quantum measurements required to minimize an objective function to within some precision . However, examining the formulation of this question already reveals a subtlety. Namely, a classical computer can simulate the quantum part of a variational algorithm in (in general) exponential time, which implies that any variational algorithm can be simulated with a variational algorithm which makes zero quantum measurements.
Of course, in practice it would generally be intractable for a classical computer to simulate a variational algorithm. In fact, it is known that the existence of an efficient classical algorithm for sampling from the output distribution of the commonly-considered variational algorithm QAOA would imply a collapse of the polynomial hierarchy [FH16]. We therefore seek a formulation of the problem which better captures the behavior of realistic algorithms. One way to do this is to strip away the classical outer loop’s knowledge of the specific objective observable that it is trying to optimize by encoding the observable in the black box, and only giving the outer loop the black box and a promise that belongs to some particular family . In practice, this essentially means that the black-box picture is applicable when the classical optimization algorithm that the outer loop runs does not depend on the details of the objective observable itself (but could depend on the family ), but is rather some more general-purpose algorithm like gradient descent, Nelder-Mead, SPSA, BFGS, etc. We are not aware of any proposed variational algorithm that is not in this class. Note that, while the classical optimization component of a variational algorithm does not exploit detailed structure of the objective observable in current proposals, the variational ansatz sometimes does (e.g. [PMS+14, FGG14, WHT15]). For example, in QAOA, the ansatz involves pulses of the form where is the objective observable. For such cases, our query upper bounds are still fully applicable. Our lower bound for zeroth-order variational optimization is ansatz-independent in the sense that for any ansatz the algorithm chooses, the lower bound still holds. Hence, the zeroth-order lower bound is still fully applicable in the setting where the ansatz may be a function of . However, there is a subtlety that if the algorithm is promised that the ansatz is a certain function of , it could use this information to learn the ground state of with fewer queries, and the lower bound may no longer apply. Essentially, this means that the zeroth-order lower bound applies for zeroth-order algorithms which do not cleverly exploit the dependence of the variational ansatz on the objective observable. This includes all “general-purpose” zeroth-order methods such as SPSA, Nelder-Mead, and gradient descent with gradients estimated via finite-differences, to name a few.
In the remainder of this section, we define our black-box formulation of variational algorithms. Specifically, we define an oracle encoding the objective observable . This definition allows us to talk about the query complexity of optimizing some family of objective observables . The classical outer loop is given as input an oracle under the promise that , and attempts to find an approximate optimum using as few queries as possible.
If the parameterization is clear from context, we may not explicitly note that is provided as input to the oracle. If we speak of querying the oracle with a state , we mean querying the oracle with a parameterization and parameter vector that describe the state . In the remainder of this section, we first define how the oracle behaves for zeroth-order queries. We then briefly review how to measure gradients (and higher-order derivatives) in low depth in variational algorithms, and then define the behavior of the oracle upon first- and higher-order queries.
Let be some objective observable. Decompose into a linear combination of products of Pauli operators as where . (The coefficients may all be assumed to be positive by absorbing the phase into the operator.) Now, defining the normalization factor and the probability distribution , we may write
From this expression, it is clear that by sampling index with probability , measuring with respect to the state , and then multiplying the outcome by we obtain an unbiased estimator for . Furthermore, since the measurement outcome of is either or , the output of this estimator is -valued. It is also clear that for all . We define the behavior of the sampling oracle for zeroth-order sampling to be essentially the above process.
Let be a decomposition of an objective observable as above, where and is a probability distribution. Given as input a parameterization , a parameter , and an empty coordinate multiset , the oracle behaves as follows. It internally prepares and measures the observable with probability proportional to . It then multiplies the outcome by and outputs the resulting -valued estimator.
2 Analytic gradient measurements
Variational algorithms typically aim to estimate the objective function at some point in parameter space, and use this information along with previous estimates of to propose a new point in parameter space. In this case, the classical outer loop essentially has a stochastic zeroth-order oracle for the objective function.
Some recent works [GS17, MNKF18, RBM+18, SBSW18, SBG+18] have instead suggested a different optimization strategy, in which one directly extracts information about the gradient of the objective function by measuring corresponding quantum observables. Quantum measurements of this type that correspond to estimates of the gradient of the objective function are often referred to as analytic gradient measurements. In this section, we review these strategies.
For notational convenience, we define to be the unitary corresponding to pulse , and for we define to be the sequence of pulses from through , inclusive. Note that in using this notation we are hiding the dependence on for visual clarity. Recall that the objective function corresponding to objective observable is given by
It is straightforward to calculate the following relation via the chain rule applied to the above expression:
We now describe how the above quantity could be measured in a variational algorithm. Denote the Pauli decomposition of as where are products of Pauli operators. As in the previous sections, denote the Pauli decomposition of as . Then by linearity we can rewrite the above derivative as
Now, we can obtain an unbiased estimator for via a (generalized) Hadamard test. In particular, the following procedure may be used for estimating .
Hadamard test for estimating 1. Initialize Register in the qubit state . Initialize Register in the state . 2. Apply to Register . 3. Apply a Controlled- gate to Register , controlled on Register . 4. Apply to Register . 5. Apply a Controlled- gate to Register , controlled on Register . 6. Measure the Pauli operator on Register . The above procedure yields a -valued unbiased estimator for , requiring one quantum measurement.
Algorithm 1: Generalized Hadamard test [EAO+02, LB17, GS17, RBM+18]
Hence, one may estimate by expanding the derivatives as above, and then estimating each term of the expansion using Algorithm 1. Alternative methods of analyticly measuring derivatives are described in [MNKF18, SBG+18], which require similar quantum resources to the scheme we just described (but do not necessarily require controlled-Pauli gates). We note that throughout this paper, one could estimate gradients using a strategy based on these methods instead, and the results would be essentially unchanged.
We now describe an equivalent way of understanding analytic gradients. Observe that
Hence, if we define the Hermitian operators
and we define , then we may write . An alternative commutator expression for the derivatives was noted in [MBS+18].
There are cases in which one may want to impose a constraint that some parameters are always equal. For example, this situation occurs for the “Hamiltonian variational” ansatz proposed in [WHT15]. Hence, one could have a -pulse ansatz but a smaller number of independent variational parameters. We note that this situation is easily addressed within the framework of this paper. For example, consider the case in which is constrained to always equal , i.e. . It is straightforward to show by linearity that , where and are defined as above. Hence, may be estimated via Algorithm 1 just as in the unconstrained case. For simplicity, we assume that there are no such constraints on the parameters. However, all results in this paper can be easily generalized to work with such constraints via this observation.
3 First-order sampling
In the previous section, we described how information about the derivatives of the objective function can be extracted in low depth using a generalized Hadamard test. In this section, we describe a specific estimator of a derivative of the objective function which requires one Pauli measurement. We will use this estimator to define the behavior of the oracle upon a first-order query.
As in the previous section, denote the Pauli expansion of as and the Pauli expansion of as , where all and coefficients are positive real numbers. Then we may write as the following expansion:
We now rewrite this expansion as a certain expectation value, similarly to what we did in the definition of zeroth-order sampling. First, we observe that some of the commutators in the expansion may trivially be zero, if the operators and act nontrivially on disjoint sets of qubits. This will often be the case in the toy model we analyze in Section 5. Removing terms that are trivially zero will improve convergence in our optimization algorithms. To this end, we define a new set of coefficients:
where denotes the set of qubits on which acts nontrivially, after removing pulses which trivially commute through and cancel the corresponding inverse pulse. We define the associated normalization factors
and probability distributions over the indices and , where is considered fixed. Note that we have the bound where . Equipped with these definitions, we may write
It is straightforward to see that for all . Given the above representation of , it is clear that the following procedure provides an unbiased estimator for which requires a single measurement.
An unbiased one-measurement estimator for . 1. Sample from the distribution as defined above. 2. Use a Hadamard test (Algorithm 1) to obtain a one-measurement unbiased estimate of . 3. Multiply the resulting number by . The estimator for described above is -valued.
Algorithm 2: unbiased, one-measurement estimator for .
Motivated by these derivative-estimating procedures, we now define the first-order behavior of the oracle .
Let denote an objective observable. Upon input of parameterization , parameter , and a coordinate multiset for some , the oracle internally prepares the state and runs Algorithm 2 above. It outputs the resulting -valued estimator for .
4 Higher-order sampling
The sampling procedure we have described above for obtaining unbiased estimates of derivatives in low depth generalizes to higher-order derivatives. In this section, we outline how the procedure would work. Start by recalling the derivative operators we derived above:
where, since is independent of , this result follows from arguments identical to those we used to derive the expression for . Also, note that from the original definition , it is clear that . We therefore have, for ,
To see how to estimate this in low depth, note that we have
From the above expression, we see how to generalize the first-order sampling procedure to higher orders. To obtain an unbiased estimate of with a single measurement, first expand , , and as linear combinations of products of Paulis. In turn, this yields an expansion of as a linear combination of real parts of inner products of states that are acted on with pulses and Paulis. For a one-measurement estimator, randomly choose one of these inner products with probability proportional to the magnitude its coefficient, and then get an unbiased estimate of the inner product by performing a Hadamard test, similarly to what we described for the first-order case. Note that, in the second-order case (or more generally for the even-order case), the Hadamard test will involve an -basis measurement instead of -basis measurement, since a real part is being estimated.
5 Query complexity in the black-box formalism
Having defined the sampling oracle , we may now quantify the cost of an algorithm by the number of queries it makes to the oracle. The general setup for a variational optimization problem in the black-box setting is that the classical “outer loop” is promised that the objective observable to be minimized belongs to a family of observables, and is given access to the sampling oracle . Note that, from the perspective of the outer loop, the problem of minimizing the objective function is a purely classical black-box optimization problem since it gives classical input to the oracle and receives classical output. We now formalize the notion of the “error” associated with some variational algorithm for optimizing a family of objective observables.
Let denote a set of objective observables, and be a (possibly randomized) classical algorithm which has access to a sampling oracle for some and outputs a description of a quantum state . Then the optimization error of with respect to , , is defined to be
where the expectation is over the possible randomness of the output state .
In other words, is the worst-case expected error in objective function value that makes over all objective observables in the set . We now make a few more definitions that will be convenient later.
Define the -optimum of an observable to be the set of all states such that . Define the -optimum of a set of observables to be the union of the -optima of each observable in the set. We say a black-box algorithm is a -vicinity algorithm for if it only queries the black box with descriptions of states that are in the -optimum of .
General upper bounds for variational algorithms in a convex region
In this section, we give general upper bounds on the query cost of variational algorithms in a region where the objective function is convex. This amounts to applying the known upper bounds for stochastic convex optimization from Section 2.2 to the setting in which estimates of the objective function, or derivatives of the objective function, come from the oracle specified in Section 3 (which is easy to implement in low depth in practice). Note that the oracle returns estimates of partial derivatives w.r.t. specific components. However, there are multiple ways of using these derivative estimates to construct a gradient estimator. We describe two such estimators. The first is designed to be used with SGD, and the second is designed to be used with SMD with an setup.
For the remainder of this section, fix some objective observable with Pauli expansion and some parameterization whose pulse generators have Pauli expansions . As in Section 3, define , . Define to be the normalization factor associated with coordinate as defined in Section 3 (see also Table 3). We collect these into a vector as
First, we specify some unbiased estimators for the gradient that we will use. The estimator of Algorithm 3 is based on sampling and designed with the goal in mind of achieving a smaller -norm of the estimator and will be used in conjunction with SGD. The estimator of Algorithm 4 is based on sampling and designed with the goal of achieving a smaller -norm of the estimator and will be used in conjunction with SMD. The latter estimator also requires a mild assumption on the -norm of the objective function. The estimators also differ in their number of samples: the former estimator uses a single sample while the latter could be called a “mini-batch” estimator which uses an asymptotically growing number of samples. We first define the two estimators, and then prove their correctness and bound them in the subsequent lemmas.
Algorithm 3: -sampling estimator for .
Algorithm 4: -sampling estimator for .
for . Using this bound with and recalling yields
By the union bound, the probability that for some is upper bounded by . If this event occurs, then we only have the trivial upper bound . Conditioned on this “bad” event not occurring, we have the bound , where we used the assumption that for all . It follows that
2 Upper bounds
Fix an objective observable , parameterization , and a closed, convex feasible set . Define , and define as above (see also Table 3). Let denote a minimizer of on .
Use the 1-query estimator of Algorithm 3 for in conjunction with Theorem 2.1. ∎
Use the 1-query estimator of Algorithm 3 for in conjunction with Theorem 2.2. ∎
Use the estimator of Algorithm 4 for in conjunction with Theorem 2.4. ∎
For comparison, we also present an upper bound for the case in which we only make zeroth-order queries to .
Note that the outputs of zeroth-order queries to have magnitude , and apply Theorem 2.5. ∎
3 When is SMD superior to SGD?
In Section 1.2, we gave intuition for why we might hope that using the 1-norm instead of 2-norm and using SMD with an setup instead of SGD might be beneficial in some cases. In particular, we noted that the 1-norm of a parameter vector has a natural interpretation as the duration of evolution from the starting state to the trial state associated with , . The -distance between two parameter vectors may be interpreted as the amount of time for while the two associated pulse sequences differ.
Comparing the upper bounds from the previous section, we see that where SGD has a factor of , SMD with an setup has instead a factor of . Note that is never larger than , and in fact can a factor of smaller. Consider for example the case in which , which may be a realistic scenario in practice. In this case, we have for the SMD bound, whereas we have for the SGD bound, which is quadratically worse in the dimension of parameter space.
On the other hand, where the SGD bounds involve a factor of , the SMD bounds involve a factor of . is never larger than , and can be significantly smaller. This could be the case when, for example, the feasible set is a Euclidean ball. On the other hand, if is a 1-ball, then , and SMD could potentially achieve substantially better performance than SGD due to the versus discrepancy.
Another consideration is the issue of strong convexity. As is evident from the above bounds, the presence of strong convexity can substantially accelerate the optimization. SGD can take advantage of strong convexity w.r.t. the 2-norm, but SMD in the setup measures strong convexity w.r.t. the 1-norm, and in fact it is straightforward to show that the strong convexity parameters are related by . In the toy problem we analyze in Section 5, we have but where is the number of qubits. However, while , so up to log factors and constants, SGD and SMD achieve the same asymptotic convergence rate for this toy model.
In conclusion, it is not clear from the upper bounds in the previous section or from the toy model we study in Section 5 whether SGD or SMD with an setup would typically achieve better upper bounds in practice. It is an interesting problem for future work to understand whether an (Euclidean) setup or an setup is usually more natural for variational algorithms.
Oracle separation between zeroth-order and first-order optimization strategies for variational algorithms
In this section, we prove a separation between algorithms which make only zeroth-order queries to the sampling oracle, and those which make first-order queries to the sampling oracle, within the vicinity of the global optimum. This separation is proven with respect to a certain simple parameterized family of objective observables on qubits. The optima of the observables in are close to each other, in the sense that for any , the ground state of is an optimum of . Precisely, we will prove the following.
For any and , let be any zeroth-order, -vicinity algorithm for the family that makes queries to the oracle. Then, if , it must hold that where the implicit factor is some fixed constant.
On the other hand, we prove that this same class of variational problems can be optimized substantially faster if the algorithm makes first-order queries to the oracle, as quantified in the following theorem. In fact, the algorithm that achieves this convergence rate is a simple stochastic gradient descent strategy. Hence, not only is the query complexity much better in this case, but the classical algorithm achieving this query complexity can be implemented efficiently. For comparison, we also obtain a zeroth-order upper bound for this class of problems using the algorithms of [FKM05] and [AFH+11]. Our first-order upper bound is given in the following theorem.
For any , there exists a first-order, -vicinity algorithm for the family that makes queries and achieves an error . Moreover, is a simple stochastic gradient descent algorithm.
For any and , suppose is an algorithm that makes queries and satisfies . Then .
Since this lower bound is achieved (up to a possible constant factor) by the upper bound of SGD, we see that SGD is essentially optimal among all black-box strategies for optimizing .
The subset of objective observables we consider are perturbed around a very simple 1-local Hamiltonian.
Intuitively, for a fixed small parameter , the set of observables are perturbed around . The parameter characterizes the strength of the perturbation, and the binary vector encodes the direction of the perturbation. It is straightforward to see that the ground state of is , where we have defined . Geometrically, the state corresponds to the pure qubit state with polarization . In the remainder of this section, we record some facts about these Hamiltonians, and define some quantities.
First, note that we may write where and is the vector of Pauli operators acting on qubit . We may now read off , and the associated eigenvector is
Next, we calculate the expectation value of with respect to any quantum state on qubits.
Suppose is a quantum state such that the polarization of , the reduced state of on qubit , is . Then .
Finally, we define the set which we will prove the separation with respect to. To do so, we first define a bias parameter associated with the precision parameter .
For a given “precision parameter” , define the associated “bias parameter”
Now, we define to be the set of such observables with bias parameter .
.
For the remainder of the paper, we often hide the dependence of on for notational simplicity, and simply write where we implicitly mean . Note that our constraint implies .
In this section, we prove Theorem 5.1. Our proof strategy for the lower bound is to reduce a statistical learning problem to the optimization problem, and then lower bound the number of oracle calls required to solve the learning problem. Precisely, we will take an appropriate subset , parameterized by some subset of the -dimensional hypercube . That is, we will have where will be strategically chosen. We prove that, if there exists an algorithm that satisfies , then the same algorithm could be used to identify the hidden parameter associated with the objective observable . By employing information theoretic methods, we will lower bound the number of oracle calls required to identify the parameter , which in turn lower bounds the number of calls required to optimize to precision .
Our proof in some parts adapts techniques from [AWBR09] and [JNR12], which lower bound the query cost of certain convex first-order and derivative-free optimization problems. These results in turn draw on methods from statistical minimax and learning theory.
We begin by defining, for fixed , a subset of objective observables that are well-separated, in the sense that if a state is close to the optimal of , then it must be far from the optimal of for any other parameter . We make this precise below.
We make use of the following classical fact about packings of the hypercube (see for example [Gun11] for a simple proof).
There exists a subset of the -dimensional hypercube of size such that, if denotes the Hamming distance between and ,
for all with .
Fix to be such a subset of , and define . The Hamming distance provides a natural distance measure between points of the hypercube. We now define a notion of distance between objective observables and . Intuitively, if is large, then a state that is close to the optimal of cannot be close to the optimal of .
For , we define the semimetric
where the minimization is over all normalized pure states on qubits.
Note that is simply , but we oftentimes write for clarity. We now define a packing parameter which quantifies how packed the subset is, with respect to the semimetric .
The packing parameter corresponding to the above subset and semimetric on the hypercube is defined to be
Suppose that for some state and parameter , . Then for all with , .
Suppose there exists some parameter , for which . From Definition 5.4, this implies that , which contradicts the assumption that is the packing parameter. ∎
We now show that any algorithm which optimizes the observables in the set with error can be used to identify the parameter with high probability.
Suppose that is an algorithm such that . Then, one may use the output of to construct an estimator such that, if the objective observable is for , then .
By assumption, if the observable that is realized is for , outputs a description of a quantum state such that
Define the estimator . Lemma 5.3 implies that, if , this estimator returns with probability one. Since this event occurs with probability at least , the estimator returns with probability at least . ∎
We have shown that the ability to optimize well implies the ability to identify the hidden parameter with high probability. We now compute the packing parameter for the family .
For the subset , semimetric , and packing parameter as defined above,
Recall that for all ,
where the minimization is over all normalized pure states on qubits. Therefore, to compute , it suffices to compute the smallest eigenvalue of .
where we used the trigonometric identities . From this expression, it is clear that the smallest eigenvalue of is , from which it follows that . By construction, for all with , we have . It follows that .
The final inequality follows from the fact that for . ∎
Any algorithm for which can be used to construct an estimator which correctly identifies the parameter of the realized observable with probability at least .
By Lemma 5.5, the packing parameter is at least . Then by Lemma 5.4, if we can optimize observables in the set with expected error at most , we can identify with probability at least . ∎
Our proof will proceed as follows. We restrict to the subset and prove a lower bound on the number of zeroth-order, -vicinity queries one must make in order to identify the hidden parameter associated with the realized objective observable . By Lemma 5.6, this number also lower bounds the number of such queries an algorithm must make to satisfy . Since is a subset of , optimizing is no harder than optimizing , and so this number also lower bounds the number of such queries needed to optimize to precision .
We next prove two simple lemmas we will need.
Suppose is in the -optimum of , i.e. . Let be the polarization of the reduced state of on qubit , and let be the (unoriented) angle between the vector and the unit vector . Then and .
Now, note that for . This gives us
It immediately follows from Jensen’s inequality that
Suppose is in the -optimum of for some . Then is in the -optimum of for any .
As in the previous lemma, let denote the polarization of the reduced state on qubit , and denote the angle between and . We have
Here, the relation can be seen geometrically. We also used the definition .
2.2 Applying Fano’s inequality
At this point, it remains to lower bound the number of zeroth-order calls to the oracle needed to correctly identify the unknown bias parameter . The results from the previous section will then allow us to turn this into a lower bound for optimization. We will need the following well-known variant of Fano’s inequality. For this result and other information-theoretic results used in this section, see (for example) [CT91].
Suppose the random variable is uniformly distributed on the discrete set , and the variable may be correlated with . Suppose is an algorithm that attempts to identify given the variable . Then the probability of error satisfies
where is the mutual information between and . When we use this inequality in our proof, we will let be the set of bias parameters associated with defined in the previous section, and will be the set of queries to and outputs from the zeroth-order sampling oracle.
First, recall how the oracle behaves for zeroth-order queries. It selects a term in the Pauli expansion of the objective observable with probability proportional to the magnitude of the coefficient of that term. Consider the objective observable . Note that the sum of coefficients of Pauli operators acting on qubit is where we have used a standard trigonometric identity. Note that this quantity is independent of the parameter . This means that, when we do a zeroth-order query of the oracle encoding this Hamiltonian, the oracle is equally likely to select or for measurement as it is or for some other . Thus, we may equivalently describe the oracle as operating in the following manner. Note that the below algorithm is simply a specialization of the zeroth-order behavior of the sampling oracle (Definition 3.2) to the particular objective observable .
Zeroth order behavior of Upon input of a parameterization , parameter , and empty coordinate multiset , 1. Select an index uniformly at random. 2. Flip a coin with probability of heads . 3. If heads, measure w.r.t. the state . If tails, measure . 4. Multiply the above measurement outcome by and output the result.
Algorithm 5: zeroth-order behavior of .
Let the parameter be uniformly distributed, and denote the associated random variable . Suppose an algorithm makes zeroth-order queries to the oracle. Let be the input to the oracle in query . Let denote the output of query . The algorithm may use information from steps one through to decide the input to query the oracle with on iteration . Formally, we have the variables , where (the algorithm’s first guess) is independent of , and is a deterministic or stochastic function of . We begin with a simple lemma. Note that versions of this relation are well-known (e.g. [AWBR09, RR11, JNR12]).
.
where in the first line we have used the chain rule for mutual information, in the second we used the fact that depends only on , in the third we used the definition of mutual information, and in the fourth we used subadditivity and the fact that depends only on and . ∎
Letting denote the relative entropy of two distributions, we have for any ,
Recall that since only queries states in the -optimum of , then for any state that is queried, by Lemma 5.8. As we have done before, let denote the angle between and . By Lemma 5.7, we know that for any state that is queried.
Suppose and are two -valued Bernoulli distributions, with and . Then
where in the second line we used the inequality with . ∎
where the last line follows from the same reasoning as in the proof of Lemma 5.8. Now, using Lemma 5.11 we have
where we have used .
2.4 Completing the proof
Combining the above bound with Lemma 5.10, upon making zeroth-order queries to the oracle within the -optimum of , and obtaining outcomes , the mutual information between the hidden vector and the inputs and outputs of the oracle is upper bounded by . Lemma 5.9 implies that for any algorithm which attempts to identify the bias vector given , the error probability is lower bounded by . Recalling that , we have
Let be value of such that the final expression above is equal to . A simple calculation shows . In particular, for , . Note that, if an algorithm makes fewer than zeroth-order queries, the probability that it can correct identify the hidden parameter is less than .
We have shown that for and , when constrained to the -optimum of , at least zeroth-order queries to the oracle are required to identify the bias parameter with probability of success at least . Our previous reduction from learning to optimization then implies that at least this many samples are required to optimize observables in the set with expected error at most . We have therefore shown Theorem 5.1.
We have shown that zeroth-order queries to the sampling oracle are required for a -vicinity algorithm to optimize the family to precision . In this section, we show that with a certain natural state parameterization and making only first-order queries to the sampling oracle, the family can be optimized to precision with queries by a -vicinity algorithm based on SGD.
We start by defining the variational ansatz that we will use in our first-order optimization procedure. We define the following -parameter parameterization :
This parameterization has a simple geometric interpretation: is the product state on qubits for which the polarization of qubit is . Clearly this ansatz is natural for the family in some sense.
Consider some objective observable . From Lemma 5.1, we have that the induced objective function is given by
The set of states associated with is contained in the -optimum of .
For any and , we have
We now will show that is -strongly convex on w.r.t. the Euclidean norm. To do so, we compute its Hessian matrices . We have for , and . Since , it must hold that where we used our assumption for the last inequality. Since all eigenvalues of for are at least , is -strongly convex on .
We now calculate for this particular parameterization and some objective observable . Expanding the gradient as in Section 3 (see also Table 3),
where, as usual, . We now remove terms which are trivially zero because the commutator involves operators which act nontrivially on disjoint qubits. In particular, since in this case, then clearly . Dropping such terms in the expansion,
Using a very similar argument to that of the proof of Theorem 5.1, we may lower bound the number of calls to required to optimize any objective observable in the family with expected error at most . In the setting of Theorem 5.1, the algorithm was restricted to querying the oracle with states in the -optimum of . In this section, the algorithm is allowed to query the oracle with states which may be outside this domain. We also allow the algorithm to make queries of any order, instead of just zeroth-order. As before, we actually prove a lower bound for the strictly easier problem of optimizing the subset .
We will essentially bound the amount of information contained in a single oracle query for any order derivative and for any state. As usual, let denote the parameterization given by . Recall from Section 3.4 that, assuming w.l.o.g. that , the expansion of in terms of nested commutators of conjugated Pauli operators is
The crucial point is that the sampling oracle cannot reveal any more information about the hidden parameter than the outcome of the internal coin flip in Step 2 of the above box. This is because, since only Step 2 in the above box depends on the hidden parameter , the algorithm can simulate the oracle if it has knowledge of the outcome of the internal coin flip. More formally, we have the following lemma.
Let be the hidden parameter, be the input to the oracle, be the outcome of the internal coin flip, and be the output of the oracle. Then .
Note from the above box that the coin flip of Step 2 is the only part of the black box’s internal procedure that depends on ; the output is simply a stochastic function of . Hence the variables form a Markov chain, and the claim follows from the data processing inequality. ∎
At this point, we may follow a virtually identical argument to that in Section 5.2.4 to find that, for and , at least oracle queries are required to identify the hidden bias parameter with probably at least . Hence, at least oracle queries are required to optimize with worst-case expected error at most .
Since this lower bound has a matching upper bound via first-order oracle queries and SGD (up to constant factors), we see that SGD is in fact essentially optimal among all black-box strategies for optimizing the family .
Conclusion and open questions
We have introduced a natural black-box setting for variational algorithms, which can be straightforwardly implemented in practice. With respect to this setting, we derived rigorous upper bounds on the query cost of variational algorithms, in the setting where the induced objective function is convex within a convex feasible set. These bounds depended on the precision, dimension of parameter space, factors from the objective observable and pulse generators, and strong convexity parameters. We derived bounds both for algorithms running SGD in a Euclidean space, and for algorithms running SMD in an space. For some settings of parameters SGD has stronger upper bounds, and for other settings of parameters SMD has stronger upper bounds. For the toy problem we analyze, SGD outperforms SMD by a factor that is merely logarithmic in the number of parameters. It is an interesting open question to understand which geometry is most natural for variational algorithms in practice.
We also introduced a simple class of objective observables on qubits, and proved a separation between the query cost of optimizing these observables in the vicinity of the optimum in the cases of zeroth-order (objective function measurements) versus first-order (analytic gradient measurements) optimization. We showed that, for this class of observables, a simple stochastic gradient descent strategy could outperform any possible variational algorithm (with any choice of ansatz) that only receives zeroth-order information from the oracle. We view these results as evidence that taking analytic gradient measurements in variational algorithms and using the measurement results to run a stochastic first-order optimization algorithm could be advantageous as compared to derivative-free strategies in some cases.
It would be interesting to understand the behavior of the objective function near a local minimum for problems and variational ansatzes which appear in practice. In particular, it would be interesting to understand how the strong convexity of typically behaves near a local minimum. Without a strong convexity guarantee, stochastic descent methods typically have query upper bounds scaling with the precision like . However, given a promise of -strong convexity, the cost is typically . For the toy model that we analyzed, we showed that with an appropriate choice of ansatz, the problem was -strongly convex with respect to the 2-norm. As a result of this property, we were able to obtain a query upper bound for optimizing this family with SGD. We apparently were able to exploit strong convexity by making a prudent choice of variational ansatz for the problem class at hand.
This situation may be viewed as the opposite of that studied in [MBS+18], which essentially considered a situation in which the variational ansatz looks random. In this situation, the gradient of the objective function is highly concentrated around zero. One way to view the difference in our models is that in our paper there are independent (i.e. commuting) degrees of freedom while in [MBS+18] different terms in the Hamiltonian and pulses have the commutation relations that we would expect from Haar-random projectors. Our model could be seen as justified by the common intuition in many-body physics that local unitaries applied to the ground state create quasiparticles, and that in an -qubit system independent quasiparticles are possible. Their model, on the other hand, could be justified by the assumption that the variational ansatz is far from a local minimum and so the pulses act like random local unitaries. It would be interesting to understand which of these scenarios is more realistic in practice. In particular, one might hope that theoretically motivated ansatzes, such as the unitary-coupled-cluster ansatz in quantum chemistry, could possess properties near an optimum (such as strong convexity) that make them more amenable to efficient optimization.
Finally, another point that we left unaddressed is the issue of noise. It would be interesting to study how to take analytic gradient measurements in the presence of noise, and what impact this has on the convergence rate of stochastic optimization methods. In particular, these methods are quite robust against unbiased noise, but their effectiveness in the presence of biased noise is less understood.
Acknowledgements
We thank an anonymous reviewer for helpful suggestions, and for pointing out the good practical effectiveness of derivative-free trust region and surrogate methods. We thank Xiaodi Wu for useful discussions. JN and AWH were funded by ARO contract W911NF-17-1-0433 and NSF grants CCF-1729369 and PHY-1818914. AWH was also funded by NSF grant CCF-1452616 and the MIT-IBM Watson AI Lab under the project Machine Learning in Hilbert space.
References
Appendix A Background on stochastic gradient and mirror descent
In this section, we review some relevant preliminaries pertaining to convex optimization and stochastic descent algorithms. Much of the material in this section follows the review [Bub15].
where is the stepsize at iteration , and is the Euclidean projection onto , . The intuition for this strategy is clear: the vector points in the direction of steepest decrease of at , and in each iteration we take a step of size in this direction and then project back into . It is sometimes helpful to think of gradient descent in an alternative, proximal picture. Namely, Eq. A.1 is equivalent to
Intuitively, the point is chosen to minimize , a linearization of around , while not making the regularization term too big.
The following result about projected gradient descent is well-known. Recall that a differentiable function is -Lipschitz with respect to if for all , where denotes the dual norm.
If the convex function is -Lipschitz w.r.t. the Euclidean norm, and is contained in a Euclidean ball of radius , then projected gradient descent with stepsize satisfies
where is a minimizer of on .
Note that this implies that iterations are sufficient for some desired precision . We now define strong convexity.
The following result about projected gradient descent for strongly convex functions is known.
Let be -strongly convex and -Lipschitz on , w.r.t. the Euclidean norm. Then projected gradient descent with satisfies
Note that this result implies that iterations are sufficient to optimize to error .
It turns out that if one does gradient descent with noisy, unbiased estimates of the gradient instead of the true gradient , the above results are qualitatively unchanged. We refer to projected gradient descent with stochastic gradient estimates as stochastic gradient descent (SGD). We now state some results formally.
A.2 Mirror descent
A reflection on gradient descent shows that the gradient descent procedure defined above in fact only makes sense when we are working in Euclidean space. For example, an iteration of gradient descent (Eq. A.1) involves adding the vectors and . When the problem is defined in Euclidean space, may be considered as living in the same space by the Riesz representation theorem. However, if (for example) the objective function is defined on an space, then the gradient lives in the dual space, and hence adding these vectors is not even formally well-defined.
Mirror descent may be viewed as a generalization of gradient descent to non-Euclidean geometries. To gain intuition for why we might want to do this, recall that minimizing a function that is -Lipschitz in the Euclidean norm to precision requires iterations using the above projected gradient descent bound. Note that this expression does not have any explicit dependence on the dimension . However, if the parameter has a dependence on , then the convergence rate could depend on implicitly. Consider for example a situation in which we know that all partial derivatives of are bounded by , so that for all in the domain. Then it follows that we can bound , and so we obtain an upper bound of for gradient descent, which has a linear dependence on . But notice that under this assumption, we have a much stronger bound on the -norm of the gradient. In particular, , so the Lipschitz constant is only with respect to this geometry. If we could somehow work in an geometry so that is the relevant quantity instead of , then perhaps we could achieve a stronger upper bound on the convergence rate. Indeed, this is possible with mirror descent.
We may associate to the mirror map its Bregman divergence,
The quantity may be thought of as a distance measure between and , generated by . If , then . If , then , the (generalized) KL divergence between and . We now define the notion of a projection onto the feasible set with respect to the Bregman divergence :
We are now ready to define the mirror descent procedure, with stepsize . Let . Then mirror descent is defined by the following iteration. For , let and be such that
In other words, we first move to a “dual space” via the mirror map , then do the gradient descent step in the dual space, then move back to the original space again via the mirror map. The resulting point, , may lie outside the feasible set , so we then project back to via the Bregman divergence generated by . We also note that a step of mirror descent can be equivalently described in the following proximal picture, which makes the relation to gradient descent clearer.
The following convergence rate can be proven for mirror descent.
If is -strongly convex on with respect to , , is convex, and is -Lipschitz with respect to , then mirror descent with satisfies
Whenever is contained in an -ball of radius centered at the origin, we have and . These assumptions on can always be achieved by shifting and scaling . We now state a mirror descent bound for an geometry.
If the convex function is -Lipschitz with respect to the norm , and is contained in a 1-ball of radius , then projected mirror descent with an appropriate choice of mirror map and stepsizes satisfies
Finally, if is strongly convex with respect to the norm , then SMD can be accelerated similarly to how SGD can be accelerated for strongly convex functions. See for example [HK14].