4  Hyperparameter Optimization

Marc Becker
Ludwig-Maximilians-Universität München

Lennart Schneider
Ludwig-Maximilians-Universität München, and Munich Center for Machine Learning (MCML)

Machine learning algorithms usually include parameters and hyperparameters. Parameters are the model coefficients or weights or other information that are determined by the learning algorithm based on the training data. In contrast, hyperparameters, are configured by the user and determine how the model will fit its parameters, i.e., how the model is built. Examples include setting the number of trees in a random forest, penalty settings in support vector machines, or the learning rate in a neural network.

Hyperparameters

The goal of hyperparameter optimization (Section 4.1) or model tuning is to find the optimal configuration of hyperparameters of an ML algorithm for a given task. There is no closed-form mathematical representation (nor analytic gradient information) for model agnostic HPO. Instead, we follow a black-box optimization approach: an ML algorithm is configured with values chosen for one or more hyperparameters, this algorithm is then evaluated (using a resampling method) and its performance is measured. This process is repeated with multiple configurations and finally the configuration with the best performance is selected. HPO closely relates to model evaluation (Chapter 3) as the objective is to find a hyperparameter configuration that optimizes the generalization performance. Broadly speaking, we could think of finding the optimal model configuration in the same way as selecting a model from a benchmark experiment, where in this case each model in the experiment is the same algorithm but with different hyperparameter configurations. For example, we could benchmark three support vector machines (SVMs) with three difference cost values. However, human trial-and-error is time-consuming, subjective and often biased, error-prone, and computationally inefficient. Instead, many sophisticated HPO methods (Section 4.1.4) (or ‘tuners’) have been developed over the past few decades for robust and efficient HPO. Besides simple approaches such as a random search or grid search, most HPO methods employ iterative techniques that propose different configurations over time, often exhibiting adaptive behavior guided towards potentially optimal hyperparameter configurations. These methods continually propose new configurations until a termination criterion is met, at which point the optimal configuration is returned. This iterative approach is depicted in a typical optimization loop as shown in Figure (Figure 4.1). For more general details on HPO and more theoretical background, we recommend Bischl et al. (2023) and Feurer and Hutter (2019).

Hyperparameter Optimization

Diagram showing 13 boxes representing model-agnostic HPO.  On the top are two boxes, one that says  "Search Space" and the other "Tuner", these are connected by a line to "Propose Hyperparameter Configurations". That box has an arrow pointing towards another box "Evaluate by Resampling", which has four blue boxes pointing towards it  "Task", "Learner", "Resampling", and "Measure". "Evaluate by Resampling" has one line below it connected to "Objective", and an arrow to the right connected to "Terminator". This "Terminator" box has an arrow pointing down to "Optimal Hyperparameter Configuration" and right to "Update Tuner". Finally the "Update Tuner" box has a line below it connecting to "Archive" and an arrow above it connecting back to "Propose Hyperparameter Configurations".

Figure 4.1: Representation of the hyperparameter optimization loop in mlr3tuning. Blue - Hyperparameter optimization loop. Purple - Objects of the tuning instance supplied by the user. Blue-Green - Internally created objects of the tuning instance. Green - Optimization Algorithm.

4.1 Model Tuning

mlr3tuning is the hyperparameter optimization package of the mlr3 ecosystem. At the heart of the package are the R6 classes

In this section, we will cover these classes as well as other supporting functions and classes. Throughout this section, we will look at optimizing an SVM classifier from e1071 on the sonar dataset as a running example.

4.1.1 Learner and Search Space

The tuning process begins by deciding which hyperparameters to tune and what range to tune them over. The first place to start is therefore picking a learner and looking at the possible hyperparameters to tune with $param_set:

as.data.table(lrn("classif.svm")$param_set)[, .(id, class, lower, upper, nlevels)]
                 id    class lower upper nlevels
 1:       cachesize ParamDbl  -Inf   Inf     Inf
 2:   class.weights ParamUty    NA    NA     Inf
 3:           coef0 ParamDbl  -Inf   Inf     Inf
 4:            cost ParamDbl     0   Inf     Inf
 5:           cross ParamInt     0   Inf     Inf
 6: decision.values ParamLgl    NA    NA       2
 7:          degree ParamInt     1   Inf     Inf
 8:         epsilon ParamDbl     0   Inf     Inf
 9:          fitted ParamLgl    NA    NA       2
10:           gamma ParamDbl     0   Inf     Inf
11:          kernel ParamFct    NA    NA       4
12:              nu ParamDbl  -Inf   Inf     Inf
13:           scale ParamUty    NA    NA     Inf
14:       shrinking ParamLgl    NA    NA       2
15:       tolerance ParamDbl     0   Inf     Inf
16:            type ParamFct    NA    NA       2

Given infinite resources, we could tune all hyperparameters jointly, but in reality that is not possible, so usually only a subset of hyperparameters can be tuned. This subset is referred to as the search space or tuning space. In this example we will tune the numeric regularization and kernel width hyperparameters, cost and gamma; see the help page for e1071::svm() for full details on the model implementation including hyperparameters. In practice, search spaces are usually more complex and can require expert knowledge to define them. Section 4.4 provides more detailed insight into the creation of tuning spaces, including using mlr3tuningspaces to load search spaces that have been established in published scientific articles.

Search Space
Tip

In rare cases parameter sets may include hyperparameters that cannot be tuned. These will usually be ‘technical’ (or ‘control’) parameters that provide information about how the model is being fit but do not control the training process itself, for example the verbose hyperparameter in classif.ranger controls how much information is displayed to the user during training.

For numeric hyperparameters (we will explore others later) one must specify the bounds to tune over. We do this by constructing a learner and using to_tune() to set the lower and upper limits for the parameters we want to tune. This function allows us to mark the hyperparameter for subsequent tuning in the specified range.

learner = lrn("classif.svm",
  type  = "C-classification",
  kernel = "radial",
  cost  = to_tune(1e-1, 1e5),
  gamma = to_tune(1e-1, 1)
)
learner
<LearnerClassifSVM:classif.svm>
* Model: -
* Parameters: type=C-classification, kernel=radial,
  cost=<RangeTuneToken>, gamma=<RangeTuneToken>
* Packages: mlr3, mlr3learners, e1071
* Predict Types:  [response], prob
* Feature Types: logical, integer, numeric
* Properties: multiclass, twoclass

Here we have constructed a classification SVM, classif.svm, selected the type of model as "C-classification", set the kernel to "radial", and specified that we plan to tune the cost and gamma parameters over the range \([0.1,100000]\) and \([0.1,1]\) respectively (though these are usually tuned on a log scale, see Section 4.1.5). Note that calling $train() on a learner with a tune token (e.g., cost=<RangeTuneToken>) will throw an error.

Now we have decided which hyperparameters to tune, we must now specify when to stop the tuning process.

4.1.2 Terminator

mlr3tuning includes many methods to specify when to terminate an algorithm (Table 4.1), which are implemented in Terminator classes.

Terminator
Table 4.1: Terminators available in mlr3tuning, their function call and default parameters. A complete and up-to-date list can be found at https://mlr-org.com/terminators.html.
Terminator Function call and default parameters
Clock Time trm("clock_time", stop_time = "2022-11-06 08:42:53 CET")
Combo trm("combo", terminators = list(run_time_100, evals_200, any = TRUE))
None trm("none")
Number of Evaluations trm("evals", n_evals = 500)
Performance Level trm("perf_reached", level = 0.1)
Run Time trm("run_time", secs = 100)
Stagnation trm("stagnation", iters = 5, threshold = 1e-5)

The most commonly used terminators are those that stop the tuning after a certain time (trm("run_time")) or the number of evaluations (trm("evals")). Choosing a runtime is often based on practical considerations and intuition. Using a time limit can be important on compute clusters where a maximum runtime for a compute job may need to be specified. The trm("perf_reached") terminator stops the tuning when a certain performance level is reached, which can be helpful if a certain performance is seen as sufficient for the practical use of the model, however if this is set too optimistically the tuning may never terminate. The trm("stagnation") terminator stops when no progress is made for a number of iterations. The minimum progress is specified by the threshold argument, note this can be difficult to select as the optimization could stop too soon for complex search spaces despite room for (possibly significant) improvement. trm("none") is used for tuners that control termination themselves. Finally, any of these terminators can be freely combined with the trm("combo") terminator, which can be used to specify if HPO finishes when any (any = TRUE) terminator is triggered or when all (any = FALSE) are triggered.

4.1.3 Tuning Instance with ti

The tuning instance collects together the tuner-agnostic information required to optimize a model, i.e., all information about the tuning process, except for the tuning algorithm itself. This includes the task to tune over, the learner to tune, the resampling method and measure used to analytically compare HPO configurations, and the terminator to determine when the measure has been optimized ‘enough’.

A tuning instance can be constructed manually with the ti() function, or automated (?sec-simplified-tuning) with the tune() function. We cover the manual approach first as this allows finer control of tuning and a more nuanced discussion about the design and use of mlr3tuning.

Tuning Instance

Continuing our example, we will construct a single-objective tuning problem (i.e., tuning over one measure) by using the ti() function to create a TuningInstanceSingleCrit, we will return to multi-objective tuning in Section 4.5.

For this example we will use three-fold cross-validation and optimize the classification error measure. Note that in the next section we will continue our example with a grid search tuner, so we select trm("none").

task_sonar = tsk("sonar")

learner = lrn("classif.svm",
  cost  = to_tune(1e-1, 1e5),
  gamma = to_tune(1e-1, 1),
  kernel = "radial",
  type = "C-classification"
)

instance = ti(
  task = task_sonar,
  learner = learner,
  resampling = rsmp("cv", folds = 3),
  measures = msr("classif.ce"),
  terminator = trm("none")
)

instance
<TuningInstanceSingleCrit>
* State:  Not optimized
* Objective: <ObjectiveTuning:classif.svm_on_sonar>
* Search Space:
      id    class lower upper nlevels
