Default prior distributions in dynamite
The default priors in dynamite
are chosen to be relatively uninformative (i.e., weakly informative) such that they provide computational stability by slightly regularizing the posterior. The motivation behind the default priors is thus similar to other popular Stan-based R packages such as brms
and rstanarm
.
Define \sigma_x=\max(1, \text{SD}(x)), where \text{SD}(x) is the standard deviation of the predictor variable x over groups and non-fixed time points (see section “Lagged responses and predictors” in the main vignette for more information: vignette("dynamite", package = "dynamite")
). Define also
\sigma_y = \begin{cases}
\max(1, \text{SD}(y)), &\text{if family is gaussian or student} \\
1, &\text{otherwise}
\end{cases},
where \text{SD}(y) is the standard deviation of the response variable as over groups and non-fixed time points.
Regression coefficients
We define the default prior for a time-invariant regression coefficients \beta as well for the first time-varying coefficient \delta_1 as zero-mean normal distribution with standard deviation of 2\sigma_y/\sigma_x. The two maximums used in definitions of \sigma_y and \sigma_x ensure that the prior standard deviation is at least 2.
Intercept
As the posterior correlations between the intercept \alpha and the regression coefficients \beta and \delta can be cause computational inefficiencies with gradient-based sampling algorithms of Stan, sampling of the intercept is performed indirectly via parameter a, so that the intercept \alpha (or \alpha_1 in case of time-varying intercept) is constructed as
\alpha = a - \bar X^\beta_1\beta - \bar X_1^\delta\delta_1,
where \bar X^\beta and \bar X^\delta are the means of the corresponding predictors at first (non-fixed) time point. The prior is then defined for a as
a \sim \text{N}(\bar y, 2\sigma_y),
where \bar y is mean of the response variable values at first time point after applying the link function (except in case of categorical and multinomial response where \bar y is set to zero).
Standard deviation parameters
The prior for the standard deviation parameter \tau of the random walk prior of the spline coefficients is half-normal with the standard deviation of 2\sigma_y/\sigma_x (with \sigma_x=1 in case of a time-varying intercept). The same prior distribution is also used for the the standard deviations of the random effects. Note that this prior can be too uninformative in some cases especially for the \tau parameters. A boundary (zero) avoiding \text{Gamma}(2, z) prior with suitable value of z can often work better for \tau when the default prior leads to divergences (alternatively, in some occasions switching between centered and noncentered parameterization in splines()
can help).
Correlation matrices
The correlation matrix of the random effects and latent factors uses a \text{LKJ}(1) prior with the Cholesky parameterization, see Stan documentation of lkj_corr_cholesky
for details. The default corresponds to uniform distribution over valid correlation matrices.
When defining latent factor term with nonzero_lambda = TRUE
, priors are set for \zeta and \kappa, with defaults \kappa \sim N(0, 1) and \kappa \sim Beta(2, 2). These are used to define standard deviation \sigma_\lambda of the loadings as \sigma_\lambda = \kappa \zeta, and \tau_\psi = (1 - \kappa)\zeta, the standard deviation of the random walk prior of \psi_t.
In case where nonzero_lambda = FALSE
, \tau_\psi is fixed to one, and prior is set directly on $_$ as N(0, 1). In both cases, prior on first \psi_1 \sim N(0, 1).
Family-specific parameters
For standard deviation parameter of gaussian and student’s t responses, we use exponential prior with a rate parameter \frac{1}{2\max(1,\sigma_y)}. The degrees-of-freedom parameter of the student’s t-distribution has a \text{Gamma}(2, 0.1) prior, whereas for other family specific parameters we set \phi \sim \text{Exponential}(1).
Defining priors of dynamite models
While dynamite()
can be used with the default prior choices, we recommend to check whether the defaults make sense for your particular problem and to codify them accordingly based on the domain knowledge.
Any univariate unbounded continuous distributions supported by Stan can be used as a prior for univariate model parameters (in case the parameter is constrained to be positive, the distribution is automatically truncated). Naturally, any univariate distribution bounded to the positive real line can be used as a prior for parameters constrained to be positive (e.g., a standard deviation parameter). See the Stan function reference at https://mc-stan.org/users/documentation/ for possible distributions. For custom priors, you should first get the default priors with the get_priors()
function, and then modify the priors
column of the obtained data frame before supplying it to the dynamite()
function:
f <- obs(y ~ -1 + z + varying(~ x + lag(y)) + random(~1 + z), "gaussian") +
random_spec(correlated = TRUE) + splines(df = 20)
p <- get_priors(f, data = gaussian_example, time = "time", group = "id")
p
#> parameter response prior type category
#> 1 sigma_nu_y_alpha y normal(0, 3.1) sigma_nu
#> 2 sigma_nu_y_z y normal(0, 3.1) sigma_nu
#> 3 alpha_y y normal(1.5, 3.1) alpha
#> 4 tau_alpha_y y normal(0, 3.1) tau_alpha
#> 5 beta_y_z y normal(0, 3.1) beta
#> 6 delta_y_x y normal(0, 3.1) delta
#> 7 delta_y_y_lag1 y normal(0, 1.8) delta
#> 8 tau_y_x y normal(0, 3.1) tau
#> 9 tau_y_y_lag1 y normal(0, 1.8) tau
#> 10 sigma_y y exponential(0.65) sigma
#> 11 L_nu lkj_corr_cholesky(1) L
p$prior[p$type == "sigma_nu"] <- "normal(0, 1)" # change prior for sigma_nu
p$prior[p$parameter == "sigma_y"] <- "student_t(3, 0, 2)" # prior for sigma_y
p
#> parameter response prior type category
#> 1 sigma_nu_y_alpha y normal(0, 1) sigma_nu
#> 2 sigma_nu_y_z y normal(0, 1) sigma_nu
#> 3 alpha_y y normal(1.5, 3.1) alpha
#> 4 tau_alpha_y y normal(0, 3.1) tau_alpha
#> 5 beta_y_z y normal(0, 3.1) beta
#> 6 delta_y_x y normal(0, 3.1) delta
#> 7 delta_y_y_lag1 y normal(0, 1.8) delta
#> 8 tau_y_x y normal(0, 3.1) tau
#> 9 tau_y_y_lag1 y normal(0, 1.8) tau
#> 10 sigma_y y student_t(3, 0, 2) sigma
#> 11 L_nu lkj_corr_cholesky(1) L
fit <- dynamite(
f,
data = gaussian_example,
time = "time",
group = "id",
priors = p
)