What hyper-parameters are, and what to do with them; an illustration with ridge...

Ridge regression

Ridge regression is used when the data you are working with has a lot of explanatory variables,

or when there is a risk that a simple linear regression might overfit to the training data, because,

for example, your explanatory variables are collinear.

If you are training a linear model and then you notice that it generalizes very badly to new,

unseen data, it is very likely that the linear model you trained overfits the data.

In this case, ridge regression might prove useful. The way ridge regression works might seem

counter-intuititive; it boils down to fitting a worse model to the training data, but in return,

this worse model will generalize better to new data.

The closed form solution of the ordinary least squares estimator is defined as:

\[
\widehat{\beta} = (X'X)^{-1}X'Y
\]

where \(X\) is the design matrix (the matrix made up of the explanatory variables) and \(Y\) is the

dependent variable. For ridge regression, this closed form solution changes a little bit:

\[
\widehat{\beta} = (X'X + \lambda I_p)^{-1}X'Y
\]

where \(\lambda \in \mathbb{R}\) is an hyper-parameter and \(I_p\) is the identity matrix of dimension \(p\)

( \(p\) is the number of explanatory variables).

This formula above is the closed form solution to the following optimisation program:

\[
\sum_{i=1}^n \left(y_i – \sum_{j=1}^px_{ij}\beta_j\right)^2
\]

such that:

\[
\sum_{j=1}^p(\beta_j)^2<c
\]

for any strictly positive \(c\) .

library(tidyverse)
library(Ecdat)
library(glmnet)

index <- 1:nrow(Housing)

set.seed(12345)
train_index <- sample(index, round(0.90*nrow(Housing)), replace = FALSE)

test_index <- setdiff(index, train_index)

train_x <- Housing[train_index, ] %>% 
    select(-price)

train_y <- Housing[train_index, ] %>% 
    pull(price)

test_x <- Housing[test_index, ] %>% 
    select(-price)

test_y <- Housing[test_index, ] %>% 
    pull(price)

I do the train/test split this way, because glmnet() requires a design matrix as input, and not

a formula. Design matrices can be created using the model.matrix() function:

train_matrix <- model.matrix(train_y ~ ., data = train_x)

test_matrix <- model.matrix(test_y ~ ., data = test_x)

To run an unpenalized linear regression, we can set the penalty to 0:

model_lm_ridge <- glmnet(y = train_y, x = train_matrix, alpha = 0, lambda = 0)

The model above provides the same result as a linear regression. Let’s compare the coefficients between the two:

coef(model_lm_ridge)

## 13 x 1 sparse Matrix of class "dgCMatrix"
##                       s0
## (Intercept) -3247.030393
## (Intercept)     .       
## lotsize         3.520283
## bedrooms     1745.211187
## bathrms     14337.551325
## stories      6736.679470
## drivewayyes  5687.132236
## recroomyes   5701.831289
## fullbaseyes  5708.978557
## gashwyes    12508.524241
## aircoyes    12592.435621
## garagepl     4438.918373
## prefareayes  9085.172469

and now the coefficients of the linear regression (because I provide a design matrix, I have to use

lm.fit() instead of lm() which requires a formula, not a matrix.)

coef(lm.fit(x = train_matrix, y = train_y))

##  (Intercept)      lotsize     bedrooms      bathrms      stories 
## -3245.146665     3.520357  1744.983863 14336.336858  6737.000410 
##  drivewayyes   recroomyes  fullbaseyes     gashwyes     aircoyes 
##  5686.394123  5700.210775  5709.493884 12509.005265 12592.367268 
##     garagepl  prefareayes 
##  4439.029607  9085.409155

as you can see, the coefficients are the same. Let’s compute the RMSE for the unpenalized linear

regression:

preds_lm <- predict(model_lm_ridge, test_matrix)

rmse_lm <- sqrt(mean(preds_lm - test_y)^2)

The RMSE for the linear unpenalized regression is equal to 2077.4197343.

Let’s now run a ridge regression, with lambda equal to 100, and see if the RMSE is smaller:

model_ridge <- glmnet(y = train_y, x = train_matrix, alpha = 0, lambda = 100)

and let’s compute the RMSE again:

preds <- predict(model_ridge, test_matrix)

rmse <- sqrt(mean(preds - test_y)^2)

best_model <- cv.glmnet(train_matrix, train_y)
# lambda that minimises the MSE
best_model$lambda.min

## [1] 66.07936

According to cv.glmnet() the best value for lambda is 66.0793576. In the

next section, we will implement cross validation ourselves, in order to find the hyper-parameters

of a random forest.

Hope you enjoyed! If you found this blog post useful, you might want to follow

me on twitter for blog post updates and

buy me an espresso or paypal.me .