Introducing bgms

Introduction

This example demonstrates how to use bgms for Bayesian analysis of binary or ordinal data with a special network model called a Markov Random Field (MRF) model. To learn more about the MRF model, check out Marsman & Haslbeck (2023), and to learn more about the Bayesian analysis of network models, check out Huth et al. (2023). A paper on the bgms software is coming soon.

We’ll examine real data on PTSD symptoms from 362 Chinese adults who survived the Wenchuan earthquake but tragically lost a child (McNally et al., 2015). The data comes from a 17-question survey where participants rated how much each symptom bothered them in the past month on a scale from “not at all” to “extremely.”

library(bgms)

Example 1 – EM

In this initial example, we’ll demonstrate how to quickly estimate a network using the bgm.em function. This method is similar to traditional network estimation approaches that seek a single optimal structure while minimizing model complexity. Unlike the complete Bayesian analysis of networks we’ll explore later, this method employs the EM algorithm to find an optimal structure (Marsman et al., 2022). The goal is to reduce the edge weights of missing connections to a value near zero, without shrinking the edge weights of existing connections.

Usage

bgm.em(
  x,
  precision = 0.975,
  convergence_criterion = sqrt(.Machine$double.eps),
  theta = 0.5,
  hierarchical = FALSE,
  indicator_alpha = 1,
  indicator_beta = 1,
  maximum_iterations = 1000,
  threshold_alpha = 1,
  threshold_beta = 1
)

Arguments

Output

The output is a list containing:

In summary, the list includes matrices for interactions, gamma, and thresholds, along with the prior inclusion probability theta when hierarchical = TRUE.

Analysis

fit <- bgm.em(x = Wenchuan)

Now let’s show the estimated edge weights and their inclusion probabilities in a plot.

par(mar = c(6, 5, 1, 1))
plot(x = fit$interactions[lower.tri(fit$interactions)], 
    y = fit$gamma[lower.tri(fit$gamma)], ylim = c(0, 1), 
    xlab = "", ylab = "", axes = FALSE, pch = 21, bg = "#bfbfbf", cex = 1.3)
abline(h = 0, lty = 2, col = "#bfbfbf")
abline(h = 1, lty = 2, col = "#bfbfbf")
abline(h = .5, lty = 2, col = "#bfbfbf")
mtext("Posterior mode edge weight", side = 1, line = 3, cex = 1.7)
mtext("(Local) posterior inclusion probability", side = 2, line = 3, cex = 1.7)
axis(1) 
axis(2, las = 1)

In the plot above, we observe that edge values close to zero are associated with low inclusion probabilities, while non-zero edge values have probabilities close to one. It is crucial to understand that these estimates are not actual posterior inclusion probabilities.

The goal of the EM algorithm is to identify the most likely network model, and to achieve this, it pushes the estimated inclusion probabilities towards extreme values (either 0 or 1). As a result, the estimated probabilities don’t represent the probability that the observed data come from a network with the edge in question (a posterior inclusion probability). Instead, they indicate the probability that the edge should be included in the most likely model.

When using EM edge selection, a median probability model is a network model that includes edges with a probability of 0.5 or higher. This approach selects the most likely network structure as the EM algorithm pushes inclusion probabilities towards extreme values. However, this could lead to a local optimum, implying that other, more optimal network structures may exist. Therefore, we refer to this as the local median probability model.

Here is some code to extract and plot the local median probability network:

library(qgraph) #For plotting the estimated network
  
posterior.inclusion <- fit$gamma[lower.tri(fit$gamma)]
tmp <- fit$interactions[lower.tri(fit$interactions)]
tmp[posterior.inclusion < 0.5] = 0
  
median.prob.model <- matrix(0, nrow = ncol(Wenchuan), ncol = ncol(Wenchuan))
median.prob.model[lower.tri(median.prob.model)] <- tmp
median.prob.model <- median.prob.model + t(median.prob.model)
  
rownames(median.prob.model) <- colnames(Wenchuan)
colnames(median.prob.model) <- colnames(Wenchuan)
  
