We are happy to finally announce the first release of mlrMBO on cran after a quite long development time. For the theoretical background and a nearly complete overview of mlrMBOs capabilities you can check our paper on mlrMBO that we presubmitted to arxiv.
The key features of mlrMBO are:
- Global optimization of expensive Black-Box functions.
- Multi-Criteria Optimization.
- Parallelization through multi-point proposals.
- Support for optimization over categorical variables using random forests as a surrogate.
For examples covering different scenarios we have Vignettes that are also available as an online documentation. For mlr users mlrMBO is especially interesting for hyperparameter optimization.
mlrMBO for mlr hyperparameter tuning was already used in an earlier blog post. Nonetheless we want to provide a small toy example to demonstrate the work flow of mlrMBO in this post.
First, we define an objective function that we are going to minimize:
set.seed(1) library(mlrMBO) ## Loading required package: mlr ## Loading required package: ParamHelpers ## Loading required package: smoof ## Loading required package: BBmisc ## ## Attaching package: 'BBmisc' ## The following object is masked from 'package:base': ## ## isFALSE ## Loading required package: checkmate fun = makeSingleObjectiveFunction( name = "SineMixture", fn = function(x) sin(x)*cos(x)/2 + 0.04 * sum(x^2), par.set = makeNumericParamSet(id = "x", len = 2, lower = -5, upper = 5) )
To define the objective function we use
makeSingleObjectiveFunction from the neat package smoof, which gives us the benefit amongst others to be able to directly visualize the function.
If you happen to be in need of functions to optimize and benchmark your optimization algorithm I recommend you to have a look at the package!
library(plot3D) plot3D(fun, contour = TRUE, lightning = TRUE)
Let’s start with the configuration of the optimization:
# In this simple example we construct the control object with the defaults: ctrl = makeMBOControl() # For this numeric optimization we are going to use the Expected # Improvement as infill criterion: ctrl = setMBOControlInfill(ctrl, crit = crit.ei) # We will allow for exactly 25 evaluations of the objective function: ctrl = setMBOControlTermination(ctrl, max.evals = 25L)
The optimization has to so start with an initial design. mlrMBO can automatically create one but here we are going to use a randomly sampled LHS design of our own:
library(ggplot2) des = generateDesign(n = 8L, par.set = getParamSet(fun), fun = lhs::randomLHS) autoplot(fun, render.levels = TRUE) + geom_point(data = des)
The points demonstrate how the initial design already covers the search space but is missing the area of the global minimum. Before we can start the Bayesian optimization we have to set the surrogate learner to Kriging. Therefore we use an mlr regression learner. In fact, with mlrMBO you can use any regression learner integrated in mlr as a surrogate allowing for many special optimization applications.
sur.lrn = makeLearner("regr.km", predict.type = "se", config = list(show.learner.output = FALSE))
Note: mlrMBO can automatically determine a good surrogate learner based on the search space defined for the objective function. For a purely numeric domain it would have chosen Kriging as well with some slight modifications to make it a bit more stable against numerical problems that can occur during optimization.
Finally, we can start the optimization run:
res = mbo(fun = fun, design = des, learner = sur.lrn, control = ctrl, show.info = TRUE) ## Computing y column(s) for design. Not provided. ## [mbo] 0: x=0.897,-3.42 : y = 0.123 : 0.0 secs : initdesign ## [mbo] 0: x=-3.76,-0.767 : y = 0.798 : 0.0 secs : initdesign ## [mbo] 0: x=-3.27,-2.48 : y = 0.623 : 0.0 secs : initdesign ## [mbo] 0: x=-1.53,1.73 : y = 0.291 : 0.0 secs : initdesign ## [mbo] 0: x=4.92,4.84 : y = 1.84 : 0.0 secs : initdesign ## [mbo] 0: x=1.52,-4.57 : y = 0.86 : 0.0 secs : initdesign ## [mbo] 0: x=-0.435,0.603 : y = -0.152 : 0.0 secs : initdesign ## [mbo] 0: x=2.66,3.25 : y = 0.473 : 0.0 secs : initdesign ## [mbo] 1: x=0.29,-0.0602 : y = 0.146 : 0.0 secs : infill_ei ## [mbo] 2: x=5,0.626 : y = 0.627 : 0.0 secs : infill_ei ## [mbo] 3: x=0.13,1.09 : y = 0.0779 : 0.0 secs : infill_ei ## [mbo] 4: x=-2.35,0.637 : y = -0.0493 : 0.0 secs : infill_ei ## [mbo] 5: x=4.65,-2.92 : y = 1.69 : 0.0 secs : infill_ei ## [mbo] 6: x=-1.22,0.511 : y = -0.34 : 0.0 secs : infill_ei ## [mbo] 7: x=-0.638,-2.33 : y = 0.438 : 0.0 secs : infill_ei ## [mbo] 8: x=-1.04,0.344 : y = -0.358 : 0.0 secs : infill_ei ## [mbo] 9: x=-1.4,0.166 : y = -0.406 : 0.0 secs : infill_ei ## [mbo] 10: x=-1.43,-3.82 : y = 1.05 : 0.0 secs : infill_ei ## [mbo] 11: x=-1.31,-0.247 : y = -0.397 : 0.0 secs : infill_ei ## [mbo] 12: x=-1.36,-0.0187 : y = -0.415 : 0.0 secs : infill_ei ## [mbo] 13: x=2.46,0.985 : y = 0.455 : 0.0 secs : infill_ei ## [mbo] 14: x=-1.73,-0.18 : y = -0.365 : 0.0 secs : infill_ei ## [mbo] 15: x=0.355,4.58 : y = 0.822 : 0.0 secs : infill_ei ## [mbo] 16: x=-5,5 : y = 2.14 : 0.0 secs : infill_ei ## [mbo] 17: x=-5,-5 : y = 2.14 : 0.0 secs : infill_ei res$x ## $x ##  -1.359755 -0.018670 res$y ##  -0.4148502
We can see that we have found the global optimum of \(y = -0.414964\) at \(x = (-1.35265,0)\) quite sufficiently.
Let’s have a look at the points mlrMBO evaluated.
Therefore we can use the
OptPath which stores all information about all evaluations during the optimization run:
opdf = as.data.frame(res$opt.path) autoplot(fun, render.levels = TRUE, render.contours = FALSE) + geom_text(data = opdf, aes(label = dob))
It is interesting to see, that for this run the algorithm first went to the local minimum on the top right in the 6th and 7th iteration but later, thanks to the explorative character of the Expected Improvement, found the real global minimum.
That is all good, but how do other optimization strategies perform?
Grid search is seldom a good idea. But especially for hyperparameter tuning it is still used. Probably because it kind of gives you the feeling that you know what is going on and have not left out any important area of the search space. In reality the grid is usually so sparse that it leaves important areas untouched as you can see in this example:
grid.des = generateGridDesign(par.set = getParamSet(fun), resolution = 5) grid.des$y = apply(grid.des, 1, fun) grid.des[which.min(grid.des$y),] ## x1 x2 y ## 12 -2.5 0 -0.04923607 autoplot(fun, render.levels = TRUE, render.contours = FALSE) + geom_point(data = grid.des)
It is no surprise, that the grid search could not cover the search space well enough and we only reach a bad result.
What about a simple random search?
random.des = generateRandomDesign(par.set = getParamSet(fun), n = 25L) random.des$y = apply(random.des, 1, fun) random.des[which.min(random.des$y),] ## x1 x2 y ## 17 0.0948004 1.061824 0.06852087 autoplot(fun, render.levels = TRUE, render.contours = FALSE) + geom_point(data = random.des)
With the random search you could always be lucky but in average the optimum is not reached if smarter optimization strategies work well.
A fair comarison
… for stochastic optimization algorithms can only be achieved by repeating the runs.
mlrMBO is stochastic as the initial design is generated randomly and the fit of the Kriging surrogate is also not deterministic.
Furthermore we should include other optimization strategies like a genetic algorithm and direct competitors like
An extensive benchmark is available in our mlrMBO paper.
The examples here are just meant to demonstrate the package.
If you want to contribute to mlrMBO we ware always open to suggestions and pull requests on github. You are also invited to fork the repository and build and extend your own optimizer based on our toolbox.