1:  cost ParamDbl   0.1 1e+05     Inf
2: gamma ParamDbl   0.1 1e+00     Inf
* Terminator: <TerminatorNone>

4.1.4 Tuner

With all the pieces of our tuning problem assembled, we can now decide how to tune our model. There are multiple Tuner classes in mlr3tuning, which implement different HPO (or more generally speaking black-box optimization) algorithms (Table 4.2).

Tuner
Table 4.2: Tuning algorithms available in mlr3tuning, their function call and the package in which the algorithm is implemented. A complete and up-to-date list can be found at https://mlr-org.com/tuners.html.
Tuner Function call Package
Random Search tnr("random_search") mlr3tuning
Grid Search tnr("grid_search") mlr3tuning
Bayesian Optimization tnr("mbo") mlr3mbo
CMA-ES tnr("cmaes") adagio
Iterative Racing tnr("irace") irace
Hyperband tnr("hyperband") mlr3hyperband
Generalized Simulated Annealing tnr("gensa") GenSA
Nonlinear Optimization tnr("nloptr") nloptr

Search strategies

Grid search and random search (Bergstra and Bengio 2012) are the most basic algorithms and are often selected first in initial experiments. The idea of grid search is to exhaustively evaluate every possible combination of given hyperparameter values. Categorical hyperparameters are usually evaluated over all possible values they can take while values to consider for numeric hyperparameters are determined by a given resolution, which is the number of distinct values to try per hyperparameter. Numeric and integer hyperparameters are spaced equidistantly in their box constraints (upper and lower bounds), and categorical hyperparameters usually have all possible values considered. Random search involves randomly selecting values for each hyperparameter independently from a pre-specified distribution, usually uniform. Both methods are non-adaptive, which means each proposed configuration ignores the performance of previous configurations. Due to their simplicity, both grid search and random search can handle mixed search spaces (i.e., hyperparameters can be numeric, integer, or categorical) as well as hierarchical search spaces (Section 4.4).

Adaptive algorithms

Adaptive algorithms learn from previously evaluated configurations to find good configurations more quickly, examples in mlr3 include Bayesian optimization (also called model-based optimization), Covariance Matrix Adaptation Evolution Strategy (CMA-ES), Iterative Racing, and Hyperband.

Bayesian optimization (Chapter 13) (e.g., Snoek, Larochelle, and Adams 2012) describes a family of iterative optimization algorithms that use a surrogate model to approximate the unknown function that is to be optimized – in HPO the mapping from a hyperparameter configuration to the estimated generalization performance. Any Bayesian optimization algorithm starts by observing an initial design of observations and then trains the surrogate model on all data points and performance values observed so far. The algorithm then uses an acquisition function that usually relies on both the mean and variance prediction of the surrogate model to determine which points of the search space are promising candidates that should be evaluated next. By optimizing the acquisition function itself, the next candidate point is choosen for evaluation, evaluated and the process repeats itself by re-fitting or updating the surrogate model on the updated set of observed data points. Bayesian optimization is very flexible (e.g. mixed and hierarchical search spaces can easily be handled by choosing a suitable surrogate model), and highly sample efficient, i.e., compared to other algorithms, much less function evaluations are needed to find good configurations. On the downside, the optimization overhead of Bayesian optimization is comparably large (e.g., in each iteration, training of the surrogate model and optimizing the acquisition function) and therefore really shines in the context of very costly function evaluations and tight optimization budget.

CMA-ES (Hansen and Auger 2011) is an evolutionary strategy that maintains a probability distribution over candidate points, with the distribution represented by a mean vector and covariance matrix. A new set of candidate points is generated by sampling from this distribution, with the probability of each candidate being proportional to its performance. The covariance matrix is adapted over time to reflect the performance landscape.

Racing algorithms work by iteratively discarding configurations that show poor performance, as determined by statistical tests. Iterative Racing (López-Ibáñez et al. 2016) starts by ‘racing’ down an initial population of randomly sampled configurations from a parametrized density and then uses the surviving configurations of the race to stochastically update the density of the subsequent race to focus on promising regions of the search space, and so on.

Multi-fidelity HPO is an adaptive method that leverages the predictive power of computationally cheap lower fidelity evaluations (i.e., poorer quality predictions such as those arising from neural networks with a small number of epochs) to improve the overall optimization efficiency. This concept is used in Hyperband (Section 4.6), a popular multi-fidelity HPO algorithm that dynamically allocates increasingly more resources to promising configurations and terminates low-performing ones. Hyperband tends to outperform random search as the optimization budget is used much more efficiently.

Other implemented algorithms for numeric search spaces are Generalized Simulated Annealing (Xiang et al. 2013; Tsallis and Stariolo 1996) and various nonlinear optimization algorithms. These can be useful if evaluations are rather cheap as they require more function evaluations and are therefore less sample efficient than other sophisticated methods (e.g., MBO) but are more commonly used for general, numeric black-box optimization.

Choosing strategies

As a rule of thumb, if the search space is small and consists of categorical hyperparameters, a grid search may be able to exhaustively evaluate the entire search space in a reasonable time. However, grid search is generally not recommended due to the curse of dimensionality – the grid size ‘blows up’ very quickly as number of parameters to tune increases – and insufficient coverage of numeric search spaces. Grid search by construction also cannot evaluate a large number of unique values per hyperparameter, which is suboptimal when some hyperparameters have minimal impact on performance while others do.

In such scenarios, a random search is often a better choice as it considers more unique values per hyperparameter compared to grid search. Random search is also suitable for small optimization budgets and quick concept experiments. However, with larger optimization budgets, more guided optimization algorithms such as evolutionary strategies or Bayesian optimization tend to perform better and are more likely to result in peak performance.

When choosing between evolutionary strategies and Bayesian optimization, the cost of function evaluation is highly relevant. If hyperparameter configurations can be evaluated quickly, evolutionary strategies often find good configurations within a reasonable timeframe. On the other hand, if model evaluations are time-consuming and the optimization budget is limited, Bayesian optimization is usually preferred.

Finally, in cases where the HPO problem involves a meaningful fidelity parameter (e.g., number of epochs, number of trees, number of boosting rounds) and optimization budget needs to be spent efficiently, multi-fidelity HPO algorithms like Hyperband may be worth considering. For further details on different tuners and practical recommendations, we refer to Bischl et al. (2023).

Tip

The $param_classes and $properties fields of a Tuner respectively provide information about which classes of hyperparameters can be handled and what properties the tuner can handle (e.g., hyperparameter dependencies (Section 4.4)):

tnr("random_search")$param_classes
[1] "ParamLgl" "ParamInt" "ParamDbl" "ParamFct"
tnr("random_search")$properties
[1] "dependencies" "single-crit"  "multi-crit"  

For our SVM example, we will use a grid search with a resolution of 5 for efficiency reasons here (in practice a larger resolution would be preferred). Recall that the resolution is the number of distinct values to try per hyperparameter, which means in our example the tuner will construct a 5x5 grid of 25 configurations of equally spaced points between the specified upper and lower bounds. All configurations will be tried by the tuner (in random order) until either all configurations are evaluated or the terminator (Section 4.1.2) signals that the budget is exhausted. For grid and random search tuners, the batch_size parameter controls how many configurations are evaluated at the same time (see Section 9.1) and also determines how many configurations should be applied before the terminator should check if the termination criterion has been reached.

tuner = tnr("grid_search", resolution = 5, batch_size = 10)
tuner
<TunerGridSearch>: Grid Search
* Parameters: resolution=5, batch_size=10
* Parameter classes: ParamLgl, ParamInt, ParamDbl, ParamFct
* Properties: dependencies, single-crit, multi-crit
* Packages: mlr3tuning

The resolution and batch_size parameters are termed control parameters, and other tuners will have other parameters that can be set, as with learners these are accessible with $param_set.

Control Parameters
tuner$param_set
<ParamSet>
                  id    class lower upper nlevels        default value
1:        batch_size ParamInt     1   Inf     Inf <NoDefault[3]>    10
2:        resolution ParamInt     1   Inf     Inf <NoDefault[3]>     5
3: param_resolutions ParamUty    NA    NA     Inf <NoDefault[3]>      

Whilst changing the control parameters of the tuner can improve optimal performance, it is likely the default settings will fit most needs.

Triggering the tuning process

Now that we have all our components, we can start the tuning process. To do this we simply pass the constructed TuningInstanceSingleCrit to the $optimize() method of the initialized Tuner, which triggers the HPO loop (Figure 4.1).

tuner$optimize(instance)
       cost gamma learner_param_vals  x_domain classif.ce
1: 25000.08   0.1          <list[4]> <list[2]>  0.2448585

The optimizer returns the best hyperparameter configuration and the corresponding performance, this information is also stored in instance$result. The first columns (here cost and gamma) will be named after the tuned hyperparameters and show the optimal values from the searched tuning spaces. The $learner_param_vals field lists the optimal hyperparameters from tuning, as well as the values of any other hyperparameters that were manually changed from the defaults, this is useful for onward model use (Section 4.1.6).

instance$result$learner_param_vals
[[1]]
[[1]]$kernel
[1] "radial"

[[1]]$type
[1] "C-classification"

[[1]]$cost
[1] 25000.08

[[1]]$gamma
[1] 0.1

The $x_domain field contains the optimal values of the tuned hyperparameters, which is most useful in the context of hyperparameter transformations, which we will briefly turn to next.

Warning

A common mistake when tuning is to report the performance estimated on the resampling sets on which the tuning was performed (instance$result$classif.ce) as an estimate of the model’s performance. The correct method is to test the model on more unseen data, which can be efficiently performed with nested resampling, we will discuss this in Section 4.3.2.

4.1.5 Logarithmic Transformations

For many hyperparameters that have infinite (or very large) bounds, tuning on a logarithmic scale is more efficient than tuning on a linear scale. By example, consider sampling uniformly in the interval \([log(1e-5), log(1e5)]\) and then exponentiating the outcome:

cost = runif(1000, log(1e-5), log(1e5))
exp_cost = exp(cost)

