Title: | Bayesian Additive Regression Kernels |
---|---|
Description: | Bayesian Additive Regression Kernels (BARK) provides an implementation for non-parametric function estimation using Levy Random Field priors for functions that may be represented as a sum of additive multivariate kernels. Kernels are located at every data point as in Support Vector Machines, however, coefficients may be heavily shrunk to zero under the Cauchy process prior, or even, set to zero. The number of active features is controlled by priors on precision parameters within the kernels, permitting feature selection. For more details see Ouyang, Z (2008) "Bayesian Additive Regression Kernels", Duke University. PhD dissertation, Chapter 3 and Wolpert, R. L, Clyde, M.A, and Tu, C. (2011) "Stochastic Expansions with Continuous Dictionaries Levy Adaptive Regression Kernels, Annals of Statistics Vol (39) pages 1916-1962 <doi:10.1214/11-AOS889>. |
Authors: | Merlise Clyde [aut, cre, ths] (ORCID=0000-0002-3595-1872), Zhi Ouyang [aut], Robert Wolpert [ctb, ths] |
Maintainer: | Merlise Clyde <[email protected]> |
License: | GPL (>= 3) |
Version: | 1.0.6 |
Built: | 2024-11-05 05:01:00 UTC |
Source: | https://github.com/merliseclyde/bark |
Implementation of Bayesian Additive Regression Kernels with Feature Selection for Nonparametric Regression for Gaussian regression and classification for binary Probit models
_PACKAGE
BARK is a Bayesian sum-of-kernels model or because of the
Bayesian priors is a Bayesian Additive Regression Kernel model.
For numeric response , we have
,
where
.
For a binary response ,
, where
denotes the standard normal cdf (probit link).
In both cases,
is the sum of many Gaussian kernel functions.
The goal is to have very flexible inference for the unknown
function
.
bark uses an approximated Cauchy process as the prior distribution
for the unknown function
.
Feature selection can be achieved through the inference
on the scale parameters in the Gaussian kernels.
BARK accepts four different types of prior distributions through setting
values for
selection
(TRUE or FALSE), which allows scale parameters
for some variables to be set to zero, removing the variables from the
kernels selection = TRUE
; this enables either soft shrinkage or hard
shrinkage for the scale
parameters. The input common_lambdas
(TRUE or FALSE) specifies whether
a common scale parameter should be used for all predictors (TRUE) or
if FALSE allows the scale parameters to differ across all variables
in the kernel.
Ouyang, Zhi (2008) Bayesian Additive Regression Kernels. Duke University. PhD dissertation, Chapter 3.
Other bark functions:
bark()
,
bark-package-deprecated
,
sim_Friedman1()
,
sim_Friedman2()
,
sim_Friedman3()
,
sim_circle()
# Simulate regression example # Friedman 2 data set, 200 noisy training, 1000 noise free testing # Out of sample MSE in SVM (default RBF): 6500 (sd. 1600) # Out of sample MSE in BART (default): 5300 (sd. 1000) traindata <- sim_Friedman2(200, sd=125) testdata <- sim_Friedman2(1000, sd=0) fit.bark.d <- bark(y ~ ., data = data.frame(traindata), testdata = data.frame(testdata), classification = FALSE, selection = FALSE, common_lambdas = TRUE) boxplot(as.data.frame(fit.bark.d$theta.lambda)) mean((fit.bark.d$yhat.test.mean-testdata$y)^2) # Simulate classification example # Circle 5 with 2 signals and three noisy dimensions # Out of sample erorr rate in SVM (default RBF): 0.110 (sd. 0.02) # Out of sample error rate in BART (default): 0.065 (sd. 0.02) traindata <- sim_circle(200, dim=5) testdata <- sim_circle(1000, dim=5) fit.bark.se <- bark(y ~ ., data= data.frame(traindata), testdata= data.frame(testdata), classification=TRUE, selection=TRUE, common_lambdas = FALSE) boxplot(as.data.frame(fit.bark.se$theta.lambda)) mean((fit.bark.se$yhat.test.mean>0)!=testdata$y)
# Simulate regression example # Friedman 2 data set, 200 noisy training, 1000 noise free testing # Out of sample MSE in SVM (default RBF): 6500 (sd. 1600) # Out of sample MSE in BART (default): 5300 (sd. 1000) traindata <- sim_Friedman2(200, sd=125) testdata <- sim_Friedman2(1000, sd=0) fit.bark.d <- bark(y ~ ., data = data.frame(traindata), testdata = data.frame(testdata), classification = FALSE, selection = FALSE, common_lambdas = TRUE) boxplot(as.data.frame(fit.bark.d$theta.lambda)) mean((fit.bark.d$yhat.test.mean-testdata$y)^2) # Simulate classification example # Circle 5 with 2 signals and three noisy dimensions # Out of sample erorr rate in SVM (default RBF): 0.110 (sd. 0.02) # Out of sample error rate in BART (default): 0.065 (sd. 0.02) traindata <- sim_circle(200, dim=5) testdata <- sim_circle(1000, dim=5) fit.bark.se <- bark(y ~ ., data= data.frame(traindata), testdata= data.frame(testdata), classification=TRUE, selection=TRUE, common_lambdas = FALSE) boxplot(as.data.frame(fit.bark.se$theta.lambda)) mean((fit.bark.se$yhat.test.mean>0)!=testdata$y)
This data set contains six measurements on 100 genuine and 100 fradulent old Swiss banknotes
data(banknotes)
data(banknotes)
a dataframe with the following variables:
the status of the banknote: genuine or counterfeit
Length of bill (mm)
Width of left edge (mm)
Width of right edge (mm)
Bottom margin width (mm)
Top margin width (mm)
Length of diagonal (mm)
Flury, B. and Riedwyl, H. (1988). Multivariate Statistics: A practical approach. London: Chapman & Hall, Tables 1.1 and 1.2, pp. 5-8.
BARK is a Bayesian sum-of-kernels model.
For numeric response , we have
,
where
.
For a binary response ,
,
where
denotes the standard normal cdf (probit link).
In both cases, is the sum of many Gaussian kernel functions.
The goal is to have very flexible inference for the unknown
function
.
BARK uses an approximation to a Cauchy process as the prior distribution
for the unknown function
.
Feature selection can be achieved through the inference on the scale parameters in the Gaussian kernels. BARK accepts four different types of prior distributions, e, d, enabling either soft shrinkage or se, sd, enabling hard shrinkage for the scale parameters.
bark( formula, data, subset, na.action = na.omit, testdata = NULL, selection = TRUE, common_lambdas = TRUE, classification = FALSE, keepevery = 100, nburn = 100, nkeep = 100, printevery = 1000, keeptrain = FALSE, verbose = FALSE, fixed = list(), tune = list(lstep = 0.5, frequL = 0.2, dpow = 1, upow = 0, varphistep = 0.5, phistep = 1), theta = list() )
bark( formula, data, subset, na.action = na.omit, testdata = NULL, selection = TRUE, common_lambdas = TRUE, classification = FALSE, keepevery = 100, nburn = 100, nkeep = 100, printevery = 1000, keeptrain = FALSE, verbose = FALSE, fixed = list(), tune = list(lstep = 0.5, frequL = 0.2, dpow = 1, upow = 0, varphistep = 0.5, phistep = 1), theta = list() )
formula |
model formula for the model with all predictors, Y ~ X. The X variables will be centered and scaled as part of model fitting. |
data |
a data frame. Factors will be converted to numerical vectors based on the using 'model.matrix'. |
subset |
an optional vector specifying a subset of observations to be used in the fitting process. |
na.action |
a function which indicates what should happen when the data contain NAs. The default is "na.omit". |
testdata |
Dataframe with test data for out of sample prediction. |
selection |
Logical variable indicating whether variable
dependent kernel parameters |
common_lambdas |
Logical variable indicating whether
kernel parameters |
classification |
TRUE/FALSE logical variable, indicating a classification or regression problem. |
keepevery |
Every keepevery draw is kept to be returned to the user |
nburn |
Number of MCMC iterations (nburn*keepevery) to be treated as burn in. |
nkeep |
Number of MCMC iterations kept for the posterior inference. |
printevery |
As the MCMC runs, a message is printed every printevery draws. |
keeptrain |
Logical, whether to keep results for training samples. |
verbose |
Logical, whether to print out messages |
fixed |
A list of fixed hyperparameters, using the default values if not
specified. |
tune |
A list of tuning parameters, not expected to change. |
theta |
A list of the starting values for the parameter theta, use defaults if nothing is given. |
BARK is implemented using a Bayesian MCMC method. At each MCMC interaction, we produce a draw from the joint posterior distribution, i.e. a full configuration of regression coefficients, kernel locations and kernel parameters.
Thus, unlike a lot of other modelling methods in R,
we do not produce a single model object
from which fits and summaries may be extracted.
The output consists of values
(and
in the numeric case)
where * denotes a particular draw.
The
is either a row from the training data (x.train)
bark
returns an object of class 'bark' with a list, including:
call |
the matched call |
fixed |
Fixed hyperparameters |
tune |
Tuning parameters used |
theta.last |
The last set of parameters from the posterior draw |
theta.nvec |
A matrix with nrow(x.train) |
theta.varphi |
A matrix with nrow(x.train)
|
theta.beta |
A matrix with nrow(x.train) |
theta.lambda |
A matrix with ncol(x.train) rows and (nkeep) columns, recording the kernel scale parameters |
thea.phi |
The vector of length nkeep, recording the precision in regression Gaussian noise (1 for the classification case) |
yhat.train |
A matrix with nrow(x.train) rows and (nkeep) columns.
Each column corresponds to a draw |
yhat.test |
Same as yhat.train but now the x's are the rows of the test data; NULL if testdata are not provided |
yhat.train.mean |
train data fits = row mean of yhat.train |
yhat.test.mean |
test data fits = row mean of yhat.test |
Ouyang, Zhi (2008) Bayesian Additive Regression Kernels. Duke University. PhD dissertation, page 58.
Other bark functions:
bark-package
,
bark-package-deprecated
,
sim_Friedman1()
,
sim_Friedman2()
,
sim_Friedman3()
,
sim_circle()
##Simulated regression example # Friedman 2 data set, 200 noisy training, 1000 noise free testing # Out of sample MSE in SVM (default RBF): 6500 (sd. 1600) # Out of sample MSE in BART (default): 5300 (sd. 1000) traindata <- data.frame(sim_Friedman2(200, sd=125)) testdata <- data.frame(sim_Friedman2(1000, sd=0)) # example with a very small number of iterations to illustrate usage fit.bark.d <- bark(y ~ ., data=traindata, testdata= testdata, nburn=10, nkeep=10, keepevery=10, classification=FALSE, common_lambdas = FALSE, selection = FALSE) boxplot(data.frame(fit.bark.d$theta.lambda)) mean((fit.bark.d$yhat.test.mean-testdata$y)^2) ##Simulate classification example # Circle 5 with 2 signals and three noisy dimensions # Out of sample erorr rate in SVM (default RBF): 0.110 (sd. 0.02) # Out of sample error rate in BART (default): 0.065 (sd. 0.02) traindata <- sim_circle(200, dim=5) testdata <- sim_circle(1000, dim=5) fit.bark.se <- bark(y ~ ., data=data.frame(traindata), testdata= data.frame(testdata), classification=TRUE, nburn=100, nkeep=200, ) boxplot(as.data.frame(fit.bark.se$theta.lambda)) mean((fit.bark.se$yhat.test.mean>0)!=testdata$y)
##Simulated regression example # Friedman 2 data set, 200 noisy training, 1000 noise free testing # Out of sample MSE in SVM (default RBF): 6500 (sd. 1600) # Out of sample MSE in BART (default): 5300 (sd. 1000) traindata <- data.frame(sim_Friedman2(200, sd=125)) testdata <- data.frame(sim_Friedman2(1000, sd=0)) # example with a very small number of iterations to illustrate usage fit.bark.d <- bark(y ~ ., data=traindata, testdata= testdata, nburn=10, nkeep=10, keepevery=10, classification=FALSE, common_lambdas = FALSE, selection = FALSE) boxplot(data.frame(fit.bark.d$theta.lambda)) mean((fit.bark.d$yhat.test.mean-testdata$y)^2) ##Simulate classification example # Circle 5 with 2 signals and three noisy dimensions # Out of sample erorr rate in SVM (default RBF): 0.110 (sd. 0.02) # Out of sample error rate in BART (default): 0.065 (sd. 0.02) traindata <- sim_circle(200, dim=5) testdata <- sim_circle(1000, dim=5) fit.bark.se <- bark(y ~ ., data=data.frame(traindata), testdata= data.frame(testdata), classification=TRUE, nburn=100, nkeep=200, ) boxplot(as.data.frame(fit.bark.se$theta.lambda)) mean((fit.bark.se$yhat.test.mean>0)!=testdata$y)
The classification problem Circle is described in the BARK paper (2008).
Inputs are dim independent variables uniformly
distributed on the interval , only the first 2
out of these dim are actually signals.
Outputs are created according to the formula
sim_circle(n, dim = 5)
sim_circle(n, dim = 5)
n |
number of data points to generate |
dim |
number of dimension of the problem, no less than 2 |
Returns a list with components
x |
input values (independent variables) |
y |
0/1 output values (dependent variable) |
Ouyang, Zhi (2008) Bayesian Additive Regression Kernels. Duke University. PhD dissertation, Chapter 3.
Other bark simulation functions:
sim_Friedman1()
,
sim_Friedman2()
,
sim_Friedman3()
Other bark functions:
bark()
,
bark-package
,
bark-package-deprecated
,
sim_Friedman1()
,
sim_Friedman2()
,
sim_Friedman3()
sim_circle(n=100, dim=5)
sim_circle(n=100, dim=5)
The regression problem Friedman 1 as described in Friedman (1991) and
Breiman (1996). Inputs are 10 independent variables uniformly
distributed on the interval , only 5 out of these 10 are actually
used. Outputs are created according to
the formula
where e is .
sim_Friedman1(n, sd = 1)
sim_Friedman1(n, sd = 1)
n |
number of data points to create |
sd |
standard deviation of noise, with default value 1 |
Returns a list with components
x |
input values (independent variables) |
y |
output values (dependent variable) |
Breiman, Leo (1996) Bagging predictors. Machine Learning 24,
pages 123-140.
Friedman, Jerome H. (1991) Multivariate adaptive regression
splines. The Annals of Statistics 19 (1), pages 1-67.
Other bark simulation functions:
sim_Friedman2()
,
sim_Friedman3()
,
sim_circle()
Other bark functions:
bark()
,
bark-package
,
bark-package-deprecated
,
sim_Friedman2()
,
sim_Friedman3()
,
sim_circle()
sim_Friedman1(100, sd=1)
sim_Friedman1(100, sd=1)
The regression problem Friedman 2 as described in Friedman (1991) and Breiman (1996). Inputs are 4 independent variables uniformly distributed over the ranges
The outputs are created according to the formula
where e is .
sim_Friedman2(n, sd = 125)
sim_Friedman2(n, sd = 125)
n |
number of data points to create |
sd |
Standard deviation of noise. The default value of 125 gives a signal to noise ratio (i.e., the ratio of the standard deviations) of 3:1. Thus, the variance of the function itself (without noise) accounts for 90% of the total variance. |
Returns a list with components
x |
input values (independent variables) |
y |
output values (dependent variable) |
Breiman, Leo (1996) Bagging predictors. Machine Learning 24,
pages 123-140.
Friedman, Jerome H. (1991) Multivariate adaptive regression
splines. The Annals of Statistics 19 (1), pages 1-67.
Other bark simulation functions:
sim_Friedman1()
,
sim_Friedman3()
,
sim_circle()
Other bark functions:
bark()
,
bark-package
,
bark-package-deprecated
,
sim_Friedman1()
,
sim_Friedman3()
,
sim_circle()
sim_Friedman2(100, sd=125)
sim_Friedman2(100, sd=125)
The regression problem Friedman 3 as described in Friedman (1991) and Breiman (1996). Inputs are 4 independent variables uniformly distributed over the ranges
The outputs are created according to the formula
where e is .
sim_Friedman3(n, sd = 0.1)
sim_Friedman3(n, sd = 0.1)
n |
number of data points to create |
sd |
Standard deviation of noise. The default value of 125 gives a signal to noise ratio (i.e., the ratio of the standard deviations) of 3:1. Thus, the variance of the function itself (without noise) accounts for 90% of the total variance. |
Returns a list with components
x |
input values (independent variables) |
y |
output values (dependent variable) |
Breiman, Leo (1996) Bagging predictors. Machine Learning 24,
pages 123-140.
Friedman, Jerome H. (1991) Multivariate adaptive regression
splines. The Annals of Statistics 19 (1), pages 1-67.
Other bark simulation functions:
sim_Friedman1()
,
sim_Friedman2()
,
sim_circle()
Other bark functions:
bark()
,
bark-package
,
bark-package-deprecated
,
sim_Friedman1()
,
sim_Friedman2()
,
sim_circle()
sim_Friedman3(n=100, sd=0.1)
sim_Friedman3(n=100, sd=0.1)