qgraph(median.prob.model, 
      theme = "TeamFortress", 
      maximum = .5,
      fade = FALSE,
      color = c("#f0ae0e"), vsize = 10, repulsion = .9, 
      label.cex = 1.1, label.scale = "FALSE", 
      labels = colnames(Wenchuan))

This method offers a quick and computationally efficient way to start exploring our data. It estimates a single network structure that seems to best fit the data we have. However, it doesn’t show how uncertain this estimate is or whether other plausible structures could also fit the data. In contrast, Bayesian model averaging (BMA) considers all possible network structures and their probabilities.

BMA combines information from multiple network structures to create a more robust and accurate representation of variable relationships. By weighting each network’s contribution based on its posterior probability, BMA addresses model uncertainty, resulting in more nuanced and stable outcomes compared to the median probability model using EM variable selection.

Example 2 – BMA

A comprehensive Bayesian analysis of the data considers both the network structure and its corresponding parameters. As numerous structures could underlie our network, we employ simulation-based methods to investigate the posterior distribution of network structures and parameters (Marsman & Haslbeck, 2023). The bgm function performs this task, iteratively simulating values from the posterior distribution of network structures and parameters. Though this method takes more time compared to the previously discussed EM approach, it offers deeper insights.

Usage

bgm(
  x,
  iter = 10000,
  burnin = 1000,
  interaction_prior = c("UnitInfo", "Cauchy"),
  cauchy_scale = 2.5,
  edge_prior = c("Bernoulli", "Beta-Bernoulli"),
  inclusion_probability = 0.5,
  beta_bernoulli_alpha = 1,
  beta_bernoulli_beta = 1,  
  threshold_alpha = 1,
  threshold_beta = 1,
  adaptive = FALSE,
  na.action = c("listwise", "impute"),
  save = FALSE,
  display_progress = TRUE
)

Arguments

Output

If save = FALSE (the default), the result is a list containing the following matrices:

If save = TRUE, the result is a list containing:

Column averages of these matrices provide the model-averaged posterior means.

Analysis

fit <-  bgm(x = Wenchuan)

To save time, we ran the algorithm using the default number of iterations, which is 10,000. However, this may not be enough to fully explore the posterior distribution of the network structures and parameters. To obtain reliable and accurate estimates, we recommend increasing the number of iterations to 100,000 or more.

Let’s reproduce the plot from the previous example.

par(mar = c(6, 5, 1, 1))
plot(x = fit$interactions[lower.tri(fit$interactions)], 
     y = fit$gamma[lower.tri(fit$gamma)], ylim = c(0, 1), 
     xlab = "", ylab = "", axes = FALSE, pch = 21, bg = "gray", cex = 1.3)
abline(h = 0, lty = 2, col = "gray")
abline(h = 1, lty = 2, col = "gray")
abline(h = .5, lty = 2, col = "gray")
mtext("Posterior mean edge weight", side = 1, line = 3, cex = 1.7)
mtext("Posterior inclusion probability", side = 2, line = 3, cex = 1.7)
axis(1)
axis(2, las = 1)

This plot looks similar to Example 1 because edge values near zero have low inclusion probabilities, and edge values far from zero have high inclusion probabilities. However, it differs in two ways:

  1. The relationship between inclusion probability and edge weights takes a different shape for edge weights close to zero.
  2. There is more variety in the inclusion probabilities in Example 2 than in Example 1.

These differences happen because of how edge weights are handled. Both bgm.em and bgm use a diffuse prior density on edge weights for included edges, but they handle absent edges differently.

bgm.em uses a continuous density tightly wrapped around zero (with a very small variance or scale value). This causes the edge weights for absent edges to be close to zero, but not exactly zero.

On the other hand, bgm sets the edge weights to exactly zero if an edge is excluded. This accounts for the difference between the curved relationship between inclusion probability and estimated edge weight around zero in Example 1 and the V-shaped relationship in Example 2.

