twalk: Run the t-walk MCMC Algorithm

View source: R/twalk_main.R

twalkR Documentation

Run the t-walk MCMC Algorithm

Description

This function implements the t-walk algorithm by Christen & Fox (2010), a general-purpose MCMC sampler that does not require manual tuning. The function can run multiple independent MCMC chains in parallel to accelerate execution and facilitate convergence diagnostics.

Usage

twalk(
  log_posterior,
  n_iter,
  x0,
  xp0,
  n_chains = 1,
  n_cores = NULL,
  show_progress = TRUE,
  ...
)

Arguments

log_posterior

A function that takes a parameter vector as its first argument and returns one numeric log posterior density. It may return '-Inf' outside the support, but must not return vectors, 'NA', 'NaN', or '+Inf'. Additional arguments can be passed to this function via '...'.

n_iter

The number of iterations to run for each chain.

x0

A numeric vector with the initial values for the first point ('x').

xp0

A numeric vector with the initial values for the second point (‘x’').

n_chains

The number of independent MCMC chains to run. Defaults to '1', which runs a single chain sequentially. If greater than 1, parallel mode is activated.

n_cores

The number of CPU cores to use in parallel mode. If 'NULL' (default), it will attempt to use all available cores minus one. Parallel random-number streams are initialized from R's current random state, so calling 'set.seed()' before 'twalk()' makes results reproducible.

show_progress

Logical; whether to display progress bars and status messages. Defaults to 'TRUE'.

...

Additional arguments to be passed to the 'log_posterior' function.

Value

A list containing:

samples

The primary t-walk trajectory, with 'n_iter' rows per chain.

companion_samples

The auxiliary trajectory maintained by the t-walk.

all_samples

Legacy concatenation of the primary and auxiliary trajectories. This is retained for compatibility and should not be treated as a single time-ordered MCMC chain.

acceptance_rate

The average Metropolis–Hastings acceptance rate across all chains. Accepted identity proposals are included, as required by the t-walk transition kernel.

move_rate

The proportion of iterations in which an accepted proposal actually changed at least one of the two t-walk points.

no_move_rate

The proportion of iterations containing an accepted identity proposal. It equals 'acceptance_rate - move_rate'.

n_iter

The number of iterations generated per chain.

n_chains

The number of independent chains.

total_iterations

The total number of primary samples generated ('n_iter * n_chains').

n_dim

The dimension of the parameter space.

individual_chains

If 'n_chains > 1', a list containing the raw results from each separate chain, useful for diagnostics like R-hat.

Examples

# Example 1: Sampling from a Bivariate Normal (sequential mode)
# The 'mvtnorm' package is required for this example
if (requireNamespace("mvtnorm", quietly = TRUE)) {
  log_post <- function(x) {
    mvtnorm::dmvnorm(x, mean = c(0, 0), sigma = matrix(c(1, 0.8, 0.8, 1), 2, 2), log = TRUE)
  }

  # Run with fewer iterations for a quick example
  # Set a seed for reproducibility
  set.seed(123)
  result_seq <- twalk(log_posterior = log_post, n_iter = 5000,
                          x0 = c(-1, 1), xp0 = c(1, -1))

  plot(result_seq$samples, pch = '.', main = "t-walk Samples (Sequential)")
}


# Example 2: The same problem in parallel (will run faster)
# Using 2 chains. n_iter is now per chain.
if (requireNamespace("mvtnorm", quietly = TRUE)) {
  set.seed(123)
  result_par <- twalk(log_posterior = log_post, n_iter = 2500,
                          x0 = c(-1, 1), xp0 = c(1, -1), n_chains = 2)

  plot(result_par$samples, pch = '.', main = "t-walk Samples (Parallel)")
}


Rtwalk documentation built on Aug. 29, 2026, 1:06 a.m.