Bayesian Workflow (Strack RRR Analysis Replication)#
from glob import glob from os.path import basename import arviz as az import bambi as bmb import numpy as np import pandas as pd
np.random.seed(12345) az.style.use('arviz-darkgrid') pd.options.mode.chained_assignment = None
In this Jupyter notebook, we do a Bayesian reanalysis of the data reported in the recent registered replication report (RRR) of a famous study by Strack, Martin & Stepper (1988). The original Strack et al. study tested a facial feedback hypothesis arguing that emotional responses are, in part, driven by facial expressions (rather than expressions always following from emotions). Strack and colleagues reported that participants rated cartoons as more funny when the participants held a pen in their teeth (unknowingly inducing a smile) than when they held a pen between their lips (unknowingly inducing a pout). The article has been cited over 1,400 times, and has been enormously influential in popularizing the view that affective experiences and outward expressions of affective experiences can both influence each other (instead of the relationship being a one-way street from experience to expression). In 2016, a Registered Replication Report led by Wagenmakers and colleagues attempted to replicate Study 1 from Strack, Martin, & Stepper (1988) in 17 independent experiments comprising over 2,500 participants. The RRR reported no evidence in support of the effect.
Here we use the Bambi model-building interface (itself just a wrapper for PyMC3) to quickly re-analyze the RRR data using a Bayesian approach. Because the emphasis here is on fitting models in Bambi, we spend very little time on quality control and data exploration; our goal is simply to show how one can replicate and extend the primary analysis reported in the RRR in a few lines of Bambi code.
We begin by importing all of the modules we’ll need, and enabling inline plotting with matplotlib.
Reading in the data#
The data for the RRR of Strack, Martin, & Stepper (henceforth SMS) is available as a set of CSV files from the project’s repository on the Open Science Framework. For the sake of completeness, we’ll show how to go from the raw CSV to the “long” data format that Bambi can use (and that’s required by most other mixed-effects modeling packages).
One slightly annoying thing about these 17 CSV files–each of which represents a different replication site–is that they don’t all contain exactly the same columns. Some labs added a column or two at the end (mostly for notes). To keep things simple, we’ll just truncate each dataset to only the first 22 columns. Because the variable names are structured in a bit of a confusing way, we’ll also just drop the first two rows in each file, and manually set the column names for all 22 variables. Once we’ve done that, we can simply concatenate all of the 17 datasets along the row axis to create one big dataset.
DL_PATH = 'data/facial_feedback/*csv' dfs =  columns = ['subject', 'cond_id', 'condition', 'correct_c1', 'correct_c2', 'correct_c3', 'correct_c4', 'correct_total', 'rating_t1', 'rating_t2', 'rating_c1', 'rating_c2', 'rating_c3', 'rating_c4', 'self_perf', 'comprehension', 'awareness', 'transcript', 'age', 'gender', 'student', 'occupation'] count = 0 for idx, study in enumerate(glob(DL_PATH)): data = pd.read_csv(study, encoding='latin1', skiprows=2, header=None, index_col=False).iloc[:, :22] data.columns = columns # Add study name data['study'] = idx # Some sites used the same subject id numbering schemes, so prepend with study to create unique ids. # Note that if we don't do this, Bambi would have no way of distinguishing two subjects who share # the same id, which would hose our results. data['uid'] = data['subject'].astype(float) + count dfs.append(data) data = pd.concat(dfs, axis=0).apply(pd.to_numeric, errors='coerce', axis=1)
Let’s see what the first few rows look like…
5 rows × 24 columns
Reshaping the data#
At this point we have our data in a pandas DataFrame with shape of (2612, 24). Unfortunately, we can’t use the data in this form. We’ll need to (a) conduct some basic quality control, and (b) “melt” the dataset–currently in so-called “wide” format, with each subject in a separate row–into long format, where each row is a single trial. Fortunately, we can do this easily in pandas:
# Keep only subjects who (i) respond appropriately on all trials, # (ii) understand the cartoons, and (iii) don't report any awareness # of the hypothesis or underlying theory. valid = data.query('correct_total==4 and comprehension==1 and awareness==0') long = pd.melt(valid, ['uid', 'condition', 'gender', 'age', 'study', 'self_perf'], ['rating_c1', 'rating_c2', 'rating_c3', 'rating_c4'], var_name='stimulus')
Notice that in the
melt() call above, we’re treating not only the unique subject ID (uid) as an identifying variable, but also gender, experimental condition, age, and study name. Since these are all between-subject variables, these columns are all completely redundant with
uid, and adding them does nothing to change the structure of our data. The point of explicitly listing them is just to keep them around in the dataset, so that we can easily add them to our models.
Fitting the model#
Now that we’re all done with our (minimal) preprocessing, it’s time to fit the model! This turns out to be a snap in Bambi. We’ll begin with a very naive (and, as we’ll see later, incorrect) model that includes only the following terms:
An overall (common) intercept.
The common effect of experimental condition (“smiling” by holding a pen in one’s teeth vs. “pouting” by holding a pen in one’s lips). This is the primary variable of interest in the study.
A group specific intercept for each of the 1,728 subjects in the ‘long’ dataset. (There were 2,576 subjects in the original dataset, but about 25% were excluded for various reasons, and we’re further excluding all subjects who lack complete data. As an exercise, you can try relaxing some of these criteria and re-fitting the models, though you’ll probably find that it makes no meaningful difference to the results.)
We’ll create a Bambi model, fit it, and store the results in a new object–which we can then interrogate in various ways.
# Initialize the model, passing in the dataset we want to use. model = bmb.Model(long, dropna=True) model.add('value ~ condition', group_specific='1|uid') # Set a custom prior on group specific factor variances—just for illustration group_specific_sd = bmb.Prior('HalfNormal', sigma=10) group_specific_prior = bmb.Prior('Normal', mu=0, sigma=group_specific_sd) model.set_priors(group_specific=group_specific_prior) # Fit the model, drawing 1,000 MCMC draws per chain results = model.fit(draws=1000)
Automatically removing 9/6940 rows from the dataset. Auto-assigning NUTS sampler... Initializing NUTS using jitter+adapt_diag... Multiprocess sampling (2 chains in 2 jobs) NUTS: [value_sigma, 1|uid_offset, 1|uid_sigma, condition, Intercept]
Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 14 seconds.
Notice that, in Bambi, the common and group specific effects are specified separately (for technical reasons we won’t get into here). The common effects can be specified using R-style formulas. The underlying parsing of the formula is done by patsy, and you can find out much more about the operators patsy supports by reading its excellent documentation. The group specific effects are specified as a list, where each term is a separate element. In this case, we only have one group specific factor (unique subject id) to include, so we can just leave it as a string.
When we run the above code, nothing much will happen for a while. The process of compiling the model in Theano (the numerical library that PyMC3–and hence, Bambi–relies on) can take a few minutes. Once sampling starts, we get a nice progress bar (courtesy of PyMC3) that keeps us updated on the sampler’s progress. Our model is fairly large, in that it includes nearly 1800 parameters (most of which are the group specific subject intercepts); even so, it should take only a few minutes or less to run on newer machines.
We can plot the prior distributions for all parameters with a call to the
And we can easily get the posterior distributions and traces for all parameters with
az.plot_trace(results, var_names=['Intercept', 'condition', 'value_sigma', '1|uid_sigma']);
If we want a numerical summary of the results, we just pass the results object to
az.summary(). By default, the summary display shows us the mean, standard deviation, and 94% highest density interval for the posterior. When we have multiple MCMC chains, the summary also includes some basic convergence diagnostic info like the effective sample sizes and the Gelman-Rubin R-hat statistic.
az.summary(results, var_names=['Intercept', 'condition', 'value_sigma', '1|uid_sigma'])
Expanding the model#
Looking at the parameter estimates produced by our model, it seems pretty clear that there’s no meaningful effect of condition. The posterior distribution is centered almost exactly on 0, with most of the probability mass on very small values. The 94% HDI spans from -0.143 to 0.083–in other words, the plausible effect of the experimental manipulation is, at best, to produce a change of < 0.2 on cartoon ratings made on a 10-point scale. For perspective, the variation between subjects is enormous
in comparison–the standard deviation
u_uid is around 1.2. We can also see that the model is behaving well, and the sampler seems to have converged nicely (the traces for all parameters look stationary).
Unfortunately, our first model has at least two pretty serious problems. First, it gives no consideration to between-study variation–we’re simply lumping all 1,728 subjects together, as if they came from the same study. A better model would properly account for study-level variation. We could model study as either a common or a group specific factor in this case–both choices are defensible, depending on whether we want to think of the 17 studies in this dataset as the only sites of interest, or as if they’re just 17 random sites drawn from some much larger population that have particular characteristics we want to account for. For present purposes, we’ll adopt the latter strategy (as an exercise, you can modify the the code below and re-run the model with study as a common factor). We’ll “keep it maximal” by adding both group specific study intercepts and group specific study slopes to the model. That is, we’ll assume that the subjects at each research site have a different baseline appreciation of the cartoons (some find the cartoons funnier than others), and that the effect of condition also varies across sites.
Second, our model also fails to explicitly model variation in cartoon ratings that should properly be attributed to the 4 stimuli. In principle, our estimate of the common effect of condition could change somewhat once we correctly account for stimulus variability (though in practice, the net effect is almost always to reduce effects, not increase them–so in this case, it’s very unlikely that adding group specific stimulus effects will produce a meaningful effect of condition). So we’ll deal with this by adding specific intercepts for the 4 stimuli. We’ll model the stimuli as group specific effect, rather than common, because it wouldn’t make sense to think of these particular cartoons as exhausting the universe of stimuli we care about (i.e., we wouldn’t really care about the facial-feedback effect if we knew that it only applied to 4 specific Far Side cartoons, and no other stimuli).
Lastly, just for fun, we can throw in some additional covariates, since they’re readily available in the dataset, and may be of interest even if they don’t directly inform the core hypothesis. Specifically, we’ll add common effects of gender and age to the model, which will let us estimate the degree to which participants’ ratings of the cartoons varies as a function of these background variables.
Once we’ve done all that, we end up with a model that’s in a good position to answer the question we care about–namely, whether the smiling/pouting manipulation has an effect on cartoon ratings that generalizes across the subjects, studies, and stimuli found in the RRR dataset.
model = bmb.Model(long, dropna=True) model.add('value ~ 1 + condition + age + gender', group_specific=['condition|study', '1|uid', 'condition|stimulus']) group_specific_sd = bmb.Prior('HalfNormal', sigma=10) group_specific_prior = bmb.Prior('Normal', mu=0, sigma=group_specific_sd) model.set_priors(group_specific=group_specific_prior) # Not we use 2000 samples for tuning and increase the taget_accept to 0.99. # The default values result in divergences. results = model.fit(draws=1000, tune=2000, target_accept=0.99)
Automatically removing 33/6940 rows from the dataset. Auto-assigning NUTS sampler... Initializing NUTS using jitter+adapt_diag... Multiprocess sampling (2 chains in 2 jobs) NUTS: [value_sigma, condition|stimulus_offset, condition|stimulus_sigma, 1|stimulus_offset, 1|stimulus_sigma, 1|uid_offset, 1|uid_sigma, condition|study_offset, condition|study_sigma, 1|study_offset, 1|study_sigma, gender, age, condition, Intercept]
Sampling 2 chains for 2_000 tune and 1_000 draw iterations (4_000 + 2_000 draws total) took 1390 seconds. There was 1 divergence after tuning. Increase `target_accept` or reparameterize. There was 1 divergence after tuning. Increase `target_accept` or reparameterize.
az.plot_trace(results, var_names=['Intercept', 'age', 'gender', 'condition', 'value_sigma', '1|study', '1|stimulus', 'condition|study', 'condition|stimulus', '1|study_sigma', '1|stimulus_sigma', 'condition|study_sigma', ], compact=True);
And the answer is…#
No. There’s still no discernible effect. Modeling the data using a mixed-effects model does highlight a number of other interesting features, however: * The stimulus-level variance
1|stimulus_sigma is quite large compared to the other factors. This is potentially problematic, because it suggests that a more conventional analysis that left individual stimulus effects out of the model could potentially run a high false positive rate. Note that this is a problem that affects both the RRR and
the original Strack study equally; the moral of the story is to deliberately sample large numbers of stimuli and explicitly model their influence. * Older people seem to rate cartoons as being (a little bit) funnier. * The variation across sites is surprisingly small–in terms of both the group specific intercepts (
1|study) and the group specific slopes (
condition|study). In other words, the constitution of the sample,
the gender of the experimenter, or any of the hundreds of others of between-site differences that one might conceivably have expected to matter, don’t really seem to make much of a difference to participants’ ratings of the cartoons.
%load_ext watermark %watermark -n -u -v -iv -w
The watermark extension is already loaded. To reload it, use: %reload_ext watermark bambi 0.2.0 pandas 1.1.5 numpy 1.19.4 arviz 0.10.0 last updated: Tue Dec 15 2020 CPython 3.8.5 IPython 7.18.1 watermark 2.0.2