The greater variety in inclusion probabilities in Example 2 results from the bgm function not using an EM algorithm to find the most likely model, which would push inclusion probabilities towards the extremes. Instead, it employs a simulation method that averages over all plausible network structures to estimate the posterior inclusion probability, which represents the probability that a network containing the edge in question generated the observed data.

Median probability network

Using the posterior inclusion probabilities, we can also identify the median probability network. In this network, we include all edges with a posterior inclusion probability greater than 0.5. We can reuse the code from earlier to create the median probability model.

library(qgraph) #For plotting the estimated network
  
posterior.inclusion <- fit$gamma[lower.tri(fit$gamma)]
tmp <- fit$interactions[lower.tri(fit$interactions)]
tmp[posterior.inclusion < 0.5] = 0
  
median.prob.model <- matrix(0, nrow = ncol(Wenchuan), ncol = ncol(Wenchuan))
median.prob.model[lower.tri(median.prob.model)] <- tmp
median.prob.model <- median.prob.model + t(median.prob.model)
  
rownames(median.prob.model) <- colnames(Wenchuan)
colnames(median.prob.model) <- colnames(Wenchuan)
  
qgraph(median.prob.model, 
       theme = "TeamFortress", 
       maximum = .5,
       fade = FALSE,
       color = c("#f0ae0e"), vsize = 10, repulsion = .9, 
       label.cex = 1.1, label.scale = "FALSE", 
       labels = colnames(Wenchuan))

It’s worth noting that this model has more connections than the local median probability model estimated with EM. Although the structure estimated with EM was the most plausible, there were other network structures that could also be deemed plausible for the given data. As a result, their connections are included in the median probability model.

Inclusion Bayes factors

One of the benefits of using a fully Bayesian approach is that it allows us to calculate the inclusion Bayes factor Huth et al. (2023). The inclusion Bayes factor represents the relative evidence for including or excluding a connection between a pair of nodes in the network. An inclusion Bayes factor of 10 suggests that the observed data is ten times more likely to have come from a network that includes the relationship. Conversely, an inclusion Bayes factor of 1/10 implies that the observed data is ten times more likely to have come from a network that excludes the relationship. It’s important to note that inclusion Bayes factors can also reveal limited support for either hypothesis.

In the current version of bgms, it is assumed that the prior inclusion probabilities are equal to 0.5. To calculate the inclusion Bayes factors, we can simply convert the estimated posterior inclusion probabilities. For easier visualization, it is common to use the natural logarithm of the Bayes factor when plotting.

# Calculate the inclusion BFs
prior.odds = 1
posterior.inclusion = fit$gamma[lower.tri(fit$gamma)]
posterior.odds = posterior.inclusion / (1 - posterior.inclusion)
log.bayesfactor = log(posterior.odds / prior.odds)
#The next line is used to truncate the extreme values of the Bayes factor in the plot
log.bayesfactor[log.bayesfactor > 5] = 5

Lets plot the relation between the estimated edge weights and the inclusion Bayes factor.

par(mar = c(5, 5, 1, 1) + 0.1)
plot(fit$interactions[lower.tri(fit$interactions)], log.bayesfactor, pch = 21, bg = "#bfbfbf", 
    cex = 1.3, axes = FALSE, xlab = "", ylab = "", ylim = c(-5, 5.5),
    xlim = c(-0.5, 1.5))
axis(1)
axis(2, las = 1)
abline(h = log(1/10), lwd = 2, col = "#bfbfbf")
abline(h = log(10), lwd = 2, col = "#bfbfbf")

text(x = 1, y = log(1 / 10), labels = "Evidence for exclusion", pos = 1,
    cex = 1.7)
text(x = 1, y = log(10), labels = "Evidence for inclusion", pos = 3, cex = 1.7)
text(x = 1, y = 0, labels = "Weak evidence", cex = 1.7)
mtext("Log-inclusion Bayes factor", side = 2, line = 3, cex = 1.5, las = 0)
mtext("Posterior mean edge weights ", side = 1, line = 3.7, cex = 1.5, las = 0)

In this example, we use a cut-off value of 10 for the inclusion Bayes factors. Values greater than 10 suggest evidence for edge inclusion, values less than 1/10 indicate evidence for edge exclusion, and values between 1/10 and 10 are considered to represent weak evidence.

