Automating approximate Bayesian computation by local linear regression
© Thornton; licensee BioMed Central Ltd. 2009
Received: 02 February 2009
Accepted: 07 July 2009
Published: 07 July 2009
In several biological contexts, parameter inference often relies on computationally-intensive techniques. "Approximate Bayesian Computation", or ABC, methods based on summary statistics have become increasingly popular. A particular flavor of ABC based on using a linear regression to approximate the posterior distribution of the parameters, conditional on the summary statistics, is computationally appealing, yet no standalone tool exists to automate the procedure. Here, I describe a program to implement the method.
The software package ABCreg implements the local linear-regression approach to ABC. The advantages are: 1. The code is standalone, and fully-documented. 2. The program will automatically process multiple data sets, and create unique output files for each (which may be processed immediately in R), facilitating the testing of inference procedures on simulated data, or the analysis of multiple data sets. 3. The program implements two different transformation methods for the regression step. 4. Analysis options are controlled on the command line by the user, and the program is designed to output warnings for cases where the regression fails. 5. The program does not depend on any particular simulation machinery (coalescent, forward-time, etc.), and therefore is a general tool for processing the results from any simulation. 6. The code is open-source, and modular.
Examples of applying the software to empirical data from Drosophila melanogaster, and testing the procedure on simulated data, are shown.
In practice, the ABCreg simplifies implementing ABC based on local-linear regression.
In many biological applications, parameter inference for models of interest from data is computationally challenging. Ideally, one would like to infer parameters using either maximum likelihood or Bayesian approaches which explicitly calculate the likelihood of the data given the parameters. While such likelihoods can be calculated for data from non-recombining regions [1, 2] and for data where all sites are independent [3, 4], full-likelihood methods are not currently feasible for many models of interest (complex demography with recombination, for example). Therefore, approximations are desirable.
Simulate a grid over the parameter space in order to calculate the likelihood of the observed summaries, given parameters [5, 6]. The maximum-likelihood estimate is the point on the grid that maximizes the likelihood of the observed summary statistics.
The maximum-likelihood algorithm can be modified to perform Bayesian inference by simulating parameters from prior distributions, calculating summary statistics, and accepting the parameters if they are "close enough" to the observed [7, 8]. The method runs until the desired number of acceptances are obtained, and can be extremely time-consuming. I refer to this approach as rejection sampling, and it has been applied in several contexts [9–11].
Decide ahead of time how many random draws to take from a prior distribution, then accept the fraction of draws which generate summary statistics closest to the data, according to some distance metric. This is the rejection-sampling approach of , and differs from the approach of [7–11] in that a finite number of simulations are performed from the prior instead of repeatedly simulating from the prior until a desired number of acceptances are recorded.
Take the parameters accepted from Method 3, and regress those acceptances onto the distance between the simulated and observed summary statistics .
The latter three methods are all forms of "Approximate Bayesian Computation" (ABC), a term which generally applies to inference problems using summary statistics instead of explicit calculations of likelihoods. The three Bayesian schemes described above are the simplest form of ABC, and the approach has been extended to use Markov Chain Monte Carlo techniques to explore the parameter space  and sequential Monte Carlo . Further developments include formalizing methods for choosing summary statistics  and methods for model selection . In this paper, I will use "regression ABC" to refer to Method 4, the regression approach of . The main appeal of regression ABC is speed, overcoming a major limitation of rejection-sampling, which is often too slow to feasibly evaluate the performance of the estimator (due to requiring high rejection rates in order to obtain reasonable estimates [8, 11]). In general, the regression ABC method has several appealing features, including simplicity of implementation, speed, and flexibility. The flexibility is a key issue, as it allows one to rapidly explore how many, and which, summary statistics to use, which is an important issue, as subtle choices can lead to surprising biases in estimation .
Currently, many tools are available for the rapid development and testing of summary-statistic based approaches to inference, including rapid coalescent simulations for both neutral models  and simple models of selection [9, 19, 20], software to calculate summary statistics from simulation output , and open-source statistical packages such as R. Currently, the only software package available to implement the regression algorithm of  is implemented in the R language, and is available from http://www.rubic.rdg.ac.uk/~mab/. The purpose of this paper is to describe a software package which automates the linear regression portion of regression ABC analyses in a fast and flexible way, with user-friendly features simplifying automation. The results from the current code have been validated against independent R implementations, and the "ABCreg" package is fully documented for use by non-programmers.
The software package is called ABCreg, and is distributed as source code from the author's web site (see below). The code compiles to generate a single binary, reg, which automates all of the regression computations. The code was written in the C++ programming language , and the linear algebra calculations for the regression are performed using the GNU Scientific Library (GSL, http://www.gnu.org/software/gsl). The C and C++ languages are ideal for this task due to the speed of the compiled programs (often an order of magnitude faster than R). Although the regression-ABC step is less computationally-demanding than simulating from the prior distribution, it does not necessarily follow that the relative speed of the simulations is the limiting step in an analysis. In practice, one may spend considerable time evaluating the utility of different sets of summary statistics, running the regression-ABC portion of the analysis multiple times on a set of simulated data. It is therefore desirable to optimize the speed of the regression-ABC step as well as the speed of the simulations.
- 1.Transformation of the parameters simulated from the prior distribution. Currently, the program implements both the natural-log transformation used in  and the transformation proposed by :
where min and max are the lower and upper bounds of the prior, respectively. The latter transformation assures that the posterior distribution is contained withing the bounds of the prior. The user may also opt to not transform the simulated values at all.
Normalisation of the observed summary statistics and summary statistics simulated from the prior
The rejection step based on accepting the closest δ of Euclidean distances between observed an simulated summary statistics. Here, δ specifies the tolerance for acceptance, and is the fraction of draws from the prior to accept, specified by the user on the command line.
The regression adjustment
Back-transformation of regression-adjusted parameter values and output to files. The program generates one output file per data set in the data file. File names are generated automatically, and the prefix of the file names is controlled by the user. The output files contain tab-delimited columns which are the regression-adjusted parameter values (i.e., the estimates of the posterior distribution), which are easily processed in R.
Use of the software requires two input files. The first file describes the data (either real or simulated), and contains a space-delimited list of the summary statistics. One can analyze multiple data sets by recording the summary statistics for each data set on a different line of the file. The second input file describes the results of simulating from the prior distribution on the model parameter(s). This "prior file" contains a space-delimited list of the parameters, and the corresponding summary statistics (in the same order as in the data file).
Additional features include a complete debugging mode, which helps identify cases where the linear regression may fail. In practice, the analysis of some data sets may return non-finite parameter values. Often, this is due to the predicted mean value of the regression being quite large, such that back-transformation (+/- the residuals from the regression) results in a value that cannot be represented on the machine. In debug mode, such cases immediately exit with an error. When not in debug mode, the program prints warnings to the screen.
In this section, I show results from applying the ABCreg software to the inference scheme of , who used rejection sampling (Method 2 above) to infer the parameters of a simple population bottleneck model from sequence data obtained from a European population sample of Drosophila melanogaster. This model has three parameters, t r , the time at which the population recovered from the bottleneck, d, the duration of the bottleneck, and f, the bottleneck severity. The parameters t r and d are scaled in units of 4N0 generations, where N0 is the effective population size at the present time, and f = N b /N0, the ratio of the bottlenecked size to the current size (0 <f ≤ N0). See  for more details of the model. The data consist of 105 X-linked, non-coding loci surveyed by  and another ten from . For each of these 115 non-coding fragments, sequence variation was surveyed in population samples from Zimbabwe, and the Netherlands. Thornton and Andolfatto used a two-step approach for the parameter inference. First, a relatively wide uniform prior was used in conjunction with a fairly liberal tolerance for acceptance. Then, the 1 st and 99 th quantiles of the resulting posterior distributions were used as the bounds on a new, uniform prior, and the acceptance criteria were made more strict. Three summary statistics were used: the variances across loci of nucleotide diversity (π, ), the number of haplotypes in the sample, and a summary of the site-frequency spectrum of mutations . The rejection sampling scheme took two weeks to run on a large computer cluster.
I repeated the analysis using the local regression approach using the same data and uniform priors on parameters (see Table one of ). The analysis was done assuming that ρ = 4N e r (the population recombination rate) is equal to 10θ (see  for details), and the value of θ (the population mutation rate, see , p. 92) at each locus was obtained by the method of  using data from a Zimbabwe population sample. C++ code was written using the GSL and the coalescent routines in libsequence to sample 5 × 106 draws from the prior distribution on the three parameters, and to record the resulting summary statistics. Simulating from the prior took 24 hours on four 2 gigahertz AMD Opteron processors. The tolerance was set such that 103 acceptances were recorded for the regression. The model has three parameters, and three summary statistics are used. Once the simulations from the prior distribution are complete, the entire ABC analysis was performed with one command:
reg -P 3 -S 3 -p prior -d data -b data -t 0.0002 -T,
where the arguments specify the number of parameters (-P), number of summary statistics (-S), names of files containing the prior (-p) and data (-d), the prefix of the output file names (-b), the tolerance (-t), and -T specifies the transformation described in . The reg command takes seconds to run on a desktop CPU. Thus, the entire inference procedure took roughly 1 day using 4 CPU, compared to the original analysis based on rejection sampling, which took many CPU-months 
The linear regression approach to ABC analysis  is a fast and flexible method of performing parameter inference from population-genetic data. The software described here facilitates such analyses in a flexible way, and is designed to interact seamlessly with widely-available tools for population-genetic simulation and statistical analysis.
Availability and requirements
The source code is distributed under the terms of the GNU public license and is available from the software section of the author's web site http://www.molpopgen.org. Documentation is also available online, as is a shell script containing a complete example. The software was developed and tested on Linux and Apple's OS X platforms, using the gcc compiler suite http://gcc.gnu.org. In order to compile and use the software, the GNU C++ compiler (g ++ ) is needed, and GSL must be installed on the system. The GSL is readily available as a pre-compiled package on many Unix-like systems, and is easily installable from source code on any system with a C compiler.
- Griffiths RC, Tavare S: Sampling theory for neutral alleles in a varying environment. Philosophical transactions: biological sciences. 1994, 334: 403-410. 10.1098/rstb.1994.0079.View ArticleGoogle Scholar
- Hey J, Nielsen R: Improved integration with the Felsenstein equation for improved Marcov Chain Monte Carlo methods in population genetics. PNAS. 2007, 104: 2785-2790. 10.1073/pnas.0611164104.PubMed CentralView ArticlePubMedGoogle Scholar
- Sawyer S, Hartl DL: Population genetics of polymorphism and divergence. Genetics. 1992, 132: 1161-1176.PubMed CentralPubMedGoogle Scholar
- Williamson S, Fledel-Alon A, Bustamante CD: Population Genetics of Polymorphism and Divergence for Diploid Selection Models With Arbitrary Dominance. Genetics. 2004, 168: 463-475. 10.1534/genetics.103.024745.PubMed CentralView ArticlePubMedGoogle Scholar
- Weiss G, von Haeseler A: Inference of Population History Using a Likelihood Approach. Genetics. 1998, 149: 1539-1546.PubMed CentralPubMedGoogle Scholar
- Wall JD: A comparison of estimators of the population recombination rate. Mol Biol Evol. 2000, 17 (1): 156-163.View ArticlePubMedGoogle Scholar
- Fu YX: Estimating the age of the common ancestor of a DNA sample using the number of segregating sites. Genetics. 1996, 144: 829-838.PubMed CentralPubMedGoogle Scholar
- Pritchard JK, Seielstad MT, Perez-Lezaun A, Feldman MW: Population growth of human Y chromosomes: A study of Y chromosome microsatellites. Mol Biol Evol. 1999, 16 (12): 1791-1798.View ArticlePubMedGoogle Scholar
- Przeworski M: Estimating the time since the fixation of a beneficial allele. Genetics. 2003, 164: 1667-1676.PubMed CentralPubMedGoogle Scholar
- Haddrill P, Thornton K, Andolfatto P, Charlesworth B: Multilocus patterns of nucleotide variability and the demographic and selection history of Drosophila melanogaster populations. Genome Research. 2005, 15: 790-799. 10.1101/gr.3541005.PubMed CentralView ArticlePubMedGoogle Scholar
- Thornton K, Andolfatto P: Approximate Bayesian Inference reveals evidence for a recent, severe, bottleneck in a Netherlands population of Drosophila melanogaster. Genetics. 2006, 172: 1607-1619. 10.1534/genetics.105.048223.PubMed CentralView ArticlePubMedGoogle Scholar
- Beaumont MA, Zhang W, Balding DJ: Approximate Bayesian Computation in Population Genetics. Genetics. 2002, 162: 2025-2035.PubMed CentralPubMedGoogle Scholar
- Marjoram P, Molitor J, Plagnol V, Tavare S: Markov chain Monte Carlo without likelihoods. Proc National Acad Sciences United States Am. 2003, 100: 15324-15328. 10.1073/pnas.0306899100.View ArticleGoogle Scholar
- Sisson SA, Fan Y, Tanaka MM: Sequential Monte Carlo without likelihoods. PNAS. 2007, 104: 1760-1765. 10.1073/pnas.0607208104.PubMed CentralView ArticlePubMedGoogle Scholar
- Joyce P, Marjoram P: Approximately sufficient statistics in Bayesian computation. Stat Appl Genet Mol Biol. 2008, 7 (1): Article26-PubMedGoogle Scholar
- Toni T, Welch D, Strelkowa N, Ipsen A, Stumpf MPH: Approximate Bayesian computation scheme for parameter inference and model selection in dynamical systems. J R Soc Interface. 2009, 6: 187-202. 10.1098/rsif.2008.0172.PubMed CentralView ArticlePubMedGoogle Scholar
- Thornton K: Recombination and the properties of Tajima's D in the context of approximate likelihood calculation. Genetics. 2005, 171: 2143-2148. 10.1534/genetics.105.043786.PubMed CentralView ArticlePubMedGoogle Scholar
- Hudson RR: Generating samples under a Wright-Fisher neutral model of genetic variation. Bioinformatics. 2002, 18: 337-338. 10.1093/bioinformatics/18.2.337.View ArticlePubMedGoogle Scholar
- Coop G, Griffiths RC: Ancestral inference on gene trees under selection. Theoretical Population Biology. 2004, 66: 219-232. 10.1016/j.tpb.2004.06.006.View ArticlePubMedGoogle Scholar
- Thornton K, Jensen JD: Controlling the false positive rate in multilocus genome scans for selection. Genetics. 2007, 175: 737-750. 10.1534/genetics.106.064642.PubMed CentralView ArticlePubMedGoogle Scholar
- Thornton K: libsequence: a C++ class library for evolutionary genetic analysis. Bioinformatics. 2003, 19: 2325-2327. 10.1093/bioinformatics/btg316.View ArticlePubMedGoogle Scholar
- R Development Core Team: R: A language and environment for statistical computing. 2004, R Foundation for Statistical Computing, Vienna, Austria, [http://www.R-project.org]Google Scholar
- Stroustrup B: The C++ programming language. 1997, Reading, MA: Addison-Wesley, 3Google Scholar
- Hamilton G, Stoneking M, Excoffier L: Molecular analysis reveals tighter social regulation of immigration in patrilocal populations than in matrilocal populations. PNAS. 2005, 102: 746-7480. 10.1073/pnas.0409253102.View ArticleGoogle Scholar
- Glinka S, Ometto L, Mousset S, Stephan W, DeLorenzo D: Demography and natural selection have shaped genetic variation in Drosophila melanogaster : A multi-locus approach. Genetics. 2003, 165: 1269-1278.PubMed CentralPubMedGoogle Scholar
- Tajima F: Evolutionary relationship of DNA sequences in finite populations. Genetics. 1983, 105: 437-460.PubMed CentralPubMedGoogle Scholar
- Zeng K, Shi S, Wu CI: Compound Tests for the Detection of Hitchhiking Under Positive Selection. Molecular Biology and Evolution. 2007, 24: 1898-1908. 10.1093/molbev/msm119.View ArticlePubMedGoogle Scholar
- Wakeley J: Coalescent Theory: An Introduction. 2009, Greenwood Village, Colorado: Roberts & Company PublishersGoogle Scholar
- Hudson RR, Kreitman M, Aguade M: A test of neutral molecular evolution based on nucleotide data. Genetics. 1987, 116: 153-159.PubMed CentralPubMedGoogle Scholar
This article is published under license to BioMed Central Ltd. This is an Open Access article distributed under the terms of the Creative Commons Attribution License (http://creativecommons.org/licenses/by/2.0), which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.