The hierr R package is an extension of regularized regression (i.e. ridge regression) that enables the incorporation of external data that may be informative for the effects of predictors on an outcome of interest. Let (y) be an n-dimensional observed outcome vector, (X) be a set of p potential predictors observed on the n observations, and (Z) be a set of q external features available for the p predictors. Our model builds off the standard two-stage regression model,
but allows regularization of both the predictors and the external features, where beta is the vector of coefficients describing the association of each predictor with the outcome and alpha is the vector of coefficients describing the association of each external feature with the predictor coefficients, beta. As an example, assume that the outcome is continuous and that we want to apply a ridge penalty to the predictors and lasso penalty to the external features. We minimize the following objective function (ignoring intercept terms):
Note that our model allows for the predictor coefficients, beta, to shrink towards potentially informative values based on the matrix (Z). In the event the external data is not informative, we can shrink alpha towards zero, returning back to a standard regularized regression. To efficiently fit the model, we rewrite this convex optimization with the variable subsitution (gamma = beta - Z * alpha). The problem is then solved as a standard regularized regression in which we allow the penalty value and type (ridge / lasso) to be variable-specific:
This package extends the coordinate descent algorithm of Friedman et al. 2010 (used in the R package glmnet) to allow for this variable-specific generalization and to fit the model described above. Currently, we allow for continuous outcomes, but plan to extend to other outcomes (i.e. binomial, count).
library(devtools) # Master branch install_github("USCbiostats/hierr") # Or the development branch install_github("USCbiostats/hierr", ref = "development")
As an example of how you might use the hierr package, we have provided a small set of simulated external data variables (ext), predictors (x), and a continuous outcome variable (y). First, load the example data:
x <- hierr::x y <- hierr::y ext <- hierr::ext
To fit a linear hierarchical regularized regression model, use the main
hierr function. At a minimum, you should specify the predictor matrix
x, outcome variable
family (outcome distribution). The
external option allows you to incorporate external data in the
regularized regression model. If you do not include external data, a
standard regularized regression model will be fit. By default, a ridge
penalty is applied to the predictors and a lasso penalty is applied to
the external data.
hierr_model <- hierr(x = x, y = y, external = ext, family = "gaussian")
To modify the regularization terms and penalty path associated with the
predictors or external data, you can use the
This function allows you to configure the following regularization
quantileused to specify quantile)
As an example, we may want to apply a ridge penalty to both the x variables and external data variables. In addition, we may want to have 30 penalty values computed for the regularization path associated with both x and external. We modify our model fitting as follows.
myPenalty <- definePenalty(penalty_type = 0, penalty_type_ext = 0, num_penalty = 30, num_penalty_ext = 30) hierr_model <- hierr(x = x, y = y, external = ext, family = "gaussian", penalty = myPenalty)
In general, we need a method to determine the penalty values that
produce the optimal out-of-sample prediction. We provide a simple
two-dimensional grid search that uses k-fold cross-validation to
determine the optimal values for the penalties. The cross-validation
cvhierr is used as follows.
cv_hierr <- cvhierr(x = x, y = y, external = ext, family = "gaussian")
To visualize the results of the cross-validation we provide a contour
plot of the mean cross-validation error across the grid of penalties
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.