R code for adaptively constructing and searching an "information landscape" in order to find the particular experimental design that will maximize the information gained as a result of running the experiment. This code accompanies the recent paper:
Fast model-selection through adapting design of experiments maximizing
information gain
Stefano Balietti, Brennan Klein, and Christoph Riedl, 2018.
arXiv:1807.07024
This code samples model parameters and simulates likely datasets that you would observe if experiment participants were parameterized as such. After sampling datasets, each outcome is assigned a likelihood based on its frequency in the sample. Using these likelihoods, the information gain is calculated by taking the Kullback-Leibler Divergence between the likelihoods of the datasets generated under several competing models.
Start with the ps_gpucbpe_testbed.R file. It initializes the process that was used in "Fast Model-Selection through Adaptive Design of Experiments Maximizing Information Gain", recreating Figure 3c.
- ps_gpucbpe_testbed.R - start here!
- ps_gpucbpe_simulateDatasets.R - this script houses the Parameter-Sampled G PUCB-PE code
- ps_gpucbpe_models.R - the four models used in our model comparison, including three from El-Gamal & Palfrey (1995)
- ps_gpucbpe_process.R - Gaussian Process script, using functions from GPfit, adapted for optimizing experimental design
- ps_gpucbpe_histories.R - enumerates all possible game histories
- ps_gpucbpe_helper.R - plotting, statistics, and file naming
- ps_gpucbpe_calc_likelihoods.R - used for calculating the likelihoods of each dataset--assuming they have not been sampled
- matches.R - enumerates possible pairings in the experiment
If you use these methods and this code in your own research, please cite our paper:
Balietti, S., Klein, B. & Riedl, C. (2018). Fast model-selection through adapting design of experiments maximizing information gain. arXiv:1807.07024
Bibtex:
@article{balietti2018optimaldesign,
title={Fast model-selection through adapting design of experiments maximizing
information gain},
author={Balietti, Stefano; Klein, Brennan and Riedl, Christoph},
journal={arXiv preprint arXiv:1807.07024},
year={2018}
}
- El-Gamal, M. A., & Palfrey, T. R. (1996). Economical experiments: Bayesian
efficient experimental design. International Journal of Game Theory, 25(4),
495-517. doi: 10.1007/BF01803953.
- motivating work from which many of these ideas derive