The histograms below show how we are initially sampling within a narrow range (\([-11.5, 11.5]\)) then exponentiating results in the majority of points being relatively small but a few being very large.

Histogram showing the distribution of uniformly sampled values from the interval [log(1e-5), log(1e5)]. The left plot shows the values on the linear scale sampled by the tuner between [-11.5,11.5] and the right plot shows the transformed values on the logarithmic scale between [1e-5, 1e5].

(a) Values on the linear scale sampled by the tuner.

Histogram showing the distribution of uniformly sampled values from the interval [log(1e-5), log(1e5)]. The left plot shows the values on the linear scale sampled by the tuner between [-11.5,11.5] and the right plot shows the transformed values on the logarithmic scale between [1e-5, 1e5].

(b) Transformed values on the logarithmic scale as seen by the learner.

Figure 4.2: Histogram of uniformly sampled values from the interval \([log(1e-5), log(1e5)]\).

To add this transformation to a hyperparameter we simply pass logscale = TRUE to to_tune().

learner = lrn("classif.svm",
  cost  = to_tune(1e-5, 1e5, logscale = TRUE),
  gamma = to_tune(1e-5, 1e5, logscale = TRUE),
  kernel = "radial",
  type = "C-classification"
)

instance = ti(
  task = task_sonar,
  learner = learner,
  resampling = rsmp("cv", folds = 3),
  measures = msr("classif.ce"),
  terminator = trm("none")
)

tuner$optimize(instance)
       cost     gamma learner_param_vals  x_domain classif.ce
1: 5.756463 -5.756463          <list[4]> <list[2]>  0.1394065

We can see from this example that using the log transformation improved our model training as classif.ce is smaller.

Note that the fields cost and gamma show the optimal values before transformation, whereas x_domain and learner_param_vals contain optimal values after transformation, it is these latter fields you would take forward for future model use.

instance$result$x_domain
[[1]]
[[1]]$cost
[1] 316.2278

[[1]]$gamma
[1] 0.003162278

In ?sec-search-space-scratch we will look at how to implement more complex, customt transformations to any hyperparameter or combination of hyperparameters. Now we will look at how to put everything into practice so we can make use of the tuned model (and the transformed hyperparameters).

4.1.6 Analyzing and using the result

Independently of whether you use ti or tune, or if you include transformations or not, the output is the same and the instance’s archive lists all evaluated hyperparameter configurations:

as.data.table(instance$archive)[, .(cost, gamma, classif.ce)]
          cost      gamma classif.ce
 1: -11.512925  -5.756463  0.4664596
 2:  -5.756463 -11.512925  0.4664596
 3:  -5.756463  11.512925  0.4664596
 4:   0.000000  -5.756463  0.2308489
 5:   5.756463  -5.756463  0.1394065
---                                 
21:   0.000000   5.756463  0.4664596
22:   0.000000  11.512925  0.4664596
23:   5.756463  11.512925  0.4664596
24:  11.512925 -11.512925  0.2161491
25:  11.512925   0.000000  0.4664596

Each row of the archive is a different evaluated configuration. The columns here show the tested configurations (before transformation) and the chosen performance measure. We can also manually inspect the archive to determine other important features such as time of evaluation, model runtime, and any errors or warnings that occurred during tuning.

as.data.table(instance$archive)[1:3,
  .(timestamp, runtime_learners, errors, warnings)]
             timestamp runtime_learners errors warnings
1: 2023-06-04 13:04:38            0.067      0        0
2: 2023-06-04 13:04:38            0.061      0        0
3: 2023-06-04 13:04:38            0.067      0        0

Another powerful feature of the instance is that we can score the internal ResampleResults on a different performance measure, for example looking at false negative rate and false positive rate as well as classification error:

as.data.table(instance$archive,
  measures = msrs(c("classif.fpr", "classif.fnr")))[1:5 ,
  .(cost, gamma, classif.ce, classif.fpr, classif.fnr)]
         cost      gamma classif.ce classif.fpr classif.fnr
1: -11.512925  -5.756463  0.4664596   1.0000000  0.00000000
2:  -5.756463 -11.512925  0.4664596   1.0000000  0.00000000
3:  -5.756463  11.512925  0.4664596   1.0000000  0.00000000
4:   0.000000  -5.756463  0.2308489   0.3185868  0.14996957
5:   5.756463  -5.756463  0.1394065   0.2089094  0.08056151

You can access all the resamplings in a BenchmarkResult object with instance$archive$benchmark_result.

Finally, to visualize the results, you can use the autoplot implemented in mlr3viz (Figure 4.3). In this example we can observe one of the flaws (by design) in grid search, despite testing 25 configurations, we only saw 5 unique values for each hyperparameter.

autoplot(instance, type = "surface")

Heatmap showing model performance during HPO. y-axis is 'gamma' parameter between (-10,10) and x-axis is 'cost' parameter between (-10,10). The heatmap shows squares covering all points on the plot and circular points indicating configurations tried in our optimisation. The top-left quadrant is all yellow indicating poor performance when gamma is high and cost is low. The bottom-right is dark blue indicating good performance when cost is high and gamma is low.

Figure 4.3: Model performance with different configurations for cost and gamma. Bright yellow regions represent the model performing worse and dark blue performing better. We can see that high cost values and low gamma values achieve the best performance. Note that we should not directly infer the performance of new unseen values from the heatmap since it is only an interpolation based on a surrogate model (regr.ranger). However, we can see the general interaction between the hyperparameters.

Training an optimized model

Once the learner has been tuned we can start to use it like any other model in the mlr3 universe. To do this we simply construct a new learner with the same underlying algorithm and set the learner hyperparameters to the optimal configuration:

svm_tuned = lrn("classif.svm")
svm_tuned$param_set$values = instance$result_learner_param_vals

Now we can train the learner on the full dataset and we are ready to make predictions. The trained model can then be used to predict new, external data:

svm_tuned$train(task_sonar)$model

Call:
svm.default(x = data, y = task$truth(), type = "C-classification", 
    kernel = "radial", gamma = 0.00316227766016838, cost = 316.227766016838, 
    probability = (self$predict_type == "prob"))


Parameters:
   SVM-Type:  C-classification 
 SVM-Kernel:  radial 
       cost:  316.2278 

Number of Support Vectors:  93

4.2 Automating with tune and auto_tuner

In the previous section, we looked at constructing and manually putting together the components of HPO by creating a tuning instance using ti(), passing this to the tuner, and then calling $optimize to start the tuning process. mlr3tuning includes two helper methods to simplify this process further.

The first helper function is tune(), which creates the tuning instance and calls $optimize for you. You may prefer the manual method with ti if you want to view and make changes to the instance before tuning.

grid_search = tnr("grid_search", resolution = 5, batch_size = 5)
lrn_svm = lrn("classif.svm",
  cost  = to_tune(1e-5, 1e5, logscale = TRUE),
  gamma = to_tune(1e-5, 1e5, logscale = TRUE),
  kernel = "radial",
  type = "C-classification"
)
rsmp3 = rsmp("cv", folds = 3)
msr_ce = msr("classif.ce")

tune(
  tuner = grid_search,
  task = task_sonar,
  learner = lrn_svm,
  resampling = rsmp3,
  measures = msr_ce
)$result
       cost     gamma learner_param_vals  x_domain classif.ce
1: 5.756463 -5.756463          <list[4]> <list[2]>  0.1443754

The other helper function is auto_tuner, which creates an object of class AutoTuner. The AutoTuner inherits from the Learner class and wraps all the information needed for tuning, which means you can treat a learner waiting to be optimized just like any other learner. Under the hood, the AutoTuner essentially runs tune on the data that is passed to the model when $train is called and then sets the learner parameters with the optimal configurations.

at = auto_tuner(
  tuner = grid_search,
  learner = lrn_svm,
  resampling = rsmp3,
  measure = msr_ce
)

at
<AutoTuner:classif.svm.tuned>
* Model: list
* Search Space:
<ParamSet>
      id    class     lower    upper nlevels        default value
1:  cost ParamDbl -11.51293 11.51293     Inf <NoDefault[3]>      
2: gamma ParamDbl -11.51293 11.51293     Inf <NoDefault[3]>      
Trafo is set.
* Packages: mlr3, mlr3tuning, mlr3learners, e1071
* Predict Type: response
* Feature Types: logical, integer, numeric
* Properties: multiclass, twoclass

And we can now call $train(), which will first tune the hyperparameters in the search space listed above before fitting the optimal model.

split = partition(task_sonar)
at$train(task_sonar, row_ids = split$train)
at$predict(task_sonar, row_ids = split$test)$score()
classif.ce 
 0.2028986 

The AutoTuner contains a tuning instance that can be analyzed like any other instance.

at$tuning_instance$result
       cost     gamma learner_param_vals  x_domain classif.ce
1: 5.756463 -5.756463          <list[4]> <list[2]>  0.1726796

We could also pass the AutoTuner to resample() and benchmark(), which would result in a nested resampling, discussed next.

4.3 Nested Resampling

Hyperparameter optimization requires additional resampling to reduce bias when estimating performance of the model. If the same data is used for determining the optimal configuration and the evaluation of the resulting model itself, the actual performance estimate of the model might be severely biased (Simon 2007). This is analogous to optimism of the training error described in James et al. (2013), which occurs when training error is taken as an estimate of out-of-sample performance.

Nested resampling separates model optimization from the process of estimating the performance of the tuned model by adding an additional resampling, i.e., while model performance is estimated using a resampling method in the ‘usual way’, tuning is then performed by resampling the resampled data (Figure 4.4). For more details and a formal introduction to nested resampling the reader is referred to Bischl et al. (2023) and Simon (2007).

Nested Resampling

The image shows three rows of large blocks representing three-fold cross-validation for the outer resampling. Below the blocks are four further rows of small blocks representing four-fold cross-validation for the inner resampling. The training sets are represented in light blue and the test sets in dark blue.

