Convergence rates of efficient global optimization algorithms
Adam D. Bull
Introduction
Many standard global optimization algorithms exist, including genetic algorithms, multistart, and simulated annealing (Pardalos and Romeijn, 2002), but these algorithms are designed for functions that are cheap to evaluate. When is expensive, we need an efficient algorithm, one which will choose its observations to maximize the information gained.
We can consider this a continuum-armed-bandit problem (Srinivas et al., 2010, and references therein), with noiseless data, and loss measured by the simple regret (Bubeck et al., 2009). At time , we choose a design point , make an observation , and then report a point where we believe will be low. Our goal is to find a strategy for choosing the and , in terms of previous observations, so as to minimize .
We would like to find a strategy which can guarantee convergence: for functions in some smoothness class, should tend to , preferably at some fast rate. The simplest method would be to fix a sequence of in advance, and set , for some approximation to . We will show that if converges in supremum norm at the optimal rate, then also converges at its optimal rate. However, while this strategy gives a good worst-case bound, on average it is clearly a poor method of optimization: the design points are completely independent of the observations .
We may therefore ask if there are more efficient methods, with better average-case performance, that nevertheless provide good guarantees of convergence. The difficulty in designing such a method lies in the trade-off between exploration and exploitation. If we exploit the data, observing in regions where is known to be low, we will be more likely to find the optimum quickly; however, unless we explore every region of , we may not find it at all (Macready and Wolpert, 1998).
Initial attempts at this problem include work on Lipschitz optimization (summarized in Hansen et al., 1992) and the DIRECT algorithm (Jones et al., 1993), but perhaps the best-known strategy is expected improvement. It is sometimes called Bayesian optimization, and first appeared in Močkus (1974) as a Bayesian decision-theoretic solution to the problem. Contemporary computers were not powerful enough to implement the technique in full, and it was later popularized by Jones et al. (1998), who provided a computationally efficient implementation. More recently, it has also been called a knowledge-gradient policy by Frazier et al. (2009). Many extensions and alterations have been suggested by further authors; a good summary can be found in Brochu et al. (2010).
Vazquez and Bect (2010) show that when is a fixed Gaussian process prior of finite smoothness, expected improvement converges on the minimum of any , and almost surely for drawn from . Grunewalder et al. (2010) bound the convergence rate of a computationally infeasible version of expected improvement: for priors of smoothness , they show convergence at a rate on drawn from . We begin by bounding the convergence rate of the feasible algorithm, and show convergence at a rate on all . We go on to show that a modification of expected improvement converges at the near-optimal rate .
For practitioners, however, these results are somewhat misleading. In typical applications, the prior is not held fixed, but depends on parameters estimated sequentially from the data. This process ensures the choice of observations is invariant under translation and scaling of , and is believed to be more efficient (Jones et al., 1998, §2). It has a profound effect on convergence, however: Locatelli (1997, §3.2) shows that, for a Brownian motion prior with estimated parameters, expected improvement may not converge at all.
We extend this result to more general settings, showing that for standard priors with estimated parameters, there exist smooth functions on which expected improvement does not converge. We then propose alternative estimates of the prior parameters, chosen to minimize the constants in the convergence rate. We show that these estimators give an automatic choice of parameters, while retaining the convergence rates of a fixed prior.
In Section 2, we briefly describe the expected-improvement algorithm, and detail our assumptions on the priors used. We state our main results in Section 3, and discuss implications for further work in Section 4. Finally, we give proofs in Appendix A.
Expected Improvement
and our goal, given , is to choose the strategy to minimize this quantity.
For this problem is very computationally intensive (Osborne, 2010, §6.3), but we can solve a simplified version of it. First, we restrict the choice of to the previous design points . (In practice this is reasonable, as choosing an we have not observed can be unreliable.) Secondly, rather than finding an optimal strategy for the problem, we derive the myopic strategy: the strategy which is optimal if we always assume we will stop after the next observation. This strategy is suboptimal (Ginsbourger et al., 2008, §3.1), but performs well, and greatly simplifies the calculations involved.
In this setting, given , if we are to stop at time we should choose , where . (In the case of ties, we may pick any minimizing .) We then suffer a loss , where . Were we to observe at before stopping, the expected loss would be
so the myopic strategy should choose to minimize this quantity. Equivalently, it should maximize the expected improvement over the current loss,
So far, we have merely replaced one optimization problem with another. However, for suitable priors, can be evaluated cheaply, and thus maximized by standard techniques. The expected-improvement algorithm is then given by choosing to maximize (1).
2 Gaussian Process Models
We still need to choose a prior for . Typically, we model as a stationary Gaussian process: we consider the values to be jointly Gaussian, with mean and covariance
for an underlying kernel with . (Note that we can always satisfy this condition by suitably scaling and .) The are the length-scales of the process: two values and will be highly correlated if each is small compared with . For now, we will assume the parameters and are fixed in advance.
For (2) and (3) to define a consistent Gaussian process, must be a symmetric positive-definite function. We will also make the following assumptions.
and by Bochner’s theorem, is non-negative and integrable.
is isotropic and radially non-increasing.
In other words, for a non-increasing function ; as a consequence, is isotropic.
for some ; or
for all (we will then say that ).
Note the condition is required for to be integrable.
is , for the largest integer less than , and at the origin, has -th order Taylor approximation satisfying
When , this is just the condition that be -Hölder at the origin; when , we instead require this condition up to a log factor.
The rate controls the smoothness of functions from the prior: almost surely, has continuous derivatives of any order (Adler and Taylor, 2007, §1.4.2). Popular kernels include the Matérn class,
where is a modified Bessel function of the second kind, and the Gaussian kernel,
Having chosen our prior distribution, we may now derive its posterior. We find
for , , and (Santner et al., 2003, §4.1.3). Equivalently, these expressions are the best linear unbiased predictor of and its variance, as given in Jones et al. (1998, §2). We will also need the reduced sum of squares,
3 Expected Improvement Strategies
Under our assumptions on , we may now derive an analytic form for (1), as in Jones et al. (1998, §4.1). We obtain
and and are the standard normal distribution and density functions respectively.
For a prior as above, expected improvement chooses to maximize (8), but this does not fully define the strategy. Firstly, we must describe how the strategy breaks ties, when more than one maximizes . In general, this will not affect the behaviour of the algorithm, so we allow any choice of maximizing (8).
Secondly, we must say how to choose , as the above expressions are undefined when . In fact, Jones et al. (1998, §4.2) find that expected improvement can be unreliable given few data points, and recommend that several initial design points be chosen in a random quasi-uniform arrangement. We will therefore assume that until some fixed time , points are instead chosen by some (potentially random) method independent of . We thus obtain the following strategy.
initial design points independently of ; and
further design points from the maximizers of (8).
So far, we have not considered the choice of parameters and . While these can be fixed in advance, doing so requires us to specify characteristic scales of the unknown function , and causes expected improvement to behave differently on a rescaling of the same function. We would prefer an algorithm which could adapt automatically to the scale of .
A natural approach is to take maximum likelihood estimates of the parameters, as recommended by Jones et al. (1998, §2). Given , the MLE ; for full generality, we will allow any choice , where . Estimates of , however, must be obtained by numerical optimization. As can vary widely in scale, this optimization is best performed over ; as the likelihood surface is typically multimodal, this requires the use of a global optimizer. We must therefore place (implicit or explicit) bounds on the allowed values of . We have thus described the following strategy.
Let be a sequence of priors, with parameters , satisfying:
for constants , ; and
An strategy satisfies 1, replacing with in (8).
Convergence Rates
which holds for all . See Aronszajn (1950), Berlinet and Thomas-Agnan (2004), Wendland (2005) and van der Vaart and van Zanten (2008).
and there is a unique minimizing this expression.
is finite. Thus, for the kernel with Fourier transform , this is just the RKHS . More generally, if satisfies our assumptions with , these spaces are equivalent in the sense of normed spaces: they contain the same functions, and have norms satisfying
If , is equivalent to the Sobolev Hilbert space .
If , is continuously embedded in for all .
Thus if , and is, say, a product of intervals , the RKHS is equivalent to the Sobolev Hilbert space , identifying each function in that space with its unique continuous extension to .
2 Fixed Parameters
We will say that converges on the optimum at rate , if
for all . Note that we do not allow to vary with ; the strategy must achieve this rate without prior knowledge of .
We begin by showing that the minimax rate of convergence is .
and this rate can be achieved by a strategy not depending on .
The upper bound is provided by a naive strategy as in the introduction: we fix a quasi-uniform sequence in advance, and take to minimize a radial basis function interpolant of the data. As remarked previously, however, this naive strategy is not very satisfying; in practice it will be outperformed by any good strategy varying with the data. We may thus ask whether more sophisticated strategies, with better practical performance, can still provide good worst-case bounds.
One such strategy is the strategy of 1. We can show this strategy converges at least at rate , up to log factors.
For , these rates are near-optimal. For , we are faced with a more difficult problem; we discuss this in more detail in Section 3.4.
3 Estimated Parameters
First, we consider the effect of the prior parameters on . While the previous result gives a convergence rate for any fixed choice of parameters, the constant in that rate will depend on the parameters chosen; to choose well, we must somehow estimate these parameters from the data. The strategy, given by 2, uses maximum likelihood estimates for this purpose. We can show, however, that this may cause the strategy to never converge.
The counterexamples constructed in the proof of the theorem may be difficult to minimize, but they are not badly-behaved (Figure 1). A good optimization strategy should be able to minimize such functions, and we must ask why expected improvement fails.
We can understand the issue by considering the constant in Theorem 2. Define
From the proof of Theorem 2, the dominant term in the convergence rate has constant
for not depending on or . In Appendix A, we will prove the following result.
is non-decreasing in , and bounded above by .
Hence for fixed , the estimate , and thus . Inserting this choice into (10) gives a constant growing exponentially in , destroying our convergence rate.
To resolve the issue, we will instead try to pick to minimize (10). The term is increasing in , and the term is decreasing in ; we may balance the terms by taking . The constant is then proportional to , which we may minimize by taking . In practice, we will not know in advance, so we must estimate it from the data; from 1, a convenient estimate is .
Suppose, then, that we make some bounded estimate of , and set . As Theorem 3 holds for any of faster than logarithmic decay, such a choice is necessary to ensure convergence. (We may also choose to minimize (10); we might then pick minimizing but our assumptions on are weak enough that we need not consider this further.)
If we believe our Gaussian-process model, this estimate is certainly unusual. We should, however, take care before placing too much faith in the model. The function in Figure 1 is a reasonable function to optimize, but as a Gaussian process it is highly atypical: there are intervals on which the function is constant, an event which in our model occurs with probability zero. If we want our algorithm to succeed on more general classes of functions, we will need to choose our parameter estimates appropriately.
To obtain good rates, we must add a further condition to our strategy. If , is identically zero, and all choices of are equally valid. To ensure we fully explore , we will therefore require that when our strategy is applied to a constant function , it produces a sequence dense in . (This can be achieved, for example, by choosing uniformly at random from when .) We have thus described the following strategy.
we instead set ; and
we require the choice of maximizing (8) to be such that, if is constant, the design points are almost surely dense in .
We cannot now prove a convergence result uniform over balls in , as the rate of convergence depends on the ratio , which is unbounded. (Indeed, any estimator of must sometimes perform poorly: can appear from the data to have arbitrarily small norm, while in fact having a spike somewhere we have not yet observed.) We can, however, provide the same convergence rates as in Theorem 2, in a slightly weaker sense.
4 Near-Optimal Rates
So far, our rates have been near-optimal only for . To obtain good rates for , standard results on the performance of Gaussian-process interpolation (Narcowich et al., 2003, §6) then require the design points to be quasi-uniform in a region of interest. It is unclear whether this occurs naturally under expected improvement, but there are many ways we can modify the algorithm to ensure it.
Perhaps the simplest, and most well-known, is an -greedy strategy (Sutton and Barto, 1998, §2.2). In such a strategy, at each step with probability we make a decision to maximize some greedy criterion; with probability we make a decision completely at random. This random choice ensures that the short-term nature of the greedy criterion does not overshadow our long-term goal.
The parameter controls the trade-off between global and local search: a good choice of will be small enough to not interfere with the expected-improvement algorithm, but large enough to prevent it from getting stuck in a local minimum. Sutton and Barto (1998, §2.2) consider the values and , but in practical work should of course be calibrated to a typical problem set.
We therefore define the following strategies.
chooses initial design points independently of ;
with probability , chooses design point as in ; or
with probability , chooses uniformly at random from .
We can show that these strategies achieve near-optimal rates of convergence for all .
Let be one of the strategies in 4. If , then for any ,
while if , the statement holds for all .
Conclusions
We have shown that expected improvement can converge near-optimally, but a naive implementation may not converge at all. We thus echo Diaconis and Freedman (1986) in stating that, for infinite-dimensional problems, Bayesian methods are not always guaranteed to find the right answer; such guarantees can only be provided by considering the problem at hand.
We might ask, however, if our framework can also be improved. Our upper bounds on convergence were established using naive algorithms, which in practice would prove inefficient. If a sophisticated algorithm fails where a naive one succeeds, then the sophisticated algorithm is certainly at fault; we might, however, prefer methods of evaluation which do not consider naive algorithms so successful.
Vazquez and Bect (2010) and Grunewalder et al. (2010) consider a more Bayesian formulation of the problem, where the unknown function is distributed according to the prior , but this approach can prove restrictive: as we saw in Section 3.3, placing too much faith in the prior may exclude functions of interest. Further, Grunewalder et al. find the same issues are present also within the Bayesian framework.
A more interesting approach is given by the continuum-armed-bandit problem (Srinivas et al., 2010, and references therein). Here the goal is to minimize the cumulative regret,
in general observing the function under noise. Algorithms controlling the cumulative regret at rate also solve the optimization problem, at rate (Bubeck et al., 2009, §3). The naive algorithms above, however, have poor cumulative regret. We might, then, consider the cumulative regret to be a better measure of performance, but this approach too has limitations. Firstly, the cumulative regret is necessarily increasing, so cannot establish rates of optimization faster than . (This is not an issue under noise, where typically , see Kleinberg and Slivkins, 2010.) Secondly, if our goal is optimization, then minimizing the regret, a cost we do not incur, may obscure the problem at hand.
Bubeck et al. (2010) study this problem with the additional assumption that has finitely many minima, and is, say, quadratic in a neighbourhood of each. This assumption may suffice in practice, and allows the authors to obtain impressive rates of convergence. For optimization, however, a further weakness is that these rates hold only once the algorithm has found a basin of attraction; they thus measure local, rather than global, performance. It may be that convergence rates alone are not sufficient to capture the performance of a global optimization algorithm, and the time taken to find a basin of attraction is more relevant. In any case, the choice of an appropriate framework to measure performance in global optimization merits further study.
Finally, we should also ask how to choose the smoothness parameter (or the equivalent parameter in similar algorithms). van der Vaart and van Zanten (2009) show that Bayesian Gaussian-process models can, in some contexts, automatically adapt to the smoothness of an unknown function . Their technique requires, however, that the estimated length-scales to tend to 0, posing both practical and theoretical challenges. The question of how best to optimize functions of unknown smoothness remains open.
We would like to thank the referees, as well as Richard Nickl and Steffen Grunewalder, for their valuable comments and suggestions.
Appendix A Proofs
and as ,
so . is thus the Fourier transform of a real continuous , satisfying the Fourier inversion formula everywhere.
If , by assumption , for a finite non-increasing function satisfying as . Hence
If , by a similar argument is continuously embedded in all . ∎
From 1, we can derive results on the behaviour of as varies. For small , we obtain the following result.
If , then for all , and
Let . As is isotropic and radially non-increasing,
Likewise, for large , we obtain the following.
If , , then for , and
for a depending only on and .
As in the proof of 3, we have constants such that
and we may argue as in the previous lemma. ∎
We can also describe the posterior distribution of in terms of ; as a consequence, we may deduce 1.
Suppose , .
solves the optimization problem
with minimum value .
with equality for some .
Let , and write for , . , so affects the optimization only through . The minimal thus has , so . The problem then becomes
The solution is given by (4) and (5), with value (7).
By symmetry, the prediction error does not depend on , so we may take . Then
for , and
Now, , as given by (6); this is a consequence of Loève’s isometry, but is easily verified algebraically. The result then follows by Cauchy-Schwarz. ∎
A.2 Fixed Parameters
We first establish the lower bound. Suppose we have functions with disjoint supports. We will argue that, given observations, we cannot distinguish between all the , and thus cannot accurately pick a minimum .
but on that event, cannot distinguish between and before time , so
As the minimax loss is non-increasing in , for we conclude
For general having non-empty interior, we can find a hypercube , with . We may then proceed as above, picking functions supported inside .
For the upper bound, consider a strategy choosing a fixed sequence , independent of the . Fit a radial basis function interpolant to the data, and pick to minimize . Then if minimizes ,
so the loss is bounded by the error in .
From results in Narcowich et al. (2003, §6) and Wendland (2005, §11.5), for suitable radial basis functions the error is uniformly bounded by
To prove Theorem 2, we first show that some observations will be well-predicted by past data.
If , then by assumption
as . If , then is differentiable, so as is symmetric, . If further , then
Similarly, if , then is , so
for a constant depending only on , and .
We next show that most design points are close to a previous . is bounded, so can be covered by balls of radius . If lies in a ball containing some earlier point , , then we may conclude
for a constant depending only on , and . Hence as there are balls, at most points can satisfy
Next, we provide bounds on the expected improvement when lies in the RKHS.
If , then by 6, , so , and the result is trivial. Suppose , and set , . From (8) and (9),
and by 6, . As , is non-decreasing, and for . Hence
If , then as is the expectation of a non-negative quantity, , and the lower bounds are trivial. Suppose . Then as , for all , and . Thus
Combining these bounds, and eliminating , we obtain
We may now prove the theorem. We will use the above bounds to show that there must be times when the expected improvement is low, and thus is close to .
holds at most times. Furthermore, , and for ,
so at most times. Since , we have also at most times. Thus there is a time , , for which and .
Let have minimum at . For large, will have been chosen by expected improvement (rather than being an initial design point, chosen at random). Then as is non-increasing in , for we have by 8,
This bound is uniform in with , so we obtain
A.3 Estimated Parameters
To prove Theorem 3, we first establish lower bounds on the posterior variance.
uniformly in the sequences , .
Given design points , there must be some such that , . By 6, the posterior mean of given these observations is the zero function. Thus for minimizing ,
As is non-increasing in , for we obtain
Next, we bound the expected improvement when prior parameters are estimated by maximum likelihood.
Let , . Set , , and . Suppose:
for some , whenever ;
for some , .
Then for as in 2, eventually . If the conditions hold on a subsequence, so does the conclusion.
Let be given by (7), and set . For , , and by 4 and 1,
Thus . Then if , for some ,
If , then , so
When , as is increasing we may upper bound using , and lower bound using . Since , and as (Abramowitz and Stegun, 1965, §7.1),
If the conditions hold on a subsequence, we may similarly argue along that subsequence. ∎
Finally, we will require the following technical lemma.
Given , fix , and pick disjoint open sets . Then
We may now prove the theorem. We will construct a function on which the strategy never observes within a region . We may then construct a function , agreeing with except on , but having different minimum. As the strategy cannot distinguish between and , it cannot successfully find the minimum of both.
We work conditional on the event , having probability at least , that , and thus for all . Suppose infinitely often, so the are not all equal. By 7, , so on a subsequence with , we have
whenever . However, by 9, there are points with , and . Hence by 10, for some , contradicting the definition of .
Construct a smooth function by adding to a function which is 0 outside , and has minimum . Then , but on the event , cannot distinguish between and , and . Thus for ,
As the behaviour of is invariant under rescaling, we may scale to have norm , and the above remains true for some . ∎
As in the proof of Theorem 2, we will show there are times when the expected improvement is small, so must be close to the minimum. First, however, we must control the estimated parameters , .
If the are all equal, then by assumption the are dense in , so is constant, and the result is trivial. Suppose the are not all equal, and let be a random variable satisfying for some . Set . is a continuous positive function, so . Let . By 4, , so by 1, for ,
As in the proof of Theorem 2, we have a constant , and some , , for which and . Then for , , arguing as in Theorem 2 we obtain
We thus have a random variable satisfying for all , and the result follows. ∎
A.4 Near-Optimal Rates
To prove Theorem 5, we first show that the points chosen at random will be quasi-uniform in .
Let be i.i.d. random variables, distributed uniformly over , and define their mesh norm,
For any , there exists such that
We will partition into regions of size , and show that with high probability we will place an in each one. Then every point will be close to an , and the mesh norm will be small.
For large, , so by the generalized Chernoff bound of Panconesi and Srinivasan (1997, §3.1),
On the event , for all . For any , we then have for some , and for some . Thus
As this bound is uniform in , we obtain . Thus for ,
and as is non-increasing in , this bound holds also for . By a change of variables, we then obtain
and the result follows by choosing large. For general , as is bounded it can be partitioned into regions of measure , so we may argue similarly. ∎
We may now prove the theorem. We will show that the points must be quasi-uniform in , so posterior variances must be small. Then, as in the proofs of Theorems 2 and 4, we have times when the expected improvement is small, so is close to .
First suppose . Let the choose initial design points independent of , and suppose . Let be the event that of the points are chosen uniformly at random, so by a Chernoff bound,
Let be the event that one of the points is chosen by expected improvement, so
Finally, let be the event that and occur, and further the mesh norm , for the constant from 12. Set . Then by 12, since ,
for a constant not depending on .
Let have prior at time , with (fixed or estimated) parameters , . Suppose , and set , so by 4, . If , then by Narcowich et al. (2003, §6),
uniformly in , for a continuous function of . Hence on the event ,
for a constant depending only on , , , and . If , the same result holds by a similar argument.
On the event , we have some chosen by expected improvement, . Let have minimum at . Then by 8,
for a constant . (Under , we have ; otherwise by 1, so .) Thus, rearranging,
On the event , we have , so
As this bound is uniform in with , the result follows. If instead , the above argument holds for any . ∎