8 Overfitting
Since overfitting is often caused by poor values of tuning parameters, we’ll focus on how to work with these values.
8.1 Requirements
You’ll need 2 packages (bestNormalize and tidymodels) for this chapter. You can install them via:
Let’s load the meta package and manage some between-package function conflicts.
8.2 Tuning Paramters
There are currently two main components in a model pipeline:
- a preprocessing method
- a supervised model fit.
In tidymodels, the type of object that can hold these two components is called a workflow. Each has arguments, many of which are for tuning parameters.
There are standardized arguments for most model parameters. For example, regularization in glmnet models and neural networks use the argument name penalty
even though the latter model would refer to this as weight decay.
tidymodels differentiates between two varieties of tuning parameters:
- main arguments are used by most engines for a model type.
- engine arguments represent more niche values specific to a few engines.
Let’s look at an example. In the embeddings chapter, the barley data had a high degree of correlation between the predictors. We discussed PCA, PLS, and other methods to deal with this (via recipe steps). We might try a neural network (say, using the brulee engine) for a model. The code to specify this pipeline would be:
pls_rec <-
recipe(barley ~ ., data = barley_train) %>%
step_zv(all_predictors()) %>%
step_orderNorm(all_numeric_predictors()) %>%
step_pls(all_numeric_predictors(),
outcome = "barley",
num_comp = 20) %>%
step_normalize(all_predictors())
nnet_spec <-
mlp(
hidden_units = 10,
activation = "relu",
penalty = 0.01,
epochs = 1000,
learn_rate = 0.1
) %>%
set_mode("regression") %>%
set_engine("brulee")
pls_nnet_wflow <- workflow(pls_rec, nnet_spec)
We’ve filled in specific values for each of these arguments, although we don’t know if these are best that we can do.
8.3 Marking Parameters for Optimization
To tune these parameters, we can give them a value of the function tune()
. This special function just returns an expression with the value “tune()
”. For example:
pls_rec <-
recipe(barley ~ ., data = barley_train) %>%
step_zv(all_predictors()) %>%
step_orderNorm(all_numeric_predictors()) %>%
step_pls(all_numeric_predictors(),
outcome = "barley",
# For demonstration, we'll use a label
num_comp = tune("pca comps")) %>%
step_normalize(all_predictors())
nnet_spec <-
mlp(
hidden_units = tune(),
activation = tune(),
penalty = tune(),
epochs = tune(),
learn_rate = tune()
) %>%
set_mode("regression") %>%
set_engine("brulee")
pls_nnet_wflow <- workflow(pls_rec, nnet_spec)
Optionally, we can give a label as an argument to the function:
str(tune("#PCA components"))
#> language tune("#PCA components")
This is useful when the pipeline has two arguments with the same name. For example, if you wanted to use splines for two predictors but allow them to have different degrees of freedom, the resulting set of parameters would not be unique since both of them would have the default label of deg_free
. In this case, one recipe step could use tune(“predictor 1 deg free”)
and another could be tune(“predictor 2 deg free”)
.
Engine arguments are set by set_engine()
. For example:
nnet_spec <-
mlp(
hidden_units = tune(),
activation = tune(),
penalty = tune(),
epochs = tune(),
learn_rate = tune()
) %>%
set_mode("regression") %>%
set_engine("brulee", stop_iter = 5, rate_schedule = tune())
pls_nnet_wflow <- workflow(pls_rec, nnet_spec)
8.4 Parameter Functions
Each tuning parameter has a corresponding function from the dials package containing information on the parameter type, parameter ranges (or possible values), and other data.
For example, the function for the penalty
argument is:
penalty()
#> Amount of Regularization (quantitative)
#> Transformer: log-10 [1e-100, Inf]
#> Range (transformed scale): [-10, 0]
This parameter has a default range from 10-10 to 1.0. It also has a corresponding transformation function (log base 10). This means that when values are created, they are uniformly distributed on the log scale. This is common for parameters that have values that span several orders of magnitude and cannot be negative.
We can change these defaults via arguments:
penalty(range = c(0, 1), trans = scales::transform_identity())
#> Amount of Regularization (quantitative)
#> Transformer: identity [-Inf, Inf]
#> Range (transformed scale): [0, 1]
In some cases, we can’t know the range a priori. Parameters like the number of possible PCA components or random forest’s \(m_{try}\) depend on the data dimensions. In the case of \(m_{try}\) , the default has an unknown in its range:
mtry()
#> # Randomly Selected Predictors (quantitative)
#> Range: [1, ?]
We would need to set this range to use the parameter.
In a few situations, the argument name to a recipe step or model function will use a dials function that has a different name than the argument. For example, there are a few different types of “degrees”. There is (real-valued) polynomial exponent degree:
degree()
#> Polynomial Degree (quantitative)
#> Range: [1, 3]
# Data type:
degree()$type
#> [1] "double"
but for the spline recipe steps we need an integer value:
# Data type:
spline_degree()$type
#> [1] "integer"
In some cases, tidymodels has methods for automatically changing the parameter function to be used, the range of values, and so on. We’ll see that in a minute.
There are also functions to manipulate individual parameters:
# Function list:
apropos("^value_")
#> [1] "value_inverse" "value_sample" "value_seq" "value_set"
#> [5] "value_transform" "value_validate"
value_seq(hidden_units(), n = 4)
#> [1] 1 4 7 10
8.5 Sets of Parameters
For our pipeline pls_nnet_wflow
, we can extract a parameter set that collects all of the parameters and their suggested information. There is a function to do this:
pls_nnet_param <- extract_parameter_set_dials(pls_nnet_wflow)
class(pls_nnet_param)
#> [1] "parameters" "tbl_df" "tbl" "data.frame"
names(pls_nnet_param)
#> [1] "name" "id" "source" "component" "component_id"
#> [6] "object"
pls_nnet_param
#> Collection of 7 parameters for tuning
#>
#> identifier type object
#> hidden_units hidden_units nparam[+]
#> penalty penalty nparam[+]
#> epochs epochs nparam[+]
#> activation activation dparam[+]
#> learn_rate learn_rate nparam[+]
#> rate_schedule rate_schedule dparam[+]
#> pca comps num_comp nparam[+]
The difference in the type
and identifier
columns only occurs when the tune()
value has a label (as with the final row).
The output ”nparam[+]”
indicates a numeric parameter, and the plus indicates that it is fully specified. If our pipeline had used \(m_{try}\), that value would show "nparam[?]”
. The rate schedule is a qualitative parameter and has a label of "dparam[+]"
(“d” for discrete).
Let’s look at the information for the learning rate parameter by viewing the parameter information set by tidymodels. It is different than the default:
pls_nnet_param %>%
filter(id == "learn_rate") %>%
pluck("object")
#> [[1]]
#> Learning Rate (quantitative)
#> Transformer: log-10 [1e-100, Inf]
#> Range (transformed scale): [-3, -0.5]
# The defaults:
learn_rate()
#> Learning Rate (quantitative)
#> Transformer: log-10 [1e-100, Inf]
#> Range (transformed scale): [-10, -1]
Why are they different? The main function has a wider range since it can be used by boosted trees, neural networks, UMAP, and other tools. The range is more narrow for this pipeline since we know that neural networks tend to work better with faster learning rates (so we set a different default).
Suppose we want to change the range to be even more narrow. We can use the update()
function to change defaults or to use a different dials parameter function:
You don’t always have to extract or modify a parameter set; this is an optional tool in case you want to change default values.
The parameter set is sometimes passed as an argument to grid creation functions or to iterative optimization functions that need to simulate/sample random candidates. For example, to create a random grid with 4 candidate values:
set.seed(220)
grid_random(pls_nnet_param, size = 4)
#> # A tibble: 4 × 7
#> hidden_units penalty epochs activation learn_rate rate_schedule `pca comps`
#> <int> <dbl> <int> <chr> <dbl> <chr> <int>
#> 1 9 2.080e-10 170 elu 0.002654 step 2
#> 2 3 1.883e- 3 112 tanh 0.002299 decay_time 4
#> 3 6 3.417e- 2 362 elu 0.1435 none 4
#> 4 6 1.593e- 9 444 tanh 0.001125 decay_time 4