Analysis with raw output

For most purposes, the default output from bgm is sufficient, providing us with the posterior means of edge indicators and parameters. However, in some cases, we may want to use the raw samples from the joint posterior distribution. This could be to estimate the posterior distribution of a specific parameter, assess how many network structures fit the given data, or create Bayes factors for hypotheses involving multiple edges. We can obtain the raw samples by setting save = TRUE.

fit <-  bgm(x = Wenchuan, save = TRUE)

Posterior density of edge weight

We can employ the following code to use the posterior samples for plotting the posterior density of a single edge:

den = density(fit$interactions[,1], bw = "SJ")
i = which.min(abs(den$x - mean(fit$interactions[,1])))[1]
x = den$x[i]
f = den$y[i]
par(cex.main = 1.5, mar = c(5, 6, 1, 1) + 0.1, mgp = c(3.5, 1, 0), cex.lab = 1.5, font.lab = 2, cex.axis = 1.3, bty = "n", las = 1)
plot(den, axes = FALSE, xlab="", ylab="", main = "", frame.plot = FALSE)
axis(1)
axis(2)
par(las = 0)
mtext(text = "Edge weight", side = 1, line = 2.5, cex = 1.5)
mtext(text = "Posterior density", side = 2, line = 2.5, cex = 1.5)
# Add a point to indicate the posterior mean
points(x, f, pch = 21, bg = "grey", cex = 1.7)

The posterior distribution of the edge weight is averaged across all structures, which can lead to greater dispersion compared to estimating it for a specific model. This is because it takes into account the uncertainty of the network structures and the parameter estimates associated with these structures.

Note that the estimate is not very smooth. This is because we only used 10,000 samples to estimate the posterior distribution.

The posterior distribution of structures

We can also use the raw samples to count the number of unique structures bgm encountered in 10,000 iterations.

G = 2 * fit$gamma - 1
S = unique(G)
nrow(S)

There are clearly many different network structures that could fit the data. Let’s estimate their posterior probabilities.

Ps = vector(length = nrow(S))
for(r in 1:nrow(S)) {
  s = S[r, ]
  tmp = G %*% s
  Ps[r] = sum(tmp == ncol(G))
}
Ps = Ps / nrow(G) * 100
max(Ps)

The most plausible model accounts for less than 1 percent of the posterior probability. In conclusion, we have significant uncertainty about the network structure that generated the data.

In the analysis by Marsman & Haslbeck (2023), it is demonstrated that even when there is uncertainty about the network structure that generated the data, inclusion Bayes factors are highly robust. They can help identify substructures of the network in which we have strong confidence. However, to perform these analyses, we need to run bgm for many more iterations. In their analysis, Marsman & Haslbeck (2023) used 1,000,000 iterations. For further details, interested readers can refer to their analysis script here.

References

Huth, K., de Ron, J., Goudriaan, A. E., Luigjes, K., Mohammadi, R., van Holst, R. J., Wagenmakers, E.-J., & Marsman, M. (2023). Bayesian analysis of cross-sectional networks: A tutorial in R and JASP. Retrieved from Https://Osf.io/Preprints/Psyarxiv/Ub5tc/.
Marsman, M., & Haslbeck, J. M. B. (2023). Bayesian analysis of the ordinal Markov random field. Retrieved from Https://Osf.io/Preprints/Psyarxiv/Ukwrf/.
Marsman, M., Huth, K., Waldorp, L. J., & Ntzoufras, I. (2022). Objective Bayesian edge screening and structure selection for Ising networks. Psychometrika, 87(1), 47–82. https://doi.org/10.1007/s11336-022-09848-8
McNally, R. J., Robinaugh, D. J., Wu, G. W. Y., Wang, L., Deserno, M. K., & Borsboom, D. (2015). Mental disorders as causal systems: A network approach to posttraumatic stress disorder. Clinical Psychological Science, 5(6), 836–849. https://doi.org/10.1177/2167702614553230