The simstudy
package is a collection of functions that allows users to generate simulated data sets to explore modeling techniques or better understand data generating processes. The user defines the distributions of individual variables, specifies relationships between covariates and outcomes, and generates data based on these specifications. The final data sets can represent randomized control trials, repeated measure designs, cluster-randomized trials, or naturally observed data processes. Other complexities that can be added include survival data, correlated data, factorial study designs, step wedge designs, and missing data processes.
Simulation using simstudy
has two fundamental steps. The user (1) defines the data elements of a data set and (2) generates the data based on these definitions. Additional functionality exists to simulate observed or randomized treatment assignment/exposures, to create longitudinal/panel data, to create multi-level/hierarchical data, to create data sets with correlated variables based on a specified covariance structure, to merge data sets, to create data sets with missing data, and to create non-linear relationships with underlying spline curves.
The overarching philosophy of simstudy
is to create data generating processes that mimic the typical models used to fit those types of data. So, the parameterization of some of the data generating processes may not follow the standard parameterizations for the specific distributions. For example, in simstudy
gamma-distributed data are generated based on the specification of a mean \(\mu\) (or \(\log(\mu)\)) and a dispersion \(d\), rather than shape \(\alpha\) and rate \(\beta\) parameters that more typically characterize the gamma distribution. When we estimate the parameters, we are modeling \(\mu\) (or some function of \((\mu)\)), so we should explicitly recover the simstudy
parameters used to generate the model - illuminating the relationship between the underlying data generating processes and the models.
This introduction provides a brief overview to the basics of defining and generating data, including treatment or exposure variables. Subsequent sections in this vignette provide more details on these processes. For information on more elaborate data generating mechanisms, please refer to other vignettes in this package that provide more detailed descriptions.
The key to simulating data in simstudy
is the creation of a series of data definition tables that look like this:
varname | formula | variance | dist | link |
---|---|---|---|---|
age | 10 | 2 | normal | identity |
female | -2 + age * 0.1 | 0 | binary | logit |
visits | 1.5 - 0.2 * age + 0.5 * female | 0 | poisson | log |
These definition tables can be generated in two ways. One option is to use any external editor that allows the creation of .csv files, which can be read in with a call to defRead
. An alternative is to make repeated calls to the function defData
. This script builds a definition table internally:
<- defData(varname = "age", dist = "normal", formula = 10,
def variance = 2)
<- defData(def, varname = "female", dist = "binary",
def formula = "-2 + age * 0.1", link = "logit")
<- defData(def, varname = "visits", dist = "poisson",
def formula = "1.5 - 0.2 * age + 0.5 * female", link = "log")
The data definition table includes a row for each variable that is to be generated, and has the following fields: varname*
, formula
, variance
, dist
, and link
. varname
provides the name of the variable to be generated. formula
is either a value or string representing any valid R formula (which can include function calls) that in most cases defines the mean of the distribution. variance
is a value or string that specifies either the variance or other parameter that characterizes the distribution; the default is 0. dist
is defines the distribution of the variable to be generated; the default is normal. The link
is a function that defines the relationship of the formula with the mean value, and can either identity, log, or logit, depending on the distribution; the default is identity.
If using defData
to create the definition table, the first call to defData
without specifying a definition name (in this example the definition name is def) creates a new data.table with a single row. An additional row is added to the table def
each time the function defData
is called. Each of these calls is the definition of a new field in the data set that will be generated.
After the data set definitions have been created, a new data set with \(n\) observations can be created with a call to function genData
. In this example, 1,000 observations are generated using the data set definitions in def
, and then stored in the object dd
:
set.seed(87261)
<- genData(1000, def)
dd dd
## id age female visits
## 1: 1 9.78 0 0
## 2: 2 10.81 0 0
## 3: 3 8.86 0 1
## 4: 4 9.83 1 1
## 5: 5 10.58 0 0
## ---
## 996: 996 8.87 1 2
## 997: 997 10.27 0 0
## 998: 998 6.84 0 1
## 999: 999 9.28 0 2
## 1000: 1000 10.80 1 2
If no data definition is provided, a simple data set is created with just id’s.
genData(1000)
## id
## 1: 1
## 2: 2
## 3: 3
## 4: 4
## 5: 5
## ---
## 996: 996
## 997: 997
## 998: 998
## 999: 999
## 1000: 1000
In many instances, the data generation process will involve a treatment or exposure. While it is possible to generate a treatment or exposure variable directly using the data definition process, trtAssign
and trtObserve
offer the ability to generate more involved types of study designs. In particular, with trtAssign
, balanced and stratified designs are possible.
<- trtAssign(dd, n = 3, balanced = TRUE, strata = c("female"),
study1 grpName = "rx")
study1
## id age female visits rx
## 1: 1 9.78 0 0 3
## 2: 2 10.81 0 0 1
## 3: 3 8.86 0 1 3
## 4: 4 9.83 1 1 3
## 5: 5 10.58 0 0 3
## ---
## 996: 996 8.87 1 2 2
## 997: 997 10.27 0 0 3
## 998: 998 6.84 0 1 1
## 999: 999 9.28 0 2 1
## 1000: 1000 10.80 1 2 3
= .(female, rx)] study1[, .N, keyby
## female rx N
## 1: 0 1 249
## 2: 0 2 248
## 3: 0 3 248
## 4: 1 1 85
## 5: 1 2 85
## 6: 1 3 85
This section elaborates on the data definition process to provide more details on how to create data sets.
The data definition table for a new data set is constructed sequentially. As each new row or variable is added, the formula (and in some cases the variance) can refer back to a previously defined variable. The first row by necessity cannot refer to another variable, so the formula must be a specific value (i.e. not a string formula). Starting with the second row, the formula can either be a value or any valid R
equation with quotes and can include any variables previously defined.
In the definition we created above, the probability being female is a function of age, which was previously defined. Likewise, the number of visits is a function of both age and female. Since age is the first row in the table, we had to use a scalar to define the mean.
<- defData(varname = "age", dist = "normal", formula = 10,
def variance = 2)
<- defData(def, varname = "female", dist = "binary",
def formula = "-2 + age * 0.1", link = "logit")
<- defData(def, varname = "visits", dist = "poisson",
def formula = "1.5 - 0.2 * age + 0.5 * female", link = "log")
Formulas can include R
functions or user-defined functions. Here is an example with a user-defined function myinv
and the log
function from base R
:
<- function(x) {
myinv 1/x
}
<- defData(varname = "age", formula = 10, variance = 2,
def dist = "normal")
<- defData(def, varname = "loginvage", formula = "log(myinv(age))",
def variance = 0.1, dist = "normal")
genData(5, def)
## id age loginvage
## 1: 1 10.31 -2.58
## 2: 2 7.90 -1.94
## 3: 3 9.83 -1.93
## 4: 4 9.10 -2.42
## 5: 5 10.18 -2.21
Replication is an important aspect of data simulation - it is often very useful to generate data under different sets of assumptions. simstudy
facilitates this in at least two different ways. There is function updateDef
which allows row by row changes of a data definition table. In this case, we are changing the formula of loginvage in def so that it uses the function log10
instead of log
:
<- updateDef(def, changevar = "loginvage", newformula = "log10(myinv(age))")
def10 def10
## varname formula variance dist link
## 1: age 10 2 normal identity
## 2: loginvage log10(myinv(age)) 0.1 normal identity
genData(5, def10)
## id age loginvage
## 1: 1 9.82 -0.338
## 2: 2 10.97 -0.633
## 3: 3 11.79 -1.267
## 4: 4 9.74 -0.882
## 5: 5 10.11 -1.519
A more powerful feature of simstudy
that allows for dynamic definition tables is the external reference capability through the double-dot notation. Any variable reference in a formula that is preceded by “..” refers to an externally defined variable:
<- 3
age_effect
<- defData(varname = "age", formula = 10, variance = 2,
def dist = "normal")
<- defData(def, varname = "agemult", formula = "age * ..age_effect",
def dist = "nonrandom")
def
## varname formula variance dist link
## 1: age 10 2 normal identity
## 2: agemult age * ..age_effect 0 nonrandom identity
genData(2, def)
## id age agemult
## 1: 1 9.69 29.1
## 2: 2 9.63 28.9
But the real power of dynamic definition is seen in the context of iteration over multiple values:
<- c(0, 5, 10)
age_effects <- list()
list_of_data
for (i in seq_along(age_effects)) {
<- age_effects[i]
age_effect <- genData(2, def)
list_of_data[[i]]
}
list_of_data
## [[1]]
## id age agemult
## 1: 1 11.4 0
## 2: 2 10.7 0
##
## [[2]]
## id age agemult
## 1: 1 11.3 56.6
## 2: 2 11.2 56.1
##
## [[3]]
## id age agemult
## 1: 1 9.32 93.2
## 2: 2 10.62 106.2
The foundation of generating data is the assumptions we make about the distribution of each variable. simstudy
provides 15 types of distributions, which are listed in the following table:
name | formula | string/value | format | variance | identity | log | logit |
---|---|---|---|---|---|---|---|
beta | mean | both | - | dispersion | X | - | X |
binary | probability | both | - | - | X | - | X |
binomial | probability | both | - | # of trials | X | - | X |
categorical | probability | string | p_1;p_2;…;p_n | a;b;c | X | - | - |
exponential | mean | both | - | - | X | X | - |
gamma | mean | both | - | dispersion | X | X | - |
mixture | formula | string | x_1 | p_1 + … + x_n | p_n | - | X | - | - |
negBinomial | mean | both | - | dispersion | X | X | - |
nonrandom | formula | both | - | - | X | - | - |
normal | mean | both | - | variance | X | - | - |
noZeroPoisson | mean | both | - | - | X | X | - |
poisson | mean | both | - | - | X | X | - |
trtAssign | ratio | string | r_1;r_2;…;r_n | stratification | X | X | - |
uniform | range | string | from ; to | - | X | - | - |
uniformInt | range | string | from ; to | - | X | - | - |
A beta distribution is a continuous data distribution that takes on values between \(0\) and \(1\). The formula
specifies the mean \(\mu\) (with the ‘identity’ link) or the log-odds of the mean (with the ‘logit’ link). The scalar value in the ‘variance’ represents the dispersion value \(d\). The variance \(\sigma^2\) for a beta distributed variable will be \(\mu (1- \mu)/(1 + d)\). Typically, the beta distribution is specified using two shape parameters \(\alpha\) and \(\beta\), where \(\mu = \alpha/(\alpha + \beta)\) and \(\sigma^2 = \alpha\beta/[(\alpha + \beta)^2 (\alpha + \beta + 1)]\).
A binary distribution is a discrete data distribution that takes values \(0\) or \(1\). (It is more conventionally called a Bernoulli distribution, or is a binomial distribution with a single trial \(n=1\).) The formula
represents the probability (with the ‘identity’ link) or the log odds (with the ‘logit’ link) that the variable takes the value of 1. The mean of this distribution is \(p\), and variance \(\sigma^2\) is \(p(1-p)\).
A binomial distribution is a discrete data distribution that represents the count of the number of successes given a number of trials. The formula specifies the probability of success \(p\), and the variance field is used to specify the number of trials \(n\). Given a value of \(p\), the mean \(\mu\) of this distribution is \(n*p\), and the variance \(\sigma^2\) is \(np(1-p)\).
A categorical distribution is a discrete data distribution taking on values from \(1\) to \(K\), with each value representing a specific category, and there are \(K\) categories. The categories may or may not be ordered. For a categorical variable with \(k\) categories, the formula
is a string of probabilities that sum to 1, each separated by a semi-colon: \((p_1 ; p_2 ; ... ; p_k)\). \(p_1\) is the probability of the random variable falling in category \(1\), \(p_2\) is the probability of category \(2\), etc. The probabilities can be specified as functions of other variables previously defined. The helper function genCatFormula
is an easy way to create different probability strings. The link
options are identity or logit. The variance
field is optional an allows to provide categories other than the default 1...n
in the same format as formula
: “a;b;c”. Numeric variance Strings (e.g. “50;100;200”) will be converted to numeric when possible. All probabilities will be rounded to 1e12 decimal points to prevent possible rounding errors.
An exponential distribution is a continuous data distribution that takes on non-negative values. The formula
represents the mean \(\theta\) (with the ‘identity’ link) or log of the mean (with the ‘log’ link). The variance
argument does not apply to the exponential distribution. The variance \(\sigma^2\) is \(\theta^2\).
A gamma distribution is a continuous data distribution that takes on non-negative values. The formula
specifies the mean \(\mu\) (with the ‘identity’ link) or the log of the mean (with the ‘log’ link). The variance
field represents a dispersion value \(d\). The variance \(\sigma^2\) is is \(d \mu^2\).
The mixture distribution is a mixture of other predefined variables, which can be defined based on any of the other available distributions. The formula is a string structured with a sequence of variables \(x_i\) and probabilities \(p_i\): \(x_1 | p_1 + … + x_n | p_n\). All of the \(x_i\)’s are required to have been previously defined, and the probabilities must sum to \(1\) (i.e. \(\sum_1^n p_i = 1\)). The result of generating from a mixture is the value \(x_i\) with probability \(p_i\). The variance
and link
fields do not apply to the mixture distribution.
A negative binomial distribution is a discrete data distribution that represents the number of successes that occur in a sequence of Bernoulli trials before a specified number of failures occurs. It is often used to model count data more generally when a Poisson distribution is not considered appropriate; the variance of the negative binomial distribution is larger than the Poisson distribution. The formula
specifies the mean \(\mu\) or the log of the mean. The variance field represents a dispersion value \(d\). The variance \(\sigma^2\) will be \(\mu + d\mu^2\).
Deterministic data can be “generated” using the nonrandom distribution. The formula
explicitly represents the value of the variable to be generated, without any uncertainty. The variance
and link
fields do not apply to nonrandom data generation.
A normal or Gaussian distribution is a continuous data distribution that takes on values between \(-\infty\) and \(\infty\). The formula
represents the mean \(\mu\) and the variance
represents \(\sigma^2\). The link
field is not applied to the normal distribution.
The noZeroPoisson distribution is a discrete data distribution that takes on positive integers. This is a truncated poisson distribution that excludes \(0\). The formula
specifies the parameter \(\lambda\) (link is ‘identity’) or log() (link
is log). The variance
field does not apply to this distribution. The mean \(\mu\) of this distribution is \(\lambda/(1-e^{-\lambda})\) and the variance \(\sigma^2\) is \((\lambda + \lambda^2)/(1-e^{-\lambda}) - \lambda^2/(1-e^{-\lambda})^2\). We are not typically interested in modeling data drawn from this distribution (except in the case of a hurdle model), but it is useful to generate positive count data where it is not desirable to have any \(0\) values.
The poisson distribution is a discrete data distribution that takes on non-negative integers. The formula
specifies the mean \(\lambda\) (link is ‘identity’) or log of the mean (link
is log). The variance
field does not apply to this distribution. The variance \(\sigma^2\) is \(\lambda\) itself.
The trtAssign distribution is an implementation of the trtAssign
functionality in the context of the defData
workflow. Sometimes, it might be convenient to assign treatment or group membership while defining other variables. The formula
specifies the relative allocation to the different groups. For example three-arm randomization with equal allocation to each arm would be specified as “1;1;1”. The variance
field defines the stratification variables, and would be specified as “s_1;s_2” if s_1 and s_2 are the stratification variables. The link
field is used to indicate if the allocations should be perfectly balanced; if nothing is specified (and link defaults to identity), the allocation will be balanced; if link is specified to be different from identity, then the allocation will not be balanced.
A uniform distribution is a continuous data distribution that takes on values from \(a\) to \(b\), where \(b\) > \(a\), and they both lie anywhere on the real number line. The formula
is a string with the format “a;b”, where a and b are scalars or functions of previously defined variables. The variance
and link
arguments do not apply to the uniform distribution.
A uniform integer distribution is a discrete data distribution that takes on values from \(a\) to \(b\), where \(b\) > \(a\), and they both lie anywhere on the integer number line. The formula
is a string with the format “a;b”, where a and b are scalars or functions of previously defined variables. The variance
and link
arguments do not apply to the uniform integer distribution.
defRepeat
allows us to specify multiple versions of a variable based on a single set of distribution assumptions. The function will add nvar
variables to the data definition table, each of which will be specified with a single set of distribution assumptions. The names of the variables will be based on the prefix
argument and the distribution assumptions are specified as they are in the defData
function. Calls to defRepeat
can be integrated with calls to defData
.
<- defRepeat(nVars = 4, prefix = "g", formula = "1/3;1/3;1/3",
def variance = 0, dist = "categorical")
<- defData(def, varname = "a", formula = "1;1", dist = "trtAssign")
def <- defRepeat(def, 3, "b", formula = "5 + a", variance = 3,
def dist = "normal")
<- defData(def, "y", formula = "0.10", dist = "binary")
def
def
## varname formula variance dist link
## 1: g1 1/3;1/3;1/3 0 categorical identity
## 2: g2 1/3;1/3;1/3 0 categorical identity
## 3: g3 1/3;1/3;1/3 0 categorical identity
## 4: g4 1/3;1/3;1/3 0 categorical identity
## 5: a 1;1 0 trtAssign identity
## 6: b1 5 + a 3 normal identity
## 7: b2 5 + a 3 normal identity
## 8: b3 5 + a 3 normal identity
## 9: y 0.10 0 binary identity
Until this point, we have been generating new data sets, building them up from scratch. However, it is often necessary to generate the data in multiple stages so that we would need to add data as we go along. For example, we may have multi-level data with clusters that contain collections of individual observations. The data generation might begin with defining and generating cluster-level variables, followed by the definition and generation of the individual-level data; the individual-level data set would be adding to the cluster-level data set.
There are several important functions that facilitate the augmentation of data sets. defDataAdd
, defRepeatAdd
, and readDataAdd
are similar to their counterparts defData
, defRepeat
, and readData
, respectively; they create data definition tables that will be used by the function addColumns
. The formulas in these “add-ing” functions are permitted to refer to fields that exist in the data set to be augmented, so all variables need not be defined in the current definition able.
<- defData(varname = "x1", formula = 0, variance = 1,
d1 dist = "normal")
<- defData(d1, varname = "x2", formula = 0.5, dist = "binary")
d1
<- defRepeatAdd(nVars = 2, prefix = "q", formula = "5 + 3*rx",
d2 variance = 4, dist = "normal")
<- defDataAdd(d2, varname = "y", formula = "-2 + 0.5*x1 + 0.5*x2 + 1*rx",
d2 dist = "binary", link = "logit")
<- genData(5, d1)
dd <- trtAssign(dd, nTrt = 2, grpName = "rx")
dd dd
## id x1 x2 rx
## 1: 1 -1.3230 1 0
## 2: 2 -0.0494 0 1
## 3: 3 -0.4064 1 0
## 4: 4 -0.5562 1 0
## 5: 5 -0.0941 0 1
<- addColumns(d2, dd)
dd dd
## id x1 x2 rx q1 q2 y
## 1: 1 -1.3230 1 0 4.589 5.70 0
## 2: 2 -0.0494 0 1 9.829 11.74 1
## 3: 3 -0.4064 1 0 2.117 4.47 0
## 4: 4 -0.5562 1 0 0.798 3.24 0
## 5: 5 -0.0941 0 1 7.601 6.98 0
In certain situations, it might be useful to define a data distribution conditional on previously generated data in a way that is more complex than might be easily handled by a single formula. defCondition
creates a special table of definitions and the new variable is added to an existing data set by calling addCondition
. defCondition
specifies a condition argument that will be based on a variable that already exists in the data set. The new variable can take on any simstudy
distribution specified with the appropriate formula
, variance
, and link
arguments.
In this example, the slope of a regression line of \(y\) on \(x\) varies depending on the value of the predictor \(x\):
<- defData(varname = "x", formula = 0, variance = 9, dist = "normal")
d
<- defCondition(condition = "x <= -2", formula = "4 + 3*x",
dc variance = 2, dist = "normal")
<- defCondition(dc, condition = "x > -2 & x <= 2", formula = "0 + 1*x",
dc variance = 4, dist = "normal")
<- defCondition(dc, condition = "x > 2", formula = "-5 + 4*x",
dc variance = 3, dist = "normal")
<- genData(1000, d)
dd <- addCondition(dc, dd, newvar = "y") dd