An Accurate Sequentially Markov Conditional Sampling Distribution for the Coalescent With Recombination (original) (raw)
Abstract
The sequentially Markov coalescent is a simplified genealogical process that aims to capture the essential features of the full coalescent model with recombination, while being scalable in the number of loci. In this article, the sequentially Markov framework is applied to the conditional sampling distribution (CSD), which is at the core of many statistical tools for population genetic analyses. Briefly, the CSD describes the probability that an additionally sampled DNA sequence is of a certain type, given that a collection of sequences has already been observed. A hidden Markov model (HMM) formulation of the sequentially Markov CSD is developed here, yielding an algorithm with time complexity linear in both the number of loci and the number of haplotypes. This work provides a highly accurate, practical approximation to a recently introduced CSD derived from the diffusion process associated with the coalescent with recombination. It is empirically demonstrated that the improvement in accuracy of the new CSD over previously proposed HMM-based CSDs increases substantially with the number of loci. The framework presented here can be adopted in a wide range of applications in population genetics, including imputing missing sequence data, estimating recombination rates, and inferring human colonization history.
THE conditional sampling distribution (CSD) describes the probability, under a particular population genetic model, that an additionally sampled DNA sequence is of a certain type, given that a collection of sequences has already been observed. In many important settings, the relevant population genetic model is the coalescent with recombination, for which the true CSD, denoted by π, does not have a known analytic formula. Nevertheless, the CSD π and, in particular, approximations thereof have found a wide range of applications in population genetics.
One important problem in which the CSD plays a fundamental role is describing the posterior distribution of genealogies under the coalescent process. Stephens and Donnelly (2000) showed that the true posterior distribution can be written in terms of π and can be approximated by using an approximate CSD, denoted . This observation has been used (Stephens and Donnelly 2000; Fearnhead and Donnelly 2001; De Iorio and Griffiths 2004a,b; Fearnhead and Smith 2005; Griffiths et al. 2008) to construct importance sampling schemes for likelihood computation and ancestral inference under the coalescent, including extensions such as recombination and population structure. In conjunction with composite-likelihood frameworks (Hudson 2001; Fearnhead and Donnelly 2002), these importance sampling methods have been used, for example, to estimate fine-scale recombination rates (McVean et al. 2004; Fearnhead and Smith 2005; Johnson and Slatkin 2009).
Li and Stephens (2003) introduced a different application of the CSD, observing that the probability of sampling a set of haplotypes can be decomposed into a product of CSDs and therefore can be approximated by a product of approximate CSDs . Similar applications of the CSD have yielded methods for estimating recombination rates (Li and Stephens 2003; Crawford et al. 2004; Stephens and Scheet 2005) and gene conversion parameters (Gay et al. 2007; Yin et al. 2009), for phasing genotype data into haplotype data (Stephens and Scheet 2005), for imputing missing data to improve power in association studies (Stephens and Scheet 2005; Li and Abecasis 2006; Scheet and Stephens 2006; Marchini et al. 2007; Howie et al. 2009), for inferring ancestry in admixed populations (Price et al. 2009), and for inferring demography (Hellenthal et al. 2008; Davison et al. 2009).
In all applications, the fidelity with which the surrogate CSD approximates the true CSD π is critical to the quality of the result. Furthermore, the time required to compute probabilities under the CSD is important, as many of the above methods are now routinely applied to genome-scale data sets. As a result, many approximate CSDs have been proposed, particularly for the coalescent with recombination. Fearnhead and Donnelly (2001) introduced an approximation in which an additionally sampled haplotype is constructed as an imperfect mosaic of previously sampled haplotypes, with mosaic breakpoints caused by recombination events and imperfections corresponding to mutation events. The resulting CSD, which we denote by , can be cast as a hidden Markov model (HMM), and the associated conditional sampling probability (CSP) can be computed with time complexity linear in both the number of previously sampled haplotypes and the number of loci. Li and Stephens (2003) proposed a related model that can be viewed as a modification to limiting the state space of the HMM, hence providing a constant factor improvement in the time complexity; we denote the corresponding CSD by .
Following the theoretical work of De Iorio and Griffiths (2004a), Griffiths et al. (2008) derived an approximate CSD from the Wright–Fisher diffusion process associated with the two-locus coalescent with recombination. More recently, Paul and Song (2010) generalized this work to an arbitrary number of loci and demonstrated that the resulting CSD, which we denote by , can also be described by a genealogical process. Though it is more accurate than both and , computing the CSP under has time complexity superexponential in the number of loci. To ameliorate this limitation, Paul and Song introduced the approximate CSD , which follows from prohibiting coalescence events in the genealogical process associated with . Computing the CSP under has time complexity exponential in the number of loci. Although this is an improvement over the superexponential complexity associated with , it is still impracticable to use for >20 loci.
In this article, we introduce an alternate approximation that is scalable in the number of loci, while maintaining the key features of that lead to high accuracy. Specifically, motivated by the sequentially Markov coalescent (SMC) introduced by McVean and Cardin (2005), we derive a sequentially Markov approximation to . The key idea is to consider the marginal genealogies at each locus sequentially, using the genealogical description of . In general, the sequence of marginal genealogies is not Markov, but, as in McVean and Cardin (2005), we make approximations to provide a Markov construction for the sequence. We denote the resulting approximation of by . The CSD can also be obtained from by prohibiting a certain class of coalescence events, a fact that mirrors the relation between the SMC and the coalescent with recombination (McVean and Cardin 2005). We formalize this relation by proving that is, in fact, equal to .
Due to its sequentially Markov construction, can be cast as an HMM. Unfortunately, the state space of the HMM is continuous, and so efficient algorithms for CSP computation and posterior inference are not known. Our solution is to discretize the state space. The discretization procedure we develop is related, though not identical, to the Gaussian quadrature method employed by Stephens and Donnelly (2000) and Fearnhead and Donnelly (2001). Although we focus on the CSD problem here, we believe that our general approach has the potential to foster applications of the SMC in other settings as well (see Hobolth et al. 2007; Dutheil et al. 2009).
Having discretized the continuous state space, we apply standard HMM theory to obtain an efficient dynamic program for computing the CSP under the discretized approximation of . The resulting time complexity is linear in both the number of previously sampled haplotypes and the number of loci. This time complexity is the same as that for and and hence is a substantial improvement over . In summary, the work presented here provides a practical approximation to , which was derived from the diffusion process associated with the coalescent with recombination. Furthermore, as detailed later, the improvement in accuracy of our new CSD over and increases substantially with the number of loci.
The remainder of this article is organized as follows. In model, we present the necessary notation and background and describe our new CSD . We also give an overview of the proof that is equivalent to and demonstrate several other useful properties. In discretization of the hmm, we describe the discretization of , and in empirical results, we provide empirical evidence that the discretized approximation performs well, with regard to both accuracy and run time. Finally, in discussion we mention some connections to existing models and describe possible applications and extensions, in particular conditionally sampling more than one haplotype.
MODEL
In this section, we describe the key transition and emission distributions for the HMM underlying . Further, we demonstrate that is equivalent to , the variant of with coalescence disallowed, and also show that the transition density satisfies several useful properties.
Notation:
We consider haplotypes in the finite-sites finite-alleles setting. Denote the set of loci by L = {1, … ,k} and the set of alleles at locus ℓ ∈ L by E_ℓ. Mutations occur at locus ℓ ∈ L at rate θℓ/2 and according to the stochastic matrix . Denote the set of breakpoints by B = {(1, 2), … , (k − 1, k)}, where recombination occurs at breakpoint b ∈ B at rate ρ_b/2.
The space of _k_-locus haplotypes is denoted by ℋ = _E_1 × … × Ek. Given a haplotype α ∈ ℋ, we denote by α[ℓ] ∈ _E_ℓ the allele at locus ℓ ∈ L and by α[1 : ℓ] the partial haplotype (α[1], … , α[ℓ]). A sample configuration of haplotypes is specified by a vector , with _n_α being the number of haplotypes of type α in the sample. The total number of haplotypes in the sample is denoted by |n| = n. Finally, we use eα to denote the singleton configuration comprising a single α haplotype.
A brief review of the CSD :
The approximate CSD is described by a genealogical process closely related to the coalescent with recombination. We provide below a brief description of the framework and refer the reader to Paul and Song (2010) for further details.
Suppose that, conditioned on having already observed a haplotype configuration , we wish to sample a new haplotype α. Define 𝒜*(n) to be the nonrandom trunk ancestry for , in which lineages associated with the haplotypes do not mutate, recombine, or coalesce with one another, but rather extend infinitely into the past. We assume that the unknown ancestry associated with is 𝒜*(n) and sample a conditional ancestry C associated with α. Within the conditional ancestry, lineages evolve backward in time with the following rates:
- Mutation: Each lineage mutates at locus ℓ ∈ L with rate θℓ/2, according to P(ℓ).
- Recombination: Each lineage undergoes recombination at breakpoint b ∈ B with rate ρ_b_/2.
- Coalescence: Each pair of lineages coalesces with rate 1.
- Absorption: Each lineage is absorbed into each lineage of 𝒜*(n) at rate 1/2.
When every lineage has been absorbed into 𝒜*(n), the process terminates. The type of every lineage in C can now be inferred, and a sample for α is generated. An illustration of this process is presented in Figure 1A.
Figure 1.—
Illustration of the corresponding genealogical and sequential interpretations for a realization of . The three loci of each haplotype are each represented by a solid circle, with the color indicating the allelic type at that locus. The trunk genealogy 𝒜*(n) and conditional genealogy C are indicated. Time is represented vertically, with the present (time 0) at the bottom of the illustration. (A) The genealogical interpretation: Mutation events, along with the locus and resulting haplotype, are indicated by small arrows. Recombination events, and the resulting haplotype, are indicated by branching events in C. Absorption events, and the corresponding absorption time [t(a) and t(b)] and haplotype [h(a) and h(b), respectively], are indicated by dotted-dashed horizontal lines. (B) The corresponding sequential interpretation: The marginal genealogies at the first, second, and third locus (_S_1, _S_2, and _S_3) are emphasized as dotted, dashed, and solid lines, respectively. Mutation events at each locus, along with resulting allele, are indicated by small arrows. Absorption events at each locus are indicated by horizontal lines.
Although a recursion for computing the CSP is known (Paul and Song 2010, Equation 7), it is computationally intractable, and Paul and Song approximate the genealogical process by disallowing coalescence within the conditional genealogy, denoting the resulting CSD by . The recursion for (Paul and Song 2010, Equation 12) is amenable to dynamic programming, though it still has time complexity exponential in the number k of loci.
The sequentially Markov coalescent:
The sequential interpretation of the coalescent with recombination was introduced by Wiuf and Hein (1999). They observed that an ancestral recombination graph (ARG) may be simulated sequentially along the chromosome. In particular, the marginal coalescent tree at a given locus can be sampled conditional on the marginal ARG for all previous loci. The full ARG is then sampled by first sampling a coalescent tree at the leftmost locus and then proceeding to the right.
McVean and Cardin (2005) proposed a simplification of this process. Though McVean and Cardin presented their work for the infinite-sites model, we state (but do not derive) the analogous results for a finite-sites, finite-alleles model. In their approach, the marginal coalescent tree at locus ℓ is sampled conditional only on the marginal coalescent tree at locus ℓ − 1. In particular, setting b = (ℓ − 1, ℓ) ∈ B, (1) recombination breakpoints are realized as a Poisson process with rate ρ_b_/2 on the marginal coalescent tree at locus ℓ − 1, (2) the lineage branching from each recombination breakpoint associated with locus ℓ − 1 is removed, and (3) the lineage branching from each recombination breakpoint associated with locus ℓ is subject to coalescence with other lineages at rate 1. The resulting tree is the marginal genealogy at locus ℓ. This approximation is called the sequentially Markov coalescent (SMC) and is equivalent to a variant of the coalescent with recombination that disallows coalescence between lineages ancestral to disjoint regions of the sequence (McVean and Cardin 2005).
The sequentially Markov CSD
We now describe a sequentially Markov approximation to the genealogical process underlying . Our construction is similar to that given by McVean and Cardin (2005), described above, though the resulting dynamics are less involved since the conditional genealogy is constructed for a single haplotype. First, observe that under , the marginal conditional genealogy at a given locus ℓ ∈ L is entirely determined by two random variables: the absorption time, which we denote _T_ℓ, and the absorption haplotype, which we denote _H_ℓ. The present corresponds to time 0 and _T_ℓ ∈ [0, ∞]. See Figure 1B for an illustration. For convenience, we write _S_ℓ = (_T_ℓ, _H_ℓ) for the random marginal conditional genealogy at locus ℓ ∈ L and _s_ℓ = (_t_ℓ, _h_ℓ) for a realization.
Within the marginal conditional genealogy at locus ℓ ∈ L, note that _T_ℓ and _H_ℓ are independent, with _T_ℓ distributed exponentially with parameter n/2 and _H_ℓ distributed uniformly over the n haplotypes of . Thus, the marginal conditional genealogy _S_ℓ at locus ℓ is distributed with density ζ(n), where
(1) |
---|
Conditioning on _S_ℓ−1= _s_ℓ−1=(_t_ℓ−1, _h_ℓ−1), the marginal conditional genealogy S_ℓ, for ℓ ≥ 2, is sampled by a process analogous to that described above for the SMC. Setting b = (ℓ − 1, ℓ) ∈ B, the sampling procedure is as follows (see Figure 2 for an accompanying illustration): (1) Recombination breakpoints are realized as a Poisson process with rate ρ_b/2 on the marginal conditional genealogy _s_ℓ−1; (2) going backward in time, the lineage associated with locus ℓ−1 branching from each recombination breakpoint is removed, so that only the lineage more recent than the first (i.e., the most recent) breakpoint remains; and (3) the lineage associated with locus ℓ branching from the first recombination breakpoint is absorbed into a particular lineage of 𝒜*(n) at rate 1/2.
Figure 2.—
Illustration of the (Markov) process for sampling the absorption time _T_ℓ given the absorption time _T_ℓ−1 = t_ℓ−1. In step 1, recombination breakpoints are realized as a Poisson process with rate ρ_b/2 on the marginal conditional genealogy with absorption time _t_ℓ−1. In step 2, the lineage branching from each breakpoint associated with locus ℓ−1 is removed, so that only the lineage more recent than the first breakpoint, at time _T_r, remains. In step 3, the lineage branching from the first recombination breakpoint associated with locus ℓ is absorbed after time _T_a distributed exponentially with rate n/2. Thus, _T_ℓ = _T_r + _T_a.
From the above description, we deduce that there is no recombination between loci ℓ−1 and ℓ with probability , and in this case the marginal conditional genealogy is unchanged; that is, _S_ℓ = _s_ℓ−1. Otherwise, the time T_r of the first recombination breakpoint is distributed exponentially with parameter ρ_b/2, truncated at time _t_ℓ−1, and the additional time _T_a until absorption is distributed exponentially with parameter n/2. Thus we have _S_ℓ = (_T_r + _T_a, _H_ℓ), where _H_ℓ is chosen uniformly at random from the sample . Taking a convolution of _T_r and _T_a, the transition density is given by
(2) |
---|
where .
Finally, conditioning on _S_ℓ = _s_ℓ, recall that mutations are realized as a Poisson process (cf. Stephens and Donnelly 2000) with rate θℓ/2. Therefore, a particular allele a ∈ _E_ℓ is observed with probability
(3) |
---|
Hereafter, we omit the superscript and the subscripts θ_ℓ_ and ρ_b_ from these densities, whenever the context is unambiguous.
The sequentially Markov approximation to can be cast as a continuous-state HMM. In generating a haplotype α, the observed state, the hidden state, and initial, transition, and emission densities are given by the following:
- Observed state: At locus ℓ ∈ L, the observed state is the allele α[ℓ].
- Hidden state: At locus ℓ ∈ L, the hidden state is the marginal genealogy _S_ℓ = (_T_ℓ, _H_ℓ).
- Initial density: ζ is defined in (1).
- Transition density: φ is defined in (2).
- Emission density: ξ is defined in (3).
Writing for the sequentially Markov approximation to , we can use the forward recursion (see, e.g., Doucet and Johansen 2008) to get
(4) |
---|
where _f_SMC (·, ·) is defined by
(5) |
---|
with base case
(6) |
---|
Though we cannot analytically solve the above recursion for , in the next section we derive a discretized approximation with time complexity linear in both the number of loci k and the number of haplotypes n. Before doing so, we briefly discuss some appealing properties of .
Properties of :
Recall that the SMC approximation of McVean and Cardin (2005) is equivalent to a variant of the coalescent with recombination disallowing coalescence events between lineages ancestral to disjoint regions. Similarly, the CSD , when used to sample a single haplotype, is a variant of disallowing the same class of coalescence events. We might therefore expect that the sequentially Markov approximation of described above is equivalent to , and in fact we can show that this is true.
Proposition 1. For an arbitrary single haplotype α ∈ ℋ and haplotype configuration , .
We present a sketch of the proof here and refer the reader to supporting information, File S1, for further details.
Sketch of Proof. The key idea of the proof is to introduce a genealogical recursion for f(α, sk), the joint density function associated with sampling haplotype α and the marginal genealogy at the last locus sk. This recursion can be constructed following the lines of Griffiths and Tavaré (1994) to explicitly incorporate coalescent time into a genealogical recursion.
By partitioning with respect to the most recent event occurring at the last locus k, it is possible to inductively show that _f_SMC(α, sk) = f(α, sk). Furthermore, the equality can be verified, and thus we conclude that
We now describe other intuitively appealing properties of . In particular, it can be verified that the detailed-balance condition
(7) |
---|
holds for the initial and transition densities, ζ and φ, respectively. This immediately implies that the initial distribution ζ is stationary under the given transition dynamics; i.e., the invariance condition
is satisfied. Thus, _S_ℓ is marginally distributed according to ζ for all loci ℓ ∈ L, and in particular the marginal distribution of _T_ℓ is exponential with rate n/2. This parallels the fact that the marginal genealogies under the SMC (and the coalescent with recombination) are distributed according to Kingman's coalescent.
Similarly, the transition density exhibits a consistency property, which we call the locus-skipping property. Intuitively, this property states that transitioning directly from locus ℓ − 1 to ℓ + 1 can be accomplished by using the transition density parameterized with the sum of the recombination rates. Formally, the following equality holds for all ρ1, ρ2 ≥ 0:
(8) |
---|
This property, in conjunction with recursion (5), is computationally useful, as it enables loci ℓ ∈ L for which α[ℓ] is unobserved to be skipped in computing the CSP .
Finally, the conditional expectation of _T_ℓ given _T_ℓ−1 = _t_ℓ−1 is
(9) |
---|
where b = (ℓ − 1, ℓ) ∈ B. Asymptotically, this expression provides several intuitive results. As ρ_b_ → ∞, E[_T_ℓ|_t_ℓ−1] → 2/n; that is, recombination happens immediately, and 2/n is the expectation of the additional absorption time T_a. As ρ_b → 0, we get E[_T_ℓ|_t_ℓ − 1] → _t_ℓ−1. In this case there is no recombination, and the absorption time does not change. Further, E[_T_ℓ|t_ℓ − 1] → 2/ρ_b + 2/n holds as _t_ℓ−1 → ∞. Here, recombination must occur, and the exponentially distributed time is not truncated, so the expectation is the sum of the expectations of two exponentials. Finally, as _t_ℓ−1 → 0 we have E[_T_ℓ|_t_ℓ − 1] → 0. No recombination can occur, and so the absorption time is unchanged.
DISCRETIZATION OF THE HMM
In the previous section we described a sequentially Markov approximation of the CSD and showed that it can be cast as an HMM. Because the absorption time component of the hidden state is continuous, the dynamic program associated with the classical HMM forward recursion is not applicable. However, by discretizing the continuous component, we are once again able to obtain a dynamic programming algorithm, resulting in an approximate CSP computation linear in both the number of loci and the number of haplotypes.
Rescaling time:
Recall from the previous section that the marginal absorption time at each locus is exponentially distributed with parameter n/2. To use the same discretization for all n, we follow Stephens and Donnelly (2000) and Fearnhead and Donnelly (2001) and transform the absorption time to a more natural scale in which the marginal absorption time is independent of n. In particular, define the transformed state Σ = (𝒯, H) where 𝒯 = (n/2)T. We denote a realization of Σ by σ = (τ, h). In the appendix, we provide expressions for the transformed quantities and derived from (1), (2), (3), and (5), respectively.
Using this time-rescaled model, the marginal absorption time at each locus is exponentially distributed with parameter 1. Because this distribution is independent of n and the coalescent model parameters ρ and θ, we expect that a single discretization of the transformed absorption time is appropriate for a wide range of haplotype configurations and parameter values.
Discretizing absorption time:
Our next objective is to discretize the absorption time 𝒯 ∈ R≥0. Let 0 = _x_0 < _x_1 < ··· < xd = ∞ be a finite strictly increasing sequence in R≥0 ∪ {∞} so that D = {Dj = [x _j_−1, xj)}_j_=1, … ,d is a _d_-partition of R≥0.
Toward formulating a _D_-discretized version of the dynamics exhibited by the transformed HMM, we define the following _D_-discretized version of the density
(10) |
---|
for all ℓ ∈ L. Unfortunately, we cannot obtain a recursion for via the definition of . Therefore, we make an additional approximation, namely that the transition and emission densities are conditionally dependent on the absorption time 𝒯 only through the event {Dj ∋ 𝒯 }; i.e., the densities depend on the interval Dj to which 𝒯 belongs but not on the actual value of 𝒯. Abusing notation, define and as the transition and emission densities, respectively, conditioned on the event {Dj ∋ 𝒯}. Formally, we make the following approximations:
(11) |
---|
(12) |
---|
Together with the building blocks of the time-rescaled HMM, these assumptions provide a recursive approximation of , which we denote by . Specifically, assumptions (11) and (12) imply that the integral recursion for reduces to the discrete recursion
(13) |
---|
with base case
(14) |
---|
where we have defined distributions and . Setting , we get
(15) |
---|
Turning to the transition density , which is conditioned on the event {Dj ∋ 𝒯}, and recalling that 𝒯 is marginally exponentially distributed with parameter 1, we obtain
(16) |
---|
with analytic expressions for y(i) and z(i,j) provided in the appendix. Note that assumption (11) is not used here; rather, the formula follows from using the time-rescaled version of the transition density (2) in the double integral. An expression for the emission density can be similarly obtained,
(17) |
---|
with an analytic expression for υ(i)(k) also given in the appendix. Again, assumption (12) is not used here; the second equality of (17) follows from using the time-rescaled version of the emission probability (3) in the integral. In summary, can be computed efficiently using (13), and can be approximated by
(18) |
---|
Equations 13–18 provide the requisite _D_-discretized versions of the transformed densities. Note that these equations characterize an HMM; that the Markov property holds on the discretized state space D follows from assumptions (11) and (12) (Rosenblatt 1959). In fact, (13–18) may alternatively be obtained by assuming that the Markov property holds on D and writing down the relevant transition and emission probabilities with the interpretations given above. In the remainder of this section, we examine some general properties of the discretized dynamics and also provide one method for choosing a discretization D.
Computational complexity of the discretized recursion:
We first consider the asymptotic complexity of computing the CSP under the _D_-discretized approximation for . Substituting Equation 16 into the key recursion (13) gives
(19) |
---|
for ℓ ≥ 2. For a fixed discretization D, the expressions , y(i), and z(i,j) depend only on the total sample size n, the mutation and recombination rates (θℓ and ρℓ), and the boundary points _x_0, … , xd of D; these may be precomputed and cached for relevant ranges of values. In conjunction with the base case (14), there is a dynamic program (see the appendix for details) for computing the CSP under the _D_-discretized approximation (18) for with time complexity O(k · (nd + _d_2)), where k is the number of loci. As in Fearnhead and Donnelly (2001), this time complexity is better than O(k · (nd)2), the result that would be obtained by naive use of the HMM forward algorithm.
Properties of the discretization:
Recall the detailed-balance condition (7) associated with . Using expressions (15) and (16), together with Bayes' rule, we find that
(20) |
---|
holds (the details are provided in the appendix). Thus, the discretized approximation of satisfies an analogous detailed balance condition. As a result, the marginal distribution at each locus of the discretized Markov chain is (again) given by and the approximation exhibits the expected symmetries; for example, equal CSPs are computed whether starting at the leftmost locus and proceeding right or starting at the rightmost locus and proceeding left.
Furthermore, recall the locus-skipping property (8) associated with . The first equality in (16) and assumption (11) imply the relation
(21) |
---|
for all ρ1, ρ2 ≥ 0 (see the appendix for details). Thus, the discretized approximation of approximately satisfies an analogous locus-skipping condition, up to the error introduced via approximation (11). This approximation is particularly useful in scenarios when data are missing (i.e., α[_ℓ_] is unknown for one or more ℓ ∈ L), since this property reduces the time complexity of the dynamic program given above. In particular, when m of the k loci are missing, the time complexity is reduced to O((k − m) · (nd + _d_2)). This is relevant, for example, in importance sampling applications (Fearnhead and Donnelly 2001).
Discretization choice and the definition of :
Finally, we discuss a method for choosing a discretization D of the absorption time. Recalling that marginally the transformed absorption time is exponentially distributed with parameter 1, let {(w(j), τ(j))}j = 1, … ,d be the _d_-point Gaussian quadrature associated with the function f(τ) = _e_−τ (Abramowitz and Stegun 1972, Section 25.4.45). Set _x_0 = 0, and set xj such that . Since , the points determine a partition D = {Dj = [x _j_−1, xj)}_j_=1, … ,d of R≥0.
The use of Gaussian quadrature evokes the work of Stephens and Donnelly (2000) and Fearnhead and Donnelly (2001). Although the method we employ is related, it is different in that we do not use the quadrature directly [for example, the values of the quadrature points {τ(j)} are never used explicitly]; rather, we use the Gaussian quadrature as a reasonable way of choosing a discretization D. We henceforth write for the _d_-point Gaussian quadrature-discretized version of .
EMPIRICAL RESULTS
In the previous section, we defined a discretized approximation of the CSD . In this section, we examine the accuracy of this approximation and also compare it to the widely used CSDs and , thereby providing evidence that is a more accurate and computationally tractable CSD.
Data simulation:
For simplicity, we consider a two-allele model with , θ_ℓ_ = θ for ℓ ∈ L and ρ_b_ = ρ for b ∈ B. We sample a _k_-locus haplotype configuration by (i) using a coalescent with recombination simulator, with ρ = ρ0 and θ = θ0, to sample a _k_0-locus (with _k_0 >> k) _n_-haplotype configuration n0, and (ii) restricting attention to the central k segregating loci in n0. This procedure corresponds to the usage of the CSD on typical genomic data, in which only segregating sites are considered.
Given a _k_-locus _n_-haplotype configuration n, we obtain a _k_-locus _n_-haplotype conditional configuration C = (α, n − eα) by withholding a single haplotype α from uniformly at random. For notational simplicity, we define π on such a conditional configuration in the natural way: π(C) = π(α | n − eα).
CSD accuracy:
We evaluate the accuracy of a CSD relative to a reference CSD π0 using the expected absolute log-ratio (ALR) error,
(22) |
---|
where N denotes the number of simulated data sets and C(i) is a _k_-locus _n_-haplotype conditional configuration sampled as indicated above, and both and π0 are evaluated using the true parameter values θ = θ0 and ρ = ρ0. For example, if , the CSP obtained using differs from that obtained by π0 by a factor of 10, on average, for a randomly sampled _k_-locus _n_-haplotype conditional configuration.
Using the ALR error, we evaluate the accuracy of several CSDs: (Fearnhead and Donnelly 2001); (Li and Stephens 2003); , evaluated using the recursion for (Paul and Song 2010); and , the _d_-point quadrature-discretized version of , for d ∈ {4, 18, 16}. We also evaluate , a variant of introduced in Paul and Song (2010) with computational time complexity O(_k_3 · n); the CSD is described in more detail in the appendix.
In what follows, we set θ0 = 0.01 and ρ0 = 0.05 and fix n = 10. For k ≤ 10, it is possible to obtain a very good approximation to the true CSD π using computationally intensive importance sampling. The resulting values of ALRErr_k_,n(· | π) are plotted in Figure 3A, as a function of k. Supporting the conclusion of Paul and Song (2010), is more accurate than both and , with the disparity increasing as k increases. Moreover, the CSD is nearly as accurate as , suggesting that the discretization is fairly accurate even for modest values of d. Finally, the CSD has accuracy that is indistinguishable from .
Figure 3.—
Absolute log-ratio error (ALRErr) of various conditional sampling distributions. See (22) for a formal definition of ALRErr_k,n_(· | ·). The accuracy of is almost indistinguishable from that of , the most accurate of all approximate CSDs considered here. As expected, discretization reduces the accuracy somewhat, but even is substantially more accurate than and . With θ0 = 0.01 and ρ0 = 0.05, we used the methodology described in the text to sample 250 conditional configurations, each with n = 10 haplotypes and k loci. (A) Error is measured relative to the true CSD π, estimated using computationally intensive importance sampling. (B) Error is measured relative to , computed by numerically solving a recursion for the equivalent CSD .
To investigate these results as k increases, we consider the ALR error relative to , which can be evaluated exactly for ; the resulting values of are plotted in Figure 3B, as a function of k. As k increases, both and continue to diverge from , suggesting that the increasing disparity in accuracy, directly observable in Figure 3A, continues for larger values of k. As expected, the discretized approximation shows increased fidelity to for larger values of d, and even is substantially more accurate, relative to , than are and .
It is too computationally expensive to compute for k > 20. However, Figure 3B suggests that the CSD is nearly indistinguishable from . Motivated by this observation, we consider the error relative to for k > 20. The values of and the analogously defined signed log-ratio (SLR) error are plotted as a function of k in Figure 4, A and B, respectively. The trends observed in Figure 3 are recapitulated in Figure 4A, suggesting that they continue to hold for substantially larger values of k. Interestingly, Figure 4B shows that and produce values significantly smaller than (and ); for example, takes values that are, on average, a factor of 10 smaller than for k = 100. In conjunction with our conclusion that is more accurate than and , this suggests a similar systematic error with respect to the true CSD.
Figure 4.—
Comparison of the accuracy of various conditional sampling distributions relative to (see Figure 3 for the accuracy of ). A and B illustrate that the improvement in accuracy of over and is amplified as the number of loci k increases and that both and produce significantly smaller values than (and ). For θ0 = 0.01 and ρ0 = 0.05, we used the methodology described in the text to sample 250 conditional configurations with n = 10 haplotypes and k loci. (A) Absolute log-ratio error. (B) Signed log-ratio error.
For a discussion of CSD accuracy in the context of the product of approximate conditionals (PAC) method (Li and Stephens 2003), we refer the reader to Paul and Song (2010). Since is very close to (as demonstrated in the present paper), we anticipate that using it produces similar results for PAC likelihood estimation and recombination rate inference.
Running time comparison:
We next consider the empirically observed running time required to compute each CSP. The results, obtained using the conditional configurations with n = 10 and k ∈ {1, … , 100} simulated as previously described, are presented in Table 1. Looking across each row, it is evident that the running time under , and depends linearly on the number of loci k, matching the asymptotic time complexity. Similarly, the running time under is well matched by the theoretical cubic dependence on k.
TABLE 1.
Asymptotic time complexity and empirically observed average running time
Next, comparing , and , observe that the running time for is approximately a factor of 10 slower than and approximately a factor of 2 slower than . Similarly, is approximately a factor of 20 and of 4 slower than and , respectively; and is approximately a factor of 40 and of 8 slower than and , respectively. Importantly, these factors are constant, depending on neither the number of loci k nor the number of haplotypes n. Also note that the time required to compute the CSD for appears to depend linearly, rather than quadratically, on d for the modest (but relevant) values considered.
DISCUSSION
We have formulated a sequentially Markov approximation of , which we call . The relationship between the genealogical process underlying and is analogous to the relationship between the coalescent with recombination and the SMC. In particular, is equivalent to with a certain class of coalescence events disallowed. In the case of sampling one additional haplotype, this corresponds to disallowing all coalescence events, the same approximation used to obtain , and so we find that .
Though the CSD can be cast as an HMM, the associated CSP cannot be evaluated using typical HMM methodology because of the continuous state space; to our knowledge, exact evaluation is possible only via the known recursion for , which has time complexity exponential in the number of loci. By discretizing the continuous state space into d intervals, obtained using Gaussian quadrature, we obtain the discretized approximation for which computing the CSP has time complexity linear in both the number of loci and the number of haplotypes. We find that, even for modest values of d, is a very good approximation of . Importantly, is more accurate than and with only a (small) constant factor penalty in run time. We remark that we investigated alternative methods for discretizing the CSP computation (e.g., point-based rather than interval-based methods), but settled on the described approach as it exhibited desirable properties and is theoretically well motivated.
We attribute the observed increase in accuracy of to the incorporation of two key features of the coalescent with recombination that are not integrated into either or . Consider the genealogy associated with two particular haplotypes within an ARG. First, observe that the times to the most recent common ancestor (MRCA) at two neighboring loci are dependent, even if ancestral lineages at the two loci are separated by a recombination event. explicitly models a Markov approximation to the analogous absorption-time dependence across breakpoints, whereas both and assume independence. Second, if the time to the MRCA at a locus is small, the probability of recombination between this locus and neighboring loci is small, since it would have had to occur prior to the MRCA. While models this property by diminishing the probability of recombination between neighboring loci if the absorption time at the first locus is small, and assume that recombination is independent of absorption time. We believe that and tend to underestimate, on average, the true CSP (as suggested in Figure 4B) due to the omission of these key features. The relationship between several CSDs, including and , is illustrated in Figure 5.
Figure 5.—
Illustration of the relationship between various CSDs. The CSD at the head of each arrow can be seen as an approximation to the CSD at the tail. Each arrow is also annotated with a (short) description of this approximation. The CSDs below the dashed line can be cast as an HMM: Those above the dotted line (including a continuous-state version of , which we denote ) have a continuous and infinite state space, while those below have a finite and discrete state space and are therefore amenable to simple dynamic programming algorithms. For more thorough descriptions of each approximation, see the main text and also Paul and Song (2010). Recall in particular that the equality holds only for conditionally sampling a single haplotype.
Toward future research, recall that the CSD can be extended to sampling more than one additional haplotype (Paul and Song 2010). Of particular importance to population genetics tools (Stephens and Scheet 2005; Marchini et al. 2007; Howie et al. 2009) for diploid organisms is sampling two additional haplotypes. Though we focused on conditionally sampling a single additional haplotype in the present work, we note that the sequentially Markov approximation to is, in principle, applicable to sampling multiple haplotypes. However, the state space of the resulting HMM description increases exponentially with the number of haplotypes. In this domain, we anticipate that randomized techniques for CSP computation, such as importance sampling and Markov chain Monte Carlo, will exhibit high accuracy and the efficiency required for modern data sets. We pursue this line of research in a forthcoming article.
We believe that it is possible to extend the ideas presented here to different demographic scenarios, for example, spatial structure or models of population subdivision (Davison et al. 2009). It should be possible to extend the principled approach of Paul and Song (2010) toward the CSD via the diffusion generator to these scenarios, as in De Iorio and Griffiths (2004b) and Griffiths et al. (2008). In other scenarios, for example varying population size, the principled approach might not be applicable, so one would have to modify the genealogical interpretation heuristically, e.g., varying coalescence rates. As in the present article, prohibiting certain coalescence events in the conditional genealogy should then allow for an efficient implementation of the resulting CSDs as HMMs.
Though the SMC has been used for simulating population genetic samples (Marjoram and Wall 2006; Chen et al. 2009), it can also be cast as an HMM and used for inference in scenarios in which using the full coalescent with recombination is cumbersome. As described above, the state space of the HMM increases exponentially with the number of haplotypes, making exact computation intractable for large numbers of haplotypes. Nevertheless, research (Hobolth et al. 2007; Dutheil et al. 2009) is in progress for modest numbers of haplotypes. We believe that choosing a discretization using Gaussian quadrature, as described in discretization of the hmm, and the forthcoming randomized techniques alluded to above, will foster progress in this area.
We conclude by recalling that a broad range of population genetic tools have been developed, and will continue to be developed, on the basis of the CSD. These tools typically employ , or a similar variant, because the underlying HMM structure admits simple and fast recursions for the relevant calculations (e.g., the CSP). We have introduced a new CSD and a discretized approximation , which also have simple underlying HMM structures and substantially improve upon the accuracy of and . We believe that , when used in the same contexts as and , has the potential to produce more accurate results, with only a small constant factor penalty in run time.
Acknowledgments
This research is supported in part by National Institutes of Health (NIH) grant R01-GM094402, an Alfred P. Sloan Research Fellowship, and a Packard Fellowship for Science and Engineering. The research of J.S.P was funded in part by an NIH National Research Service Award Trainee appointment on T32-HG00047.
APPENDIX
Time-transformed model:
Rewriting the HMM Equations 1–6 in terms of the transformed state Σ = (𝒯, H) introduced in discretization of the hmm yields
(A1) |
---|
where the transformed density is given by
(A2) |
---|
with the base case
(A3) |
---|
The transformed initial, transition, and emission densities are given by
(A4) |
---|
(A5) |
---|
(A6) |
---|
Note that care must be taken upon transforming the Dirac-δ in the expression for .
Analytic expressions for emission and transition probabilities:
We now provide analytic expressions for the quantities y(i), z(i,j), and υ(i)(k) introduced for the transition probability (16) and the emission probability (17). Recalling that Di = [x _i_−1, xi) and Dj = [x _j_−1, xj) and evaluating the associated integrals, we get
(A7) |
---|
(A8) |
---|
for ρ_b_ ≠ n,
(A9) |
---|
for ρ_b_ = n, and
(A10) |
---|
Note that the recursive structure of υ(i)(k) [together with and the sum in Equation 17] suggests an efficient implementation.
Description of the dynamic program for _D_-discretized :
Let D = {_D_1, … , _D_d} be a finite partition of R≥0 as described in the text. Recalling the recursion for given in Equation 19, consider the following dynamic programming algorithm for computing the _D_-discretized approximation of :
- For each Dj ∈ D and h ∈ ℋ such that nh > 0, compute using (14), and set
- For each ℓ ∈ {2, … , k},
- For each Dj ∈ D, compute
- For each Dj ∈ D and h ∈ ℋ such that nh > 0, compute
- 3. Compute _D_-discretized approximation .
The time complexities of steps 2a and 2b are O(_d_2) and O(nd), respectively. The time complexities of steps 1 and 3 are both O(nd). We can therefore conclude that the time complexity of the dynamic program is .
Detailed balance and locus skipping:
The detailed-balance condition (20) for the discretized model can be shown using expressions (15) and (16). Together with Bayes' rule, we find that the following holds:
(A11) |
---|
Using expression (16) and assumption (11) we can show that
(A12) |
---|
holds; thus the locus-skipping property (21) for the discretized model holds only approximately. Here we make explicit that the error is introduced by approximation (11) in the third step. Thus it is possible to explicitly assess the error and it goes to zero as the number of intervals used for the discretization becomes large.
A description of :
Computing the CSP for can be done via a genealogical recursion (Paul and Song 2010, Equation 12), but has time complexity exponential in the number of loci, k. To improve upon this result, Paul and Song suggest using the genealogical recursion until the first mutation, and thereafter using a fast alternative CSD (Paul and Song 2010, Equation 13). In particular, choosing yields , for which CSP computation has asymptotic time complexity O(_k_3· n).
Similarly, choosing yields , for which CSP computation has the same asymptotic time complexity O(_k_3 · n). Importantly, is more accurate than , and so the resulting CSD is more accurate than .
References
- Abramowitz, M., and I. A. Stegun (Editors), 1972. Handbook of Mathematical Functions With Formulas, Graphs, and Mathematical Tables. Dover, New York.
- Chen, G. K., P. Marjoram and J. D. Wall, 2009. Fast and flexible simulation of DNA sequence data. Genome Res. 19 136–142. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Crawford, D. C., T. Bhangale, N. Li, G. Hellenthal, M. J. Rieder_et al._, 2004. Evidence for substantial fine-scale variation in recombination rates across the human genome. Nat. Genet. 36 700–706. [DOI] [PubMed] [Google Scholar]
- Davison, D., J. K. Pritchard and G. Coop, 2009. An approximate likelihood for genetic data under a model with recombination and population splitting. Theor. Popul. Biol. 75(4): 331–345. [DOI] [PMC free article] [PubMed] [Google Scholar]
- De Iorio, M., and R. C. Griffiths, 2004. a Importance sampling on coalescent histories. I. Adv. Appl. Probab. 36(2): 417–433. [Google Scholar]
- De Iorio, M., and R. C. Griffiths, 2004. b Importance sampling on coalescent histories. II: Subdivided population models. Adv. Appl. Probab. 36(2): 434–454. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Doucet, A., and A. M. Johansen, 2011. A tutorial on particle filtering and smoothing: fifteen years later, in Handbook of Nonlinear Filtering, edited by D. Crisan and B. Rozovsky. Oxford University Press, Oxford (in press).
- Dutheil, J. Y., G. Ganapathy, A. Hobolth, T. Mailund, M. K. Uoyenoyama_et al._, 2009. Ancestral population genomics: the coalescent hidden Markov model approach. Genetics 183 259–274. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Fearnhead, P., and P. Donnelly, 2001. Estimating recombination rates from population genetic data. Genetics 159 1299–1318. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Fearnhead, P., and P. Donnelly, 2002. Approximate likelihood methods for estimating local recombination rates. J. R. Stat. Soc. B 64 657–680. [Google Scholar]
- Fearnhead, P., and N. G. Smith, 2005. A novel method with improved power to detect recombination hotspots from polymorphism data reveals multiple hotspots in human genes. Am. J. Hum. Genet. 77 781–794. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Gay, J., S. R. Myers and G. A. T. McVean, 2007. Estimating meiotic gene conversion rates from population genetic data. Genetics 177 881–894. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Griffiths, R. C., and S. Tavaré, 1994. Sampling theory for neutral alleles in a varying environment. Philos. Trans. R. Soc. Lond. B Biol. Sci. 344 403–410. [DOI] [PubMed] [Google Scholar]
- Griffiths, R. C., P. A. Jenkins and Y. S. Song, 2008. Importance sampling and the two-locus model with subdivided population structure. Adv. Appl. Probab. 40(2): 473–500. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Hellenthal, G., A. Auton and D. Falush, 2008. Inferring human colonization history using a copying model. PLoS Genet. 4(5): e1000078. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Hobolth, A., O. F. Christensen, T. Mailund and M. H. Schierup, 2007. Genomic relationships and speciation times of human, chimpanzee, and gorilla inferred from a coalescent hidden Markov model. PLoS Genet. 3(2): e7. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Howie, B. N., P. Donnelly and J. Marchini, 2009. A flexible and accurate genotype imputation method for the next generation of genome-wide association studies. PLoS Genet. 5(6): e1000529. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Hudson, R. R., 2001. Two-locus sampling distributions and their application. Genetics 159 1805–1817. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Johnson, P. L. F., and M. Slatkin, 2009. Inference of microbial recombination rates from metagenomic data. PLoS Genet. 5(10): e1000674. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Li, N., and M. Stephens, 2003. Modeling linkage disequilibrium, and identifying recombination hotspots using SNP data. Genetics 165 2213–2233. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Li, Y., and G. R. Abecasis, 2006. Mach 1.0: rapid haplotype reconstruction and missing genotype inference. Am. J. Hum. Genet. S79 2290. [Google Scholar]
- Marchini, J., B. Howie, S. R. Myers, G. A. T. McVean and P. Donnelly, 2007. A new multipoint method for genome-wide association studies by imputation of genotypes. Nat. Genet. 39(7): 906–913. [DOI] [PubMed] [Google Scholar]
- Marjoram, P., and J. D. Wall, 2006. Fast “coalescent” simulation. BMC Genet. 7 16. [DOI] [PMC free article] [PubMed] [Google Scholar]
- McVean, G. A. T., and N. J. Cardin, 2005. Approximating the coalescent with recombination. Philos. Trans. R. Soc. Lond. B Biol. Sci. 360 1387–1393. [DOI] [PMC free article] [PubMed] [Google Scholar]
- McVean, G. A. T., S. R. Myers, S. Hunt, P. Deloukas, D. R. Bentley_et al._, 2004. The fine-scale structure of recombination rate variation in the human genome. Science 304 581–584. [DOI] [PubMed] [Google Scholar]
- Paul, J. S., and Y. S. Song, 2010. A principled approach to deriving approximate conditional sampling distributions in population genetics models with recombination. Genetics 186 321–338. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Price, A. L., A. Tandon, N. Patterson, K. C. Barnes, N. Rafaels_et al._, 2009. Sensitive detection of chromosomal segments of distinct ancestry in admixed populations. PLoS Genet. 5(6): e1000519. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Rosenblatt, M., 1959. Functions of a Markov process that are Markovian. J. Math. Mech. 8 585–596. [Google Scholar]
- Scheet, P., and M. Stephens, 2006. A fast and flexible method for large-scale population genotype data: applications to inferring missing genotypes and haplotypic phase. Am. J. Hum. Genet. 78(4): 629–644. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Stephens, M., and P. Donnelly, 2000. Inference in molecular population genetics. J. R. Stat. Soc. Ser. B Stat. Methodol. 62(4): 605–655. [Google Scholar]
- Stephens, M., and P. Scheet, 2005. Accounting for decay of linkage disequilibrium in haplotype inference and missing-data imputation. Am. J. Hum. Genet. 76(3): 449–462. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Wiuf, C., and J. Hein, 1999. Recombination as a point process along sequences. Theor. Popul. Biol. 55 248–259. [DOI] [PubMed] [Google Scholar]
- Yin, J., M. I. Jordan and Y. S. Song, 2009. Joint estimation of gene conversion rates and mean conversion tract lengths from population SNP data. Bioinformatics 25(12): i231–i239. [DOI] [PMC free article] [PubMed] [Google Scholar]