Figure 4.4: An illustration of nested resampling. The large blocks represent 3-fold cross-validation for the outer resampling for model evaluation and the small blocks represent 4-fold cross-validation for the inner resampling for HPO. The light blue blocks are the training sets and the dark blue blocks are the test sets.

Figure 4.4 represents the following example of nested resampling:

  1. Outer resampling – Instantiate 3-fold cross-validation to create different testing and training datasets.
  2. Inner resampling – Within the outer training data instantiate 4-fold cross-validation to create different inner testing and training datasets.
  3. HPO – Tune the hyperparameters on the outer training set (large, light blue blocks) using the inner data splits.
  4. Training – Fit the learner on the outer training dataset using the optimal hyperparameter configuration obtained from the inner resampling (small blocks).
  5. Evaluation – Evaluate the performance of the learner on the outer testing data (large, dark blue block).
  6. Cross-validation – Repeat (2)-(5) for each of the three folds.
  7. Aggregation – Take the sample mean of the three performance values for an unbiased performance estimate.

The inner resampling produces generalization performance estimates for each configuration and selects the optimal configuration to be evaluated on the outer resampling. The outer resampling then produces generalization estimates for these optimal configurations. The result from the outer resampling can be used for comparison to other models trained and tested on the same outer folds.

Tip

Nested resampling is computationally expensive, three outer folds and four inner folds with a grid search of resolution 5 used to tune 2 parameters, results in 3*4*5*5 = 300 iterations of model training/testing. If you have the resources we recommend utilizing parallelization when tuning (Section 9.1).

A common mistake is to try and use nested resampling as a method to select optimal model configurations. Nested resampling is a method to compare models and to estimate the generalization performance of a tuned model, however, this is the performance based on multiple different configurations (one from each outer fold) and not performance based on a single configuration (Section 4.3.2). If you are interested in identifying optimal configurations, then use tune/ti or auto_tuner with $train().

4.3.1 Resampling an AutoTuner

Whilst the theory of nested resampling may seem complicated, it is all automated in mlr3tuning by simply passing an AutoTuner to resample() or benchmark(). Continuing with our previous example, we will use the auto-tuner to resample a support vector classifier with 3-fold cross-validation in the outer-resampling and 4-fold cross-validation in the inner resampling.

at = auto_tuner(
  tuner = grid_search,
  learner = lrn_svm,
  resampling = rsmp("cv", folds = 4),
  measure = msr_ce,
)

rr = resample(task_sonar, at, rsmp3, store_models = TRUE)

rr
<ResampleResult> with 3 resampling iterations
 task_id        learner_id resampling_id iteration warnings errors
   sonar classif.svm.tuned            cv         1        0      0
   sonar classif.svm.tuned            cv         2        0      0
   sonar classif.svm.tuned            cv         3        0      0

Note that we set store_models = TRUE so that the AutoTuner models (fitted on the outer training data) are stored, which also enables investigation of the inner tuning instances. Whilst we used K-fold cross-validation for both the inner and outer resampling strategy, you could use different resampling strategies (Section 3.2) and also different parallelization methods (Section 9.1.4).

The estimated performance of a tuned model is reported as the aggregated performance of all outer resampling iterations, which is an unbiased estimate of future model performance.

rr$aggregate()
classif.ce 
 0.1589372 

In addition to the methods described in Section 3.2, extract_inner_tuning_results() and extract_inner_tuning_archives() return the optimal configurations (across all outer folds) and full tuning archives, respectively.

extract_inner_tuning_results(rr)[,
  .(iteration, cost, gamma, classif.ce)]
   iteration     cost     gamma classif.ce
1:         1 11.51293 -5.756463  0.2174370
2:         2 11.51293 -5.756463  0.2086134
3:         3 11.51293 -5.756463  0.1796218
extract_inner_tuning_archives(rr)[1:3,
  .(iteration, cost, gamma, classif.ce)]
   iteration      cost      gamma classif.ce
1:         1 -11.51293 -11.512925  0.5285714
2:         1 -11.51293  11.512925  0.5285714
3:         1   0.00000  -5.756463  0.2981092

4.3.2 The right (and wrong) way to estimate generalization performance

This section covers advanced ML or technical details that can be skipped.

In this short section we will demonstrate with a brief example that only nested resampling with the $aggregate() method provides a reliable estimate of the model’s generalization performance.

We will do this by tuning and (un-nested) resampling a learner on training data and evaluating the in-sample performance, which is the performance of the best performing hyperparameter. Then we will compute the out-of-sample, generalization, performance, which is the performance of the tuned learner trained on the full dataset and evaluated on a test dataset. Finally, we will show how using nested resampling returns a good estimate of the generalization performance, even when only the training data is used.

In this experiment we tune several parameters from classif.xgboost, in doing so we can simultaneously show that nested resampling returns the best estimate of the generalization performance and that the insample performance will be overly-optimistic if hyperparameters are ‘meta’-overfitting the data (we use the term ‘meta-overfitting’ as they are overfitting to a subset of the training data during resampling as opposed to the full training data). To best estimate the generalization performance we make use of the "moons" TaskGenerator. The TaskGenerator class is used when you want to simulate data for use in experiments, these are very useful in cases such as this experiment when you need access to an infinite number of datapoints to estimate quantities such as the generalization error.

TaskGenerator

We begin by loading our learner, task generator, and generating 100 training data points and 1,000,000 testing data points.

XGB = lrn("classif.xgboost",
  eta               = to_tune(1e-4, 1, logscale = TRUE),
  max_depth         = to_tune(1, 20),
  colsample_bytree  = to_tune(1e-1, 1),
  colsample_bylevel = to_tune(1e-1, 1),
  lambda            = to_tune(1e-3, 1e3, logscale = TRUE),
  alpha             = to_tune(1e-3, 1e3, logscale = TRUE),
  subsample         = to_tune(1e-1, 1)
)
task = tgen("moons")
train_task = task$generate(100)
test_task = task$generate(1000000)

Now we will tune the learner with respect to the classification error and use holdout resampling and random search with 700 evaluations, we compute the in-sample performance as the performance for the optimal hyperparameter configuration.

tnr_rdm = tnr("random_search")
holdout = rsmp("holdout")
trm_700 = trm("evals", n_evals = 700)

instance = tune(
  tuner = tnr_rdm,
  task = train_task,
  learner = XGB,
  resampling = holdout,
  measures = msr_ce,
  terminator = trm_700
)

insample = instance$result_y

Next we compute the out-of-sample performance, which is the aggregate of scores over the outer resampling method (below we use 5-fold CV) using an `AutoTuner

# same setup as above
at = auto_tuner(
  tuner = tnr_rdm,
  learner = XGB,
  resampling = holdout,
  measure = msr_ce,
  terminator = trm_700
)

outsample = resample(train_task, at, rsmp("cv", folds = 5))$aggregate()

And finally we estimate the generalization error by training the tuned learner (i.e., using the values from the instance above) on the full training data again and predicting on the test data.

xgb_tuned = lrn("classif.xgboost")
xgb_tuned$param_set$set_values(
  .values = instance$result_learner_param_vals)
generalization = xgb_tuned$train(train_task)$predict(test_task)$score()

Now we can compare these three values:

round(c(generalization = as.numeric(generalization),
  insample = as.numeric(insample),
  outsample = as.numeric(outsample)), 2)
generalization       insample      outsample 
          0.31           0.06           0.20 

We find that the insample performance overestimates the model performance, which indicates meta-overfitting to the optimal hyperparameters. The out-of-sample performance from nested resampling provides a much better estimate of the generalization performance, despite the nested resampling only accessing 100 data points and not all 1,000,000.

4.4 More Advanced Search Spaces

Up until now we have only considered tuning overly simplistic search spaces limited to numeric hyperparameters. In this section we will first look at how to tune different scalar parameter classes with to_tune, and then how to define your own search space with ParamSet to create more advanced search spaces that may include tuning over vectors, transformations, and handling parameter dependencies. Finally, we will consider how to access a database of standardized search spaces from the literature.

4.4.1 Scalar parameter tuning

The to_tune function can be used to tune parameters of any class, whether they are scalar or vectors. To best understand this function, we will consider what is happening behind the scenes. When to_tune is used in a learner, implicitly a ParamSet is created just for the tuning search space:

learner = lrn("classif.svm",
  cost  = to_tune(1e-1, 1e5),
  gamma = to_tune(1e-1, 1),
  kernel = "radial",
  type = "C-classification"
)

learner$param_set$search_space()
<ParamSet>
      id    class lower upper nlevels        default value
1:  cost ParamDbl   0.1 1e+05     Inf <NoDefault[3]>      
2: gamma ParamDbl   0.1 1e+00     Inf <NoDefault[3]>      

Recall from Section 2.2.3, that the class field corresponds to the hyperparameter class as defined in paradox. In this example, we can see that gamma hyperparameter has class ParamDbl, with lower = 0.1 and upper = 1, which was automatically created by to_tune as we passed two numeric values to this function. If we wanted to tune over a non-numeric hyperparameter, we can still use to_tune, which will ‘guess’ the correct class to construct in the resulting parameter set. For example, say we wanted to tune the numeric cost, factor kernel, and logical scale hyperparameter in our SVM:

learner = lrn("classif.svm",
  cost  = to_tune(1e-1, 1e5),
  kernel = to_tune(c("radial", "linear")),
  shrinking = to_tune(),
  type = "C-classification"
)

learner$param_set$search_space()
<ParamSet>
          id    class lower upper nlevels        default value
1:      cost ParamDbl   0.1 1e+05     Inf <NoDefault[3]>      
2:    kernel ParamFct    NA    NA       2 <NoDefault[3]>      
3: shrinking ParamLgl    NA    NA       2           TRUE      

Here the kernel hyperparameter is a factor, so we simply pass in a vector corresponding to the levels we want to tune over. The shrinking hyperparameter is a logical, there are only two possible values this could take so we do not need to pass anything to to_tune, it will automatically recognize this is a logical from learner$param_set and passes this detail to learner$param_set$search_space(). In fact, for factor parameters, we could also use to_tune without any arguments if we want to tune over all possible values:

lrn("classif.svm", kernel = to_tune(c("radial", "linear")))$param_set$search_space()
<ParamSet>
       id    class lower upper nlevels        default value
1: kernel ParamFct    NA    NA       2 <NoDefault[3]>      

Finally, we can only use to_tune to treat numeric parameters as factors if we only want to discretize them over a small subset of possible values, for example if we wanted to find the optimal number of trees in a random forest we might only consider three scenarios: 100, 200, or 400 trees:

lrn("classif.ranger", num.trees = to_tune(c(100, 200, 400)))$param_set$search_space()
<ParamSet>
          id    class lower upper nlevels        default value
1: num.trees ParamFct    NA    NA       3 <NoDefault[3]>      
Trafo is set.

Before we look at tuning over vectors, we must first learn how to create parameter sets from scratch.

Warning

Treating an integer as a factor for tuning results in “unordered” hyperparameters. Therefore algorithms that make use of ordering information will perform worse when ordering is ignored. For these algorithms, it would make more sense to define a ParamDbl or ParamInt (Section 4.4.2) with a more fitting transformation (Section 4.4.3).

4.4.2 Defining Search Spaces with ps

As we have seen, to_tune is a helper function that creates a parameter set that will go on to be used by ti/tune/auto_tune during the tuning process. However, there will be use-cases where you will need to create a parameter set manually using paradox::ps(). This function takes named arguments of class Param, which can be created using the sugar functions in Table 4.3.

Table 4.3: Domain Constructors and their resulting Param.
Constructor Description Underlying Class
p_dbl Real valued parameter (“double”) ParamDbl
p_int Integer parameter ParamInt
p_fct Discrete valued parameter (“factor”) ParamFct
p_lgl Logical / Boolean parameter ParamLgl
p_uty Untyped parameter ParamUty

As a simple example, let’s look at how to create a search space to tune cost and gamma again:

search_space = ps(
  cost  = p_dbl(lower = 1e-1, upper = 1e5),
  kernel = p_fct(c("radial", "linear")),
  shrinking = p_lgl()
)

This search space would then be passed to the search_space argument in auto_tuner:

ti(task_sonar, lrn("classif.svm", type = "C-classification"), rsmp3,
  msr_ce, trm("none"), search_space = search_space)
<TuningInstanceSingleCrit>
* State:  Not optimized
* Objective: <ObjectiveTuning:classif.svm_on_sonar>
* Search Space:
          id    class lower upper nlevels
1:      cost ParamDbl   0.1 1e+05     Inf
2:    kernel ParamFct    NA    NA       2
3: shrinking ParamLgl    NA    NA       2
* Terminator: <TerminatorNone>
Warning

When manually creating search spaces, make sure all numeric hyperparameters in your search space are bounded, e.g., if you are trying to tune a hyperparameter that could take any in value \((-\infty, \infty)\) then the tuning process will throw an error if you do not pass lower and upper limits to p_dbl or p_int. You can use $is_bounded on the constructed ParamSet if you are unsure:

ps(cost = p_dbl(lower = 0.1, upper = 1))$is_bounded
[1] TRUE
ps(cost = p_dbl(lower = 0.1, upper = Inf))$is_bounded
[1] FALSE

4.4.3 Transformations and tuning over vectors

This section covers advanced ML or technical details that can be skipped.

In Section 4.1.5 we saw how to quickly apply log transformations with to_tune. As you now know, to_tune is just a wrapper that creates ParamSet objects, so let’s look at what is actually taking place when we set logscale = TRUE:

lrn("classif.svm", cost = to_tune(1e-5, 1e5, logscale = TRUE))$param_set$search_space()
<ParamSet>
     id    class     lower    upper nlevels        default value
1: cost ParamDbl -11.51293 11.51293     Inf <NoDefault[3]>      
Trafo is set.

Notice that now the lower and upper fields correspond to the transformed bounds, i.e. \([log(1e-5), log(1e5)]\). To manually create the same transformation, we can pass the transformation to the trafo argument in p_dbl:

ss = ps(cost = p_dbl(1e-5, 1e5, trafo = function(x) exp(x)))
ss
<ParamSet>
     id    class lower upper nlevels        default value
1: cost ParamDbl 1e-05 1e+05     Inf <NoDefault[3]>      
Trafo is set.

In this case we do not see the transformation in the lower and upper fields, however we can confirm it is correctly set by making use of the $trafo() method, which takes a named list and applies the specified transformations

ss$trafo(list(cost = 1))
$cost
[1] 2.718282

Where transformations become the most powerful is in the ability to pass arbitrary functions that can act on single parameters of the entire parameter set. As an example, consider a simple transformation to add ‘2’ to our range:

ss = ps(cost = p_dbl(0, 3, trafo = function(x) x + 2))
ss$trafo(list(cost = 1))
$cost
[1] 3

Simple transformations such as this can even be added directly to a learner by passing a Param object to to_tune:

lrn("classif.svm", cost = to_tune(p_dbl(0, 3, trafo = function(x) x + 2)))
<LearnerClassifSVM:classif.svm>
* Model: -
* Parameters: cost=<ObjectTuneToken>
* Packages: mlr3, mlr3learners, e1071
* Predict Types:  [response], prob
* Feature Types: logical, integer, numeric
* Properties: multiclass, twoclass

More complex transformations that require multiple arguments require the .extra_trafo argument in ps:

ss = ps(
  cost = p_dbl(-1, 1, trafo = function(x) exp(x)),
  kernel = p_fct(c("polynomial", "radial")),
  .extra_trafo = function(x, param_set) {
    if (x$kernel == "polynomial") {
      x$cost = x$cost + 2
    }
    x
  }
)
ss$trafo(list(cost = 1, kernel = "radial"))
$cost
[1] 2.718282

$kernel
[1] "radial"
ss$trafo(list(cost = 1, kernel = "polynomial"))
$cost
[1] 4.718282

$kernel
[1] "polynomial"

In this transformation we first exponentiate the value of cost and then add ‘2’ if the kernel is "polynomial". Note that .extra_trafo takes a function with parameters x and param_set. During tuning, x will be a list containing the configuration being tested, and param_set is the whole parameter set.

Vector transformations

This section covers advanced ML or technical details that can be skipped.

Any function can be passed to trafo and .extra_trafo, which enables tuning of ‘untyped’ parameters of class ParamUty that could be vectors, functions, or any non-atomic class. By example, consider the class.weights parameter of the SVM, which takes a named vector of class weights with one entry for each target class. To tune this parameter we could tune a scalar and then transform this to a vector:

search_space = ps(
  class.weights = p_dbl(lower = 0.1, upper = 0.9,
    trafo = function(x) c(M = x, R = 1 - x))
)

In other cases, we may need to tune two or more ‘pseudoparameters’ that do not exist in our learner’s parameter set, but are required to tune a vector parameter. For example, say we want to tune the architecture of a neural network, in which we need to decide the number of layers and the number of nodes in each layer, this is the case in the num_nodes hyperparameter in the lrn("surv.coxtime") learner. In this case the learner expects a vector where each element corresponds to the number of nodes and the length corresponds to the number of layers. We could then tune this as follows:

ss = ps(
  num_layers = p_int(lower = 1, upper = 20),
  num_nodes_per_layer = p_int(4, 64),
  .extra_trafo = function(x, param_set) {
    x$num_nodes = rep(x$num_nodes_per_layer, x$num_layers)
    x$num_layers = NULL
    x$num_nodes_per_layer = NULL
    x
  }
)

Here we are tuning the pseudo-parameter num_layers between 1 and 20, then tuning the pseudo-parameter num_nodes_per_layer between 4 and 64, then we are combining these into a vector called num_nodes (the real hyperparameter) and removing the pseudo-parameters.

ss$trafo(list(num_layers = 4, num_nodes_per_layer = 12))
$num_nodes
[1] 12 12 12 12

Once again, we could pass this straight to the learner using to_tune but this time passing the whole ParamSet object:

learner = lrn("surv.coxtime")
learner$param_set$set_values(num_nodes = to_tune(ss))
learner$param_set$search_space()
<ParamSet>
                    id    class lower upper nlevels        default value
1:          num_layers ParamInt     1    20      20 <NoDefault[3]>      
2: num_nodes_per_layer ParamInt     4    64      61 <NoDefault[3]>      
Trafo is set.

However note that in this case the transformation would only have access to the pseudo-hyperparameters and not the rest of the parameters in the learner’s parameter set.

4.4.4 Hyperparameter dependencies

This section covers advanced ML or technical details that can be skipped.

Hyperparameter dependencies occur when a hyperparameter should only be set if another hyperparameter has a particular value. For example, the degree parameter in SVM is only valid when kernel is “polynomial”. In the ps() function, we specify this using the depends argument, which takes a named argument of the form <param> == value or <param> %in% <vector>:

ps(
  kernel = p_fct(c("polynomial", "radial")),
  degree = p_int(1, 3, depends = (kernel == "polynomial")),
  gamma = p_dbl(1e-5, 1e5, depends = (kernel %in% c("polynomial", "radial")))
)
<ParamSet>
       id    class lower upper nlevels        default parents value
1: degree ParamInt 1e+00 3e+00       3 <NoDefault[3]>  kernel      
2:  gamma ParamDbl 1e-05 1e+05     Inf <NoDefault[3]>  kernel      
3: kernel ParamFct    NA    NA       2 <NoDefault[3]>              

Above we have said that degree should only be set if kernel is (==) "polynomial", and gamma should only be set if kernel is one of (%in%) "polynomial", "radial", or "sigmoid". In practice, some underlying implementations ignore unused parameters and others throw errors, either way this is problematic during tuning if, for example, we were wasting time trying to tune degree when the kernel was not polynomial. Hence setting the dependency tells the tuning process to tune degree iff kernel is "polynomial" and to ignore it otherwise.

Dependencies can also be passed straight into a learner using to_tune:

lrn("classif.svm",
  kernel = to_tune(c("polynomial", "radial")),
  degree = to_tune(p_int(1, 3, depends = (kernel == "polynomial")))
)$param_set$search_space()
<ParamSet>
       id    class lower upper nlevels        default       parents value
1: degree ParamInt     1     3       3 <NoDefault[3]> kernel,kernel      
2: kernel ParamFct    NA    NA       2 <NoDefault[3]>                    

4.4.5 Recommended Search Spaces with mlr3tuningspaces

This section covers advanced ML or technical details that can be skipped.

Selected search spaces can require a lot of background knowledge or expertise. The package mlr3tuningspaces tries to make HPO more accessible by providing implementations of published search spaces for many popular machine learning algorithms, the hope is that these search spaces are applicable to a wide range of datasets. The search spaces are stored in the dictionary mlr_tuning_spaces.

library(mlr3tuningspaces)
as.data.table(mlr_tuning_spaces)[1:3, .(key, label)]
                      key                             label
1: classif.glmnet.default   Classification GLM with Default
2:    classif.glmnet.rbv1 Classification GLM with RandomBot
3:    classif.glmnet.rbv2 Classification GLM with RandomBot

The tuning spaces are named according to the scheme {learner-id}.{tuning-space-id}. The default tuning spaces are published in Bischl et al. (2023), other tuning spaces are part of the random bot experiments rbv1 and rbv2 published in Kuehn et al. (2018) and Binder, Pfisterer, and Bischl (2020). The sugar function lts() is used to retrieve a TuningSpace.

rpart_hypers = lts("classif.rpart.default")
rpart_hypers
<TuningSpace:classif.rpart.default>: Classification Rpart with Default
          id lower upper levels logscale
1:  minsplit 2e+00 128.0            TRUE
2: minbucket 1e+00  64.0            TRUE
3:        cp 1e-04   0.1            TRUE

A tuning space can be passed to ti() or auto_tuner() as the search_space.

instance = ti(
  task = task_sonar,
  learner = lrn("classif.rpart"),
  resampling = rsmp("cv", folds = 3),
  measures = msr("classif.ce"),
  terminator = trm("evals", n_evals = 20),
  search_space = rpart_hypers
)

Alternatively, as loaded search spaces are just a collection of tune tokens, we could also pass these straight to a learner:

vals = rpart_hypers$values
vals
$minsplit
Tuning over:
range [2, 128] (log scale)


$minbucket
Tuning over:
range [1, 64] (log scale)


$cp
Tuning over:
range [1e-04, 0.1] (log scale)
learner = lrn("classif.rpart")
learner$param_set$set_values(.values = vals)
learner$param_set
<ParamSet>
                id    class lower upper nlevels        default
 1:             cp ParamDbl     0     1     Inf           0.01
 2:     keep_model ParamLgl    NA    NA       2          FALSE
 3:     maxcompete ParamInt     0   Inf     Inf              4
 4:       maxdepth ParamInt     1    30      30             30
 5:   maxsurrogate ParamInt     0   Inf     Inf              5
 6:      minbucket ParamInt     1   Inf     Inf <NoDefault[3]>
 7:       minsplit ParamInt     1   Inf     Inf             20
 8: surrogatestyle ParamInt     0     1       2              0
 9:   usesurrogate ParamInt     0     2       3              2
10:           xval ParamInt     0   Inf     Inf             10
1 variable not shown: [value]

We could also apply the default search spaces from Bischl et al. (2023) by passing the learner to lts():

lts(lrn("classif.rpart"))
<LearnerClassifRpart:classif.rpart>: Classification Tree
* Model: -
* Parameters: xval=0, minsplit=<RangeTuneToken>,
  minbucket=<RangeTuneToken>, cp=<RangeTuneToken>
* Packages: mlr3, rpart
* Predict Types:  [response], prob
* Feature Types: logical, integer, numeric, factor, ordered
* Properties: importance, missings, multiclass, selected_features,
  twoclass, weights

Finally, it is possible to overwrite a predefined tuning space in construction, for example changing the range of the maxdepth hyperparameter in a decision tree:

lts("classif.rpart.rbv2", maxdepth = to_tune(1, 20))
<TuningSpace:classif.rpart.rbv2>: Classification Rpart with RandomBot
          id lower upper levels logscale
1:        cp 1e-04     1            TRUE
2:  maxdepth 1e+00    20           FALSE
3: minbucket 1e+00   100           FALSE
4:  minsplit 1e+00   100           FALSE

4.5 Multi-Objective Tuning

So far we have considered optimizing a model with respect to one metric, but multi-criteria, or multi-objective optimization, is also possible. A simple example of multi-objective optimization might be optimizing a classifier to minimize false positive and false negative predictions. In another example, consider the single-objective problem of tuning a deep neural network to minimize classification error. The best performing model is likely to be quite complex, possibly with many layers that will take a long time to train, which would not be appropriate when you have limited resources. In this case we might want to simultaneously minimize the classification error and model complexity.

Multi-objective

By definition, optimization of multiple metrics means these will be in competition (otherwise we would only optimize one of them) and therefore in general no single configuration exists that optimizes all metrics. We therefore instead focus on the concept of Pareto optimality. One hyperparameter configuration is said to Pareto-dominate another if the resulting model is equal or better in all metrics and strictly better in at least one metric. For example say we are minimizing classification error, CE, and complexity, CP, for configurations A and B with CE of \(CE_A\) and \(CE_B\) respectively and CP of \(CP_A\) and \(CP_B\) respectively. Then, A pareto-dominates B if: 1) \(CE_A \geq CE_B\) and \(CP_A > CP_B\) or; 2) \(CE_A > CE_B\) and \(CP_A \geq CP_B\). All configurations that are not Pareto-dominated by any other configuration are called Pareto-efficient and the set of all these configurations is the Pareto set. The metric values corresponding to these Pareto set are referred to as the Pareto front.

Pareto SetPareto Front

The goal of multi-objective HPO is to approximate the Pareto front. We will now demonstrate multi-objective HPO by tuning a decision tree on the sonar dataset with respect to the classification error, as a measure of model performance, and the number of selected features, as a measure of model complexity (in a decision tree the number of selected features is straightforward to obtain by simply counting the number of unique splitting variables). Methodological details on multi-objective HPO can be found in Karl et al. (2022).

We will tune cp, minsplit, and maxdepth:

learner = lrn("classif.rpart",
  cp = to_tune(1e-04, 1e-1),
  minsplit = to_tune(2, 64),
  maxdepth = to_tune(1, 30)
)

measures = msrs(c("classif.ce", "selected_features"))

Note that as we tune with respect to multiple measures, the function ti creates a TuningInstanceMultiCrit instead of a TuningInstanceSingleCrit. We also have to set store_models = TRUE as this is required by the selected features measure.

instance = ti(
  task = task_sonar,
  learner = learner,
  resampling = rsmp("cv", folds = 5),
  measures = measures,
  terminator = trm("evals", n_evals = 30),
  store_models = TRUE
)
instance
<TuningInstanceMultiCrit>
* State:  Not optimized
* Objective: <ObjectiveTuning:classif.rpart_on_sonar>
* Search Space:
         id    class lower upper nlevels
1:       cp ParamDbl 1e-04   0.1     Inf
2: minsplit ParamInt 2e+00  64.0      63
3: maxdepth ParamInt 1e+00  30.0      30
* Terminator: <TerminatorEvals>

As before we will select and run a tuning algorithm, in this example we will use random search:

tuner = tnr("random_search", batch_size = 30)
tuner$optimize(instance)

Finally, we inspect the best-performing configurations, i.e., the Pareto set, and visualize the corresponding estimated Pareto front (Figure 13.7).

instance$archive$best()[, .(cp, minsplit, maxdepth, classif.ce, selected_features)]
            cp minsplit maxdepth classif.ce selected_features
1: 0.004160380       21        1  0.2938444               1.0
2: 0.005934033       13       14  0.2838560               6.8

Scatter plot with selected_features on x-axis and classif.ce on y-axis. Purple dots represent simulated tested configurations of selected_features vs. classif.ce and blue dots and a blue line along the bottom-left of the plot shows the Pareto front.

Figure 4.5: Pareto front of selected features and classification error. Purple dots represent tested configurations, each blue dot individually represents a Pareto-optimal configuration and all blue dots together represent the Pareto front.

Determining which configuration to use from the Pareto front is up to you. By definition there is no optimal configuration so this may depend on your use-case, for example if you would prefer lower complexity at the cost of higher error than you might prefer a configuration where selected_features = 1. You can select one configuration and pass it to a learner for training using $result_learner_param_vals, so if we want to select the first configuration we would run:

learner = lrn("classif.rpart")
learner$param_set$values = instance$result_learner_param_vals[[1]]
learner$param_set
<ParamSet>
                id    class lower upper nlevels        default      value
 1:             cp ParamDbl     0     1     Inf           0.01 0.00416038
 2:     keep_model ParamLgl    NA    NA       2          FALSE           
 3:     maxcompete ParamInt     0   Inf     Inf              4           
 4:       maxdepth ParamInt     1    30      30             30          1
 5:   maxsurrogate ParamInt     0   Inf     Inf              5           
 6:      minbucket ParamInt     1   Inf     Inf <NoDefault[3]>           
 7:       minsplit ParamInt     1   Inf     Inf             20         21
 8: surrogatestyle ParamInt     0     1       2              0           
 9:   usesurrogate ParamInt     0     2       3              2           
10:           xval ParamInt     0   Inf     Inf             10          0

As multi-objective tuning requires manual intervention to select a configuration, it is currently not possible to use auto_tuner. Furthermore, it can also be quite difficult to compare multiple models over multiple measures.

4.6 Multi-Fidelity Tuning via Hyperband

Increasingly large datasets and search spaces and costly to train models make hyperparameter optimization a time-consuming task. Recent HPO methods often also make use of evaluating a configuration at multiple fidelity levels. For example, a neural network can be trained for an increasing number of epochs, gradient boosting can be performed for an increasing number of boosting iterations and training data can always be subsampled to a smaller fraction of all available data. The general idea of multi-fidelity HPO is that the performance of a model obtained by using computationally cheap lower fidelity evaluation (few numbers of epochs or boosting iterations, only using a small sample of all available data for training) is predictive of the performance of the model obtained using computationally expensive full model evaluation and this concept can be leveraged to make HPO more efficient (e.g., only continuing to evaluate those configurations on higher fidelities that appear to be promising with respect to their performance). The fidelity parameter is part of the search space and controls the trade-off between the runtime and preciseness of the performance approximation.

A popular multi-fidelity HPO algorithm is Hyperband (Li et al. 2018). After having evaluated randomly sampled configurations on low fidelities, Hyperband iteratively allocates more resources to promising configurations and terminates low-performing ones. In the following example, we will optimize XGBoost and use the number of boosting iterations as the fidelity parameter. This means Hyperband will allocate increasingly more boosting iterations to well-performing hyperparameter configurations. Increasing the number of boosting iterations increases the time to train a model but generally also the performance. It is therefore a suitable fidelity parameter. However, as already mentioned, Hyperband is not limited to machine learning algorithms that are trained iteratively. In the second example, we will tune a support vector machine and use the size of the training data as the fidelity parameter. Some prior knowledge about pipelines (Chapter 6) is beneficial but not necessary to fully understand the examples. In the following, the terms fidelity and budget are often used interchangeably.

4.6.1 Hyperband Tuner

Hyperband (Li et al. 2018) builds upon the Successive Halving algorithm by Jamieson and Talwalkar (2016). Successive Halving is initialized with the number of starting configurations \(n\), the proportion of configurations discarded in each stage \(\eta\), and the minimum \(r{_{min}}\) and maximum \(r{_{max}}\) budget of a single evaluation. The algorithm starts by sampling \(n\) random configurations and allocating the minimum budget \(r{_{min}}\) to them. The configurations are evaluated and \(\frac{1}{\eta}\) of the worst-performing configurations are discarded. The remaining configurations are promoted to the next stage and evaluated on a larger budget. This continues until one or more configurations are evaluated on the maximum budget \(r{_{max}}\) and the best-performing configuration is selected. The total number of stages is calculated so that each stage consumes approximately the same overall budget. Successive Halving has the disadvantage that is not clear whether we should choose a large \(n\) and try many configurations on a small budget or choose a small \(n\) and train more configurations on the full budget.

Hyperband solves this problem by running Successive Halving with different numbers of stating configurations starting on different budget levels. The algorithm is initialized with the same parameters as Successive Halving except for \(n\). Each run of Successive Halving is called a bracket and starts with a different budget \(r{_{0}}\). A smaller starting budget means that more configurations can be evaluated. The most exploratory bracket is allocated the minimum budget \(r{_{min}}\). The next bracket increases the starting budget by a factor of \(\eta\). In each bracket, the starting budget increases further until the last bracket \(s = 0\) essentially performs a random search with the full budget \(r{_{max}}\). The number of brackets \(s{_{max}} + 1\) is calculated with \(s{_{max}} = {\log_\eta \frac{r{_{max}} }{r{_{min}}}}\). Under the condition that \(r{_{0}}\) increases by \(\eta\) with each bracket, \(r{_{min}}\) sometimes has to be adjusted slightly in order not to use more than \(r{_{max}}\) resources in the last bracket. The number of configurations in the base stages is calculated so that each bracket uses approximately the same amount of budget. Table 4.4 shows a full run of the Hyperband algorithm. The bracket \(s = 3\) is the most exploratory bracket and \(s = 0\) essentially performs a random search using the full budget.

Table 4.4: Hyperband schedule with the number of configurations \(n_{i}\) and resources \(r_{i}\) for each bracket \(s\) and stage \(i\), when \(\eta = 2\) , \(r{_{min}} = 1\) and \(r{_{max}} = 8\)
\(s = 3\) \(s = 2\) \(s = 1\) \(s = 0\)
\(i\) \(n_{i}\) \(r_{i}\) \(n_{i}\) \(r_{i}\) \(n_{i}\) \(r_{i}\) \(n_{i}\) \(r_{i}\)
0 8 1 6 2 4 4 4 8
1 4 2 3 4 2 8
2 2 4 1 8
3 1 8

4.6.2 Example XGBoost

In this practical example, we will optimize the hyperparameters of XGBoost on the spam dataset. We begin by constructing the learner.

library(mlr3hyperband)

learner = lrn("classif.xgboost")

As the next step we define the search space. The nrounds parameter controls the number of boosting iterations. We specify a range from 16 to 128 boosting iterations. This is used as \(r{_{min}}\) and \(r{_{max}}\) within Hyperband. We need to tag the parameter with "budget"to identify it as a fidelity parameter. For the other hyperparameters, we take the search space for XGBoost from Bischl et al. (2023). This search space usually work well for a wide range of datasets.

learner$param_set$set_values(
  nrounds           = to_tune(p_int(16, 128, tags = "budget")),
  eta               = to_tune(1e-4, 1, logscale = TRUE),
  max_depth         = to_tune(1, 20),
  colsample_bytree  = to_tune(1e-1, 1),
  colsample_bylevel = to_tune(1e-1, 1),
  lambda            = to_tune(1e-3, 1e3, logscale = TRUE),
  alpha             = to_tune(1e-3, 1e3, logscale = TRUE),
  subsample         = to_tune(1e-1, 1)
)

We proceed to construct the tuning instance. Note that we use trm("none") because Hyperband terminates itself after all brackets have been evaluated.

instance = ti(
  task = tsk("spam"),
  learner = learner,
  resampling = rsmp("holdout"),
  measures = msr("classif.ce"),
  terminator = trm("none")
)

We then construct the Hyperband tuner and specify eta = 2. In general, Hyperband can start all over from the beginning once the last bracket is evaluated. We control the number of Hyperband runs with the repetition argument. The setting repetition = Inf is useful when a terminator should stop the optimization, for example based on runtime.

tuner = tnr("hyperband", eta = 2, repetitions = 1)

Using eta = 2 and 16 to 128 boosting iterations results in the following schedule. This only prints a data table with the schedule and does not modify the tuner.

hyperband_schedule(r_min = 16, r_max = 128, eta = 2)
    bracket stage budget n
 1:       3     0     16 8
 2:       3     1     32 4
 3:       3     2     64 2
 4:       3     3    128 1
 5:       2     0     32 6
 6:       2     1     64 3
 7:       2     2    128 1
 8:       1     0     64 4
 9:       1     1    128 2
10:       0     0    128 4

We can now proceed with the tuning:

tuner$optimize(instance)

The result is the configuration with the best performance.

instance$result[, .(classif.ce, nrounds)]
   classif.ce nrounds
1: 0.05671447      64

Note that the archive resulting of a Hyperband run contains the additional columns bracket and stage:

as.data.table(instance$archive)[,
  .(bracket, stage, classif.ce, eta, max_depth, colsample_bytree)]
    bracket stage classif.ce         eta max_depth colsample_bytree
 1:       3     0 0.08865711 -6.30670834         9        0.8778884
 2:       3     0 0.09843546 -4.04120120        16        0.7876059
 3:       3     0 0.10625815 -7.16124801         3        0.5142202
 4:       3     0 0.09778357 -7.25284702        11        0.6225641
 5:       3     0 0.13298566 -2.83262403        12        0.1009884
---                                                                
31:       0     0 0.13102999 -7.44473775         3        0.1423377
32:       3     3 0.05801825 -0.06164547        15        0.1774943
33:       2     2 0.09256845 -7.38837441        14        0.2005979
34:       1     1 0.07692308 -7.52958477         7        0.6703289
35:       1     1 0.29791395 -0.56189030         3        0.9722843

4.6.3 Example Support Vector Machine

In this example, we will optimize the hyperparameters of a support vector machine on the sonar dataset. We begin by constructing the learner and setting type to "C-classification".

learner = lrn("classif.svm", id = "svm", type = "C-classification")

The mlr3pipelines package features a PipeOp for subsampling data. This will be helpful when using the size of the training data as a fidelity parameter.

po("subsample")
PipeOp: <subsample> (not trained)
values: <frac=0.6321, stratify=FALSE, replace=FALSE>
Input channels <name [train type, predict type]>:
  input [Task,Task]
Output channels <name [train type, predict type]>:
  output [Task,Task]

This pipeline operator controls the size of the training dataset with the frac parameter. We connect the po("subsample") with the learner and get a GraphLearner.

graph_learner = as_learner(
  po("subsample") %>>%
  learner
)

The graph learner subsamples and then fits a support vector machine on the data subset. The parameter set of the graph learner is a combination of the parameter sets of the pipeline operator and learner.

as.data.table(graph_learner$param_set)[, .(id, lower, upper, levels)]
                     id lower upper                             levels
 1:      subsample.frac     0   Inf                                   
 2:  subsample.stratify    NA    NA                         TRUE,FALSE
 3:   subsample.replace    NA    NA                         TRUE,FALSE
 4:       svm.cachesize  -Inf   Inf                                   
 5:   svm.class.weights    NA    NA                                   
 6:           svm.coef0  -Inf   Inf                                   
 7:            svm.cost     0   Inf                                   
 8:           svm.cross     0   Inf                                   
 9: svm.decision.values    NA    NA                         TRUE,FALSE
10:          svm.degree     1   Inf                                   
11:         svm.epsilon     0   Inf                                   
12:          svm.fitted    NA    NA                         TRUE,FALSE
13:           svm.gamma     0   Inf                                   
14:          svm.kernel    NA    NA   linear,polynomial,radial,sigmoid
15:              svm.nu  -Inf   Inf                                   
16:           svm.scale    NA    NA                                   
17:       svm.shrinking    NA    NA                         TRUE,FALSE
18:       svm.tolerance     0   Inf                                   
19:            svm.type    NA    NA C-classification,nu-classification

Next, we create the search space. We have to prefix the hyperparameters with the id of the pipeline operators, because this reflects the way how they are represented in the parameter set of the graph learner. The subsample.frac is the fidelity parameter that must be tagged with "budget" in the search space. In the following, the dataset size is increased from 3.7% to 100%. For the other hyperparameters, we take the search space for support vector machines from Binder, Pfisterer, and Bischl (2020). This search space usually work well for a wide range of datasets.

graph_learner$param_set$set_values(
  subsample.frac  = to_tune(p_dbl(3^-3, 1, tags = "budget")),
  svm.kernel      = to_tune(c("linear", "polynomial", "radial")),
  svm.cost        = to_tune(1e-4, 1e3, logscale = TRUE),
  svm.gamma       = to_tune(1e-4, 1e3, logscale = TRUE),
  svm.tolerance   = to_tune(1e-4, 2, logscale = TRUE),
  svm.degree      = to_tune(2, 5)
)

Support vector machines can often crash during training or take an extensive time to train given certain hyperparameters. We therefore set a timeout of 30 seconds and specify a fallback learner (Section 4.7.1) to handle these cases.

graph_learner$encapsulate = c(train = "evaluate", predict = "evaluate")
graph_learner$timeout = c(train = 30, predict = 30)
graph_learner$fallback = lrn("classif.featureless")

Let us create the tuning instance. Again, we use trm("none") because Hyperband controls the termination itself.

instance = ti(
  task = task_sonar,
  learner = graph_learner,
  resampling = rsmp("cv", folds = 3),
  measures = msr("classif.ce"),
  terminator = trm("none")
)

We create the tuner and set eta = 3.

tuner = tnr("hyperband", eta = 3)

Using eta = 3 and a lower bound of 3.7% for the dataset size results in the following Hyperband schedule.

hyperband_schedule(r_min = 3^-3, r_max = 1, eta = 3)
    bracket stage     budget  n
 1:       3     0 0.03703704 27
 2:       3     1 0.11111111  9
 3:       3     2 0.33333333  3
 4:       3     3 1.00000000  1
 5:       2     0 0.11111111 12
 6:       2     1 0.33333333  4
 7:       2     2 1.00000000  1
 8:       1     0 0.33333333  6
 9:       1     1 1.00000000  2
10:       0     0 1.00000000  4

We can now start the tuning.

tuner$optimize(instance)

We observe that the best model is a support vector machine with a polynomial kernel.

instance$result[, .(classif.ce, subsample.frac, svm.kernel)]
   classif.ce subsample.frac svm.kernel
1:  0.1824707              1 polynomial

The archive contains all evaluated configurations. We can proceed to further investigate the 8 configurations that were evaluated on the full dataset. The configuration with the best classification error on the full dataset was sampled in the second bracket. The classification error was estimated to be 30% on 33% of the dataset and decreased to 14% on the full dataset (see bright purple line in Figure 4.6).

Image showing the performance of 8 configurations evaluated on different training dataset sizes. The classification error decreases on larger training set sizes.

Figure 4.6: Optimization paths of the 8 configurations evaluated on the complete dataset.

4.7 Advanced Tuning

This section covers advanced ML or technical details that can be skipped.

This section is devoted to advanced tuning techniques, which are not necessary for a basic understanding of tuning but will make tuning more efficient and robust. We will first look at how to handle any errors that may occur during tuning, and then how to manage tuning experiments that may use up a lot of computer memory.

4.7.1 Error Handling

Error handling is discussed in detail in Section 9.2, however it is very important in the context of tuning so here we will just practically demonstrate how to make use of encapsulation and fallback learners and explain why they are essential during HPO.

Even in simple ML problems, there is a lot of potential for things to go wrong. For example when learners do not converge, run out of memory, or terminate with an error due to issues in the underlying data or bugs in the code. As a common example, learners can fail if there are factor levels present in the test data that were not in the training data, models fail in this case as there have been no weights/coefficients trained for these new factor levels:

task_bh = tsk("boston_housing")
cv10 = rsmp("cv", folds = 10)
msr_rmse = msr("regr.rmse")
rnd_srch = tnr("random_search")
learner = lrn("regr.lm", df = to_tune(1, 2))
instance = tune(
  tuner = rnd_srch,
  task = task_bh,
  learner = learner,
  resampling = cv10,
  measures = msr_rmse,
  term_evals = 10
)
Error in model.frame.default(Terms, newdata, na.action = na.action, xlev = object$xlevels): factor town has new levels Cohasset, Manchester

In the above example, we can see the tuning process breaks and we lose all information about the HPO process as the instance cannot be saved. This is even worse in nested resampling or benchmarking, when errors could cause us to lose all progress across multiple configurations or even learners and tasks.

Encapsulation (Section 9.2.1) allows errors to be isolated and handled, without disrupting the tuning process. We can tell a learner to encapsulate an error by setting the $encapsulate field as follows:

learner$encapsulate = c(train = "evaluate", predict = "evaluate")

Note by passing "evaluate" to both train and predict, we are telling the learner to setup encapsulation in both the training and predicting stages, however we could have only set it for one stage.

Another common issue that cannot be easily solved during HPO is learners not converging and the process running indefinitely. We can prevent this happening by setting the timeout field in a learner, which signals the learner to stop if it has been running for that much time, again this can be set for training and predicting individually:

learner$timeout = c(train = 30, predict = 30)

Now if either an error occurs, or the model timeout threshold is reached, then instead of breaking, the learner will simply not make predictions when errors are found. Unfortunately, if predictions are not made, then our HPO experimement will still fail as for any resampling iteration with errors, the result will be NA, and so are unable to aggregate results across resampling iterations. Therefore it is essential to also select a fallback learner (Section 9.2.2), which is essentiall a learner that will be fitted if the learner of interest fails. A common approach is to use a featureless baseline, classif/regr.featureless. Below we set regr.featureless, which always predicts the mean response, by passing this learner to the $fallback field.

learner$fallback = lrn("regr.featureless")

We can now run our experiment and see errors that occurred during tuning in the archive.

instance = tune(
  tuner = rnd_srch,
  task = task_bh,
  learner = learner,
  resampling = cv10,
  measures = msr_rmse,
  term_evals = 10
)

as.data.table(instance$archive)[, .(df, regr.rmse, errors)]
          df regr.rmse errors
 1: 1.662280   6.87099      7
 2: 1.322637   6.87099      7
 3: 1.497020   6.87099      7
 4: 1.531233   6.87099      7
 5: 1.467291   6.87099      7
 6: 1.373006   6.87099      7
 7: 1.156667   6.87099      7
 8: 1.990731   6.87099      7
 9: 1.561569   6.87099      7
10: 1.305218   6.87099      7
# Reading the error in the first resample result
instance$archive$resample_result(1)$errors
   iteration
1:         1
2:         3
3:         6
4:         7
5:         8
6:         9
7:        10
1 variable not shown: [msg]

The learner was tuned without breaking because the errors were encapsulated and logged before the fallback learners were used for fitting and predicting:

instance$result
        df learner_param_vals  x_domain regr.rmse
1: 1.66228          <list[1]> <list[1]>   6.87099

4.7.2 Memory Management

Running a large tuning experiment requires a lot of working memory, especially when using nested resampling. Most of the memory is consumed by the models since each resampling iteration creates one new model. The option store_models in the functions ti() and auto_tuner() allows us to enable the storage of the models. Storing the models is disabled by default and in most cases is not required.

The archive stores a ResampleResult for each evaluated hyperparameter configuration. The contained Prediction objects can take up a lot of memory, especially with large datasets and many resampling iterations. We can disable the storage of the resample results by setting store_benchmark_result = FALSE in the functions ti() and auto_tuner(). Note that without the resample results, it is no longer possible to score the configurations on another measure.

When we run nested resampling with many outer resampling iterations, additional memory can be saved if we set store_tuning_instance = FALSE in the auto_tuner() function. However, the functions extract_inner_tuning_results() and extract_inner_tuning_archives() would then no longer work.

The option store_models = TRUE sets store_benchmark_result and store_tuning_instance to TRUE because the models are stored in the benchmark results which in turn is part of the instance. This also means that store_benchmark_result = TRUE sets store_tuning_instance to TRUE.

Finally, we can set store_models = FALSE in the resample() or benchmark() functions to disable the storage of the auto tuners when running nested resampling. This way we can still access the aggregated performance (rr$aggregate()) but lose information about the inner resampling.

4.8 Conclusion

In this chapter, we learned how to optimize a model using tuning instances, about different tuners and terminators, search spaces and transformations, how to make use of automated methods for quicker implementation in larger experiments, and the importance of nested resampling. The most important functions and classes we learned about are in Table 4.5 alongside their R6 classes.

Table 4.5: Core S3 ‘sugar’ functions for model optimization in mlr3 with the underlying R6 class that are constructed when these functions are called (if applicable) and a summary of the purpose of the functions.
S3 function R6 Class Summary
tnr() Tuner Determines an optimisation algorithm
trm() Terminator Controls when to terminate the tuning algorithm
ti() TuningInstanceSingleCrit or TuningInstanceMultiCrit Stores tuning settings and save results
paradox::to_tune() paradox::TuneToken Sets which parameters in a learner to tune and over what search space
auto_tuner() AutoTuner Automates the tuning process
extract_inner_tuning_results() - Extracts inner results from nested resampling
extract_inner_tuning_archives() - Extracts inner archives from nested resampling

We have a few practical examples of tuning that may be useful for more specific use-cases, including:

4.9 Exercises

  1. Tune the mtry, sample.fraction, num.trees hyperparameters of a random forest model (regr.ranger) on the Motor Trend dataset (mtcars). Use a simple random search with 50 evaluations and select a suitable batch size. Evaluate with a 3-fold cross-validation and the root mean squared error.
  2. Evaluate the performance of the model created in Question 1 with nested resampling. Use a holdout validation for the inner resampling and a 3-fold cross-validation for the outer resampling. Print the unbiased performance estimate of the model.
  3. Tune and benchmark an XGBoost model against a logistic regression and determine which has the best Brier score. Use mlr3tuningspaces and nested resampling.