Abstract
We propose a new space of phylogenetic trees which we call wald space. The motivation is to develop a space suitable for statistical analysis of phylogenies, but with a geometry based on more biologically principled assumptions than existing spaces: in wald space, trees are close if they induce similar distributions on genetic sequence data. As a point set, wald space contains the previously developed Billera–Holmes–Vogtmann (BHV) tree space; it also contains disconnected forests, like the edge-product (EP) space but without certain singularities of the EP space. We investigate two related geometries on wald space. The first is the geometry of the Fisher information metric of character distributions induced by the two-state symmetric Markov substitution process on each tree. Infinitesimally, the metric is proportional to the Kullback–Leibler divergence, or equivalently, as we show, to any f-divergence. The second geometry is obtained analogously but using a related continuous-valued Gaussian process on each tree, and it can be viewed as the trace metric of the affine-invariant metric for covariance matrices. We derive a gradient descent algorithm to project from the ambient space of covariance matrices to wald space. For both geometries we derive computational methods to compute geodesics in polynomial time and show numerically that the two information geometries (discrete and continuous) are very similar. In particular, geodesics are approximated extrinsically. Comparison with the BHV geometry shows that our canonical and biologically motivated space is substantially different.
Keywords: Phylogenetic tree, Information geometry, Tree space
Introduction
Evolutionary relationships between species are represented by phylogenetic trees, in which the leaves represent present-day species, internal vertices represent speciation events, and edge lengths represent the degree of evolutionary divergence between species (Semple and Steel 2003). Evolutionary relationships are often subject to a high degree of uncertainty, and so it is natural to consider the space of all possible relationships and probability distributions on this space. Billera et al. (2001) were the first to construct a space of all phylogenetic trees on a fixed set of leaves. This space, known as Billera–Holmes–Vogtmann or BHV tree space, has a very rich geometry: in particular there is a unique geodesic, or shortest length path, between any two points in the space. BHV tree space is a so-called CAT(0) space (Billera et al. 2001), meaning it has globally non-positive curvature, and many of its attractive geometric properties follow from this condition. A polynomial time algorithm for computing geodesics and their lengths was subsequently developed (Owen and Provan 2011). A number of statistical methods for analysing samples of phylogenetic trees have been established, which rely fundamentally on the geometry of BHV tree space by transferring conventional multivariate statistical methods into the new geometrical context. Algorithms have been developed for computing sample means (Bačák 2014; Miller et al. 2015), for constructing confidence regions for the population mean (Willis 2019), and for performing principal component analysis (Nye 2011; Feragen et al. 2013; Nye 2014; Nye et al. 2017). An alternative geometry for phylogenetic trees, known as the tropical tree space (Speyer and Sturmfels 2004; Lin et al. 2018), arises from regarding phylogenetic trees as distance matrices between the species at the leaves. Statistical methods such as calculation of sample means (Lin and Yoshida 2018) and principal component analysis (Yoshida et al. 2019), have also been developed in tropical tree space.
In the BHV and tropical tree spaces, trees are regarded primarily as geometric or algebraic objects, without specific consideration to how phylogenetic trees are estimated or interpreted. Phylogenetic trees are typically inferred from genetic sequence data via Markov models of sequence evolution over the edges of the tree (Yang 2006), and we are only concerned with such trees. Each phylogenetic tree can therefore be regarded as a probability model for genetic sequence data, and a space of all tree-like probability models can be constructed. This idea was first considered by Kim (2000), and then developed more formally in subsequent papers (Moulton and Steel 2004; Gill et al. 2008). The space is known as the phylogenetic orange space or edge-product space. While the space has been studied from the viewpoint of algebraic geometry (Zwiernik and Smith 2012; Engström et al. 2013), metric geometry on the space has received little attention. Recently, methods for approximately computing ‘probabilistic’ metrics on the edge-product space have been developed (Garba et al. 2018). These metrics are defined by mapping each tree to its associated distribution on sequence data, and using a metric between these probability distributions. Specifically, each tree represents a distribution on characters, where a character is a map from the N leaves of the tree to some alphabet of letters . The Hellinger and Jensen–Shannon metrics are defined between distributions on and are pulled back to give metrics between trees. Exact calculation of these metrics involves summation over all possible characters, and so when N is large, Garba et al. (2018) use a simulation procedure to estimate the distance between any pair of trees. The probabilistic metrics have substantially different properties than the BHV and tropical metrics. For example, if all the edge lengths in a pair of given trees are scaled up linearly, then the BHV and tropical distance between the trees both scale in the same way, while in contrast the probabilistic metrics eventually tend to zero. This is because the letters at the leaves of the trees become independent from one another as the edge lengths increase, due to genetic saturation. The distributions on characters represented by the two trees therefore converge to one another as the edge lengths are scaled up, and the distance tends to zero (see Fig. 2 in Garba et al. (2018)).
The metrics studied by Garba et al. (2018) arise from embedding tree space into the larger ‘ambient’ space of all distributions on characters. They are obtained from the lengths of ‘chordal paths’ in the ambient space which do not generally lie within the embedded tree space, and are hence called extrinsic metrics. In contrast, the BHV metric is an intrinsic metric, since it is obtained from the lengths of paths lying within tree space. For trees sharing a common branching pattern the BHV metric agrees with the corresponding extrinsic metric obtained via an embedding into Euclidean space. The statistical methods developed on BHV tree space rely heavily on the intrinsic nature of the metric, and this motivated us to seek intrinsic analogs of the probabilistic metrics.
The aim of this article is to realize intrinsic metrics and their associated geodesics in a new space of forests, the wald space,1 that is related to the edge-product tree space (for the subtle, yet essential differences see the discussion in Sect. 6), when the underlying assumptions are similar to those for the probabilistic metrics. We assume that the infinitesimal distance between two trees is measured using the Fisher information matrix. We show that this is equivalent to assuming the infinitesimal squared distance is the Kullback–Leibler divergence, or equivalently, any f-divergence. Our approach uses ideas from information geometry, which is the study of Riemannian differential geometry on spaces of probability distributions. The purpose of developing this geometry on this space of forests is with the ultimate aim of obtaining statistical methods analogous to those on other tree spaces. The probabilistic metrics and the information geometry have an important advantage over the BHV and tropical geometries: they have by definition a direct biological interpretation in terms of the evolution of genetic sequences. In the information geometry, two trees are close when they determine similar distributions of characters, and as a result they would be potentially indistinguishable if inferred from experimental samples of sequence data. Conversely, trees are distant in the information geometry when they induce substantially different distributions. In contrast, the BHV and tropical metrics are defined more abstractly without reference to evolutionary models or processes. Examples of the biological interpretation of the probabilistic metrics were given by Garba et al. (2018).
Our approach has two main parts. First, we consider geodesics in the information geometry when the model associated with each phylogenetic tree is the two-state symmetric Markov process. This is the simplest discrete Markov model of sequence evolution, for which there are two letters in the alphabet, . This model is introduced in Sect. 2 along with a formal definition of the wald space and a brief review of BHV space. The thesis of Garba (2019) contains some comparisons of results obtained using the two-state model versus models with the DNA alphabet. Geodesics in wald space are constructed locally by numerically integrating a certain differential equation determined by the assumptions on the Riemannian metric. We explore geodesics on the space of unrooted trees with 5 leaves, for which visualization is relatively straightforward, and compare the results with those for BHV tree space. This forms Sect. 3 of the paper. Secondly, in order to improve computational tractability, we consider an alternative continuous-valued model of evolution on each tree. This consists of a Gaussian process which approximates the two-state Markov process by matching its moments. The continuous random variables at the leaves of the tree have a multivariate normal distribution with zero mean, for which the covariance matrix is related to the matrix of path lengths between the leaves. Numerically solving the differential equations for geodesics is much faster under this set of assumptions, and the geodesics closely resemble those for the two-state model. However, solutions are still restricted to trees sharing a common branching pattern, or topology. The definition of the Gaussian process on trees and numerical solution of geodesics in the corresponding information geometry are described in Sect. 4. The information geometry of multivariate normal distributions with zero mean corresponds to a certain geometry on the space of symmetric positive definite matrices, known as the Fisher–Rao or affine-invariant geometry, and the map from the wald space to covariance matrices is an isometric embedding in this space. The geometry on the space of symmetric positive definite matrices is analytically tractable, and geodesics can be computed in polynomial time. The embedding therefore gives intrinsic and extrinsic metrics on the wald space. We describe a projection algorithm from the space of symmetric positive definite matrices into the embedded wald space. We then use this algorithm to project geodesics in the ambient space down into wald space in various ways to obtain approximate geodesics between trees with different topologies. The embedding in the space of symmetric positive definite matrices and the associated geometry is described in Sect. 5. We conclude in Sect. 6 with a detailed discussion of the promises and challenges of our new wald space.
Background and the new wald space
Phylogenetic trees
For we define to be the set of unrooted phylogenetic trees on N taxa. More specifically, a tree T is an element of if it satisfies the following conditions. First, T contains exactly N vertices with degree 1, which are called leaves, and these are bijectively labelled . Secondly, T must contain no vertices with degree 2. Thirdly, each edge e in T is assigned a length with if e contains a leaf. An edge in a tree is called a pendant edge if it contains a leaf; otherwise it is called an internal edge. Similarly, the vertices which are not leaves are called internal vertices.
The edge lengths on any given tree define a path length distance between any pair of leaves. The path length on T between will be denoted .
Each tree contains at most edges, in which case the tree is called fully resolved or bifurcating, and all internal vertices have degree 3. Trees with fewer edges are called unresolved, and for , these contain at least one vertex with degree 4 or more. Trees which contain only the N pendant edges joined at a single degree-N internal vertex are called star trees.
A tree T is rooted when some internal point is labelled as being the root. This is conveniently achieved by adding an additional taxon labelled 0 to the tree via a pendant edge of length zero. It follows that the set of rooted phylogenetic trees satisfies the same conditions as , except the leaves are bijectively labelled , and the pendant edge containing taxon 0 has zero length. We will work with unrooted trees, but our results are easily transferred to the space of rooted trees via this relationship.
Every fully resolved tree will correspond to a fully resolved BHV-tree (reviewed in Sect. 2.2) and to a fully resolved wald, as introduced below in Sect. 2.4. In both BHV tree space and in wald space, unresolved trees will be identified with other trees with certain internal edges having zero length, so that conceptually a missing edge is the same as a zero length edge.
Billera–Holmes–Vogtmann tree space
Billera et al. (2001) defined a space of phylogenetic trees, subsequently known as BHV tree space, and described its geometry. BHV tree space can be described via an embedding in for dimension d which increases exponentially with the number of leaves. However, we have chosen to describe BHV tree space in a way different from the original authors, and we define it as a quotient space. As a result, the wald space introduced in the next section is a superset of BHV tree space when the spaces are regarded simply as sets, clarifying the relationship between the two spaces. Importantly, we allow internal edges on trees to have length zero, and under the quotient these are equivalent to trees with those edges missing. A second difference is that while Billera et al. (2001) worked with rooted trees, we work with unrooted trees. As described in Sect. 2.3, the distribution on binary characters determined by a tree does not depend on the root position under the two-state symmetric model, and so unrooted trees are more natural to work with.
BHV tree space is defined using the notion of splits, where a split is a bipartition of the leaf labels into two disjoint sets. Cutting an edge of a tree induces such a bipartition of the leaves, and so each edge on a tree corresponds to a split, and the terms split and edge can be used interchangeably. The set of splits represented by a tree is called its topology.
Arbitrary sets of splits do not typically determine valid tree topologies: the splits of a tree must satisfy a compatibility condition. For example, the splits and are incompatible, since leaf 1 cannot be grouped next to both 2 and 3 on the same tree. For any topology with k internal edges, , the set of trees in with that topology is bijectively parametrized by where the first term in the product parametrizes the pendant edge lengths that, by definition, are strictly positive, and parametrizes the internal edge lengths.
The set is called the orthant associated with topology , and we identify the set of all trees with topology with . Under this identification, the set of all trees , as defined in Sect. 2.1, is the disjoint union
where the disjoint union is taken over all possible topologies .
The unrooted BHV tree space is obtained by taking the quotient of with respect to an equivalence relation:
Two trees in are equivalent under if and only if they are identical modulo the presence of internal splits with zero length, as shown in Fig. 1.
The quotient space factorizes as
where the first term parametrizes the lengths of the pendant edges and the space parametrizes the topology and internal edge lengths of the BHV-trees. When is fully resolved, is called a maximal orthant. Unresolved trees correspond to points on the boundaries of maximal orthants; they can be obtained from fully resolved trees by shrinking internal edge lengths down to zero.
Since there are fully resolved unrooted topologies, can be thought of as being constructed by gluing this number of maximal orthants together along their boundaries, where two points are identified if they correspond to the same tree. For example, when , there are three fully resolved topologies, each of which contains a single internal edge. The space therefore consists of three copies of glued together at the origin. The origin corresponds to the star trees, while the location along each of the three copies of gives the length of the internal edge in each of the three possible fully resolved topologies. For there are 15 possible unrooted tree topologies, each of which contains two internal edges. It follows that consists of 15 copies of glued along their boundaries. At each codimension-1 boundary, three maximal orthants are joined together. This is because when a single internal edge is contracted to length zero, a degree 4 vertex is obtained, and there are 3 possible ways to add in an edge, including the original edge, in order to obtain a fully resolved topology, as illustrated by Fig. 2.
The metric on is constructed as follows. The basic idea is that for trees with the same fully-resolved topology but different vectors of internal edge lengths, say and , the distance is the Euclidean distance , and the corresponding geodesic is the straight line segment in the orthant containing the trees. Billera et al. (2001) showed that there exists a unique shortest path between any two points in , for which path length is measured using the Euclidean distance in each orthant, and the length of these defines a metric on which we denote . A metric on , denoted , is obtained as the product metric when the metric on pendant edges is taken to be the Euclidean distance. An algorithm has been developed which constructs geodesics and calculates their lengths in time (Owen and Provan 2011).
The two-state symmetric Markov model
Genetic sequence evolution is typically modelled using discrete-valued continuous-time Markov processes defined over the edges of a tree T (Yang 2006; Bryant et al. 2005). DNA sequence evolution is modelled by associating to each point , a random variable X(t) which takes values in an alphabet . In this paper, however, we will consider the two-state symmetric Markov process with alphabet . This simplification is made in order to make the mathematics more tractable and for computational speed. Nonetheless, some of the calculations using the two-state symmetric can readily be performed using DNA models. More details are given in the thesis of Garba (2019) in which simulations show similarity of geometries obtained from the two- and the four-state process. The transition probability of the symmetric two-state model is defined in terms of the path length between any two points :
2.1 |
The stationary distribution of this Markov process is Bern(1/2), and the process is assumed to be in its stationary state over the tree. As a result, for all , X(t) has a marginal Bernoulli distribution, . While the random variables at the leaves of the tree have the same marginal distributions, they are not independent since the tree imposes a dependence structure. The following lemma determines certain moments of the process X(t) giving insight on the dependence structure of . The proof is straightforward using the transition probabilities in Eq. (2.1).
Lemma 2.1
If are the random variables at the leaves of a tree determined by the discrete Markov process defined above, then where is the path length between leaves and .
If are path length apart, then the conditional distribution of given has variance .
It is straightforward to simulate realizations of X(t) in the following way. First simulate a Poisson process with rate 1 independently on each edge of the tree. The positions of the simulated events correspond to points at which X(t) changes parity. Secondly, pick any point which is not a change point and sample from Bern(1/2). The change points generated from the Poisson process then determine the value of X(t) for all other . The distribution obtained is independent of the choice of , because the Markov process is reversible. In particular, the Markov process is independent of the choice of , which could be considered as a root.
Under the model, each edge length can be interpreted as the expected number of change points that occur over the edge. Internal edges are allowed to have length zero, which means that no change in X(t) occurs over the edge. On the other hand, when edges are long, the number of changes is likely to be large, and the letters at either end of the edge are weakly correlated. Biologists refer to this effect as saturation. A fixed change of edge length therefore has more effect on the distribution of characters when applied to a short edge as opposed to a long edge in some given tree. For example, an increase of to an edge with length approximately doubles the probability that the letters at either end of the edge are different, but the same change to an edge of length has almost no effect on this probability, which due to saturation is very close to 1/2. This idea becomes important when we consider defining distances between trees via the information they represent, in particular using the probability mass function of the nontrivial distribution of .
Remark 2.1
The probability mass function of determined by T is denoted where is called a binary character. Given any binary character s, the values of can be evaluated via a recursive algorithm (Bryant et al. 2005), described in Appendix A. Appendix A also contains a modified form of the algorithm which is used to compute exactly the derivatives of with respect to the edge lengths.
Also, as shown in Appendix A, is a multivariate polynomial in and where ranges over the edge lengths in T.
Crucially, the map from fully resolved trees in to the space of probability mass functions on is injective up to the equivalence relation introduced in Sect. 2.2 (Rogers 1997; Allman et al. 2008). This has two implications: first that the probability mass function uniquely characterizes each element of , and secondly that metrics on distributions of characters pull back to define a metric between fully resolved trees, as described in Sect. 3.2.
A new forest space: the wald space
The following wald space gives an alternative viewpoint of phylogenetic trees by regarding them as Markov models for sequence evolution (Kim 2000; Moulton and Steel 2004; Gill et al. 2008). We will give a description of the wald space that is related to the edge-product space of previous authors, by defining it as a quotient space which adds trees with infinitely long edges to BHV tree space. As described in the introduction, each probabilistic metric considered by Garba et al. (2018) converges to zero in the limit as all the edge lengths in a pair of trees simultaneously tend to infinity. This behaviour indicates that shortest paths might cross a tree with infinitely long edges, which is why we add this boundary at infinity. Furthermore, we allow pendant edges of length zero under certain conditions. Thus, in the wald space, edge lengths take values in . It is convenient to reparametrize to the -parametrization by defining weights , so . Under this transformation, becomes a set of unit cubes, rather than orthants, glued along faces where for one or more edges. The wald space is defined by imposing additional gluing rules on faces where .
In order to be able to identify trees with infinitely long edges along faces where , we construct the wald space from forests, that is, disjoint unions of unrooted trees. We start with a preliminary space leading to the definition of the wald space further below. Let be the collection of forests satisfying the following necessary and sufficient conditions for each .
The forest F contains exactly N labelled vertices, these are called leaves and labelled .
There are no unlabelled vertices in F of degree 0, 1 and 2.
For every pair of leaves in the same tree in F, at least one edge e on the unique path from to satisfies .
Clearly, . The condition on the edge weights ensures that no pair of leaves is coincident and consequently that metrics are always well-defined, as described in Sect. 3.2. We impose an equivalence relation on , defined by the following two rules.
BHV boundary rule: Given , suppose all internal edges with are removed from the forests, and the vertices at either end of each such edge are merged. If the resulting forests are identical, then . The rule is the same as that in Fig. 1.
Boundary at infinity rule: Suppose contains an edge with , and that F is modified as follows. The edge with is removed, disconnecting the tree it belongs to. If this results in any unlabelled vertex having degree 2, then those vertices are removed. If v is such a vertex, and the two adjacent edges have weights , then are replaced by a single edge with weight , as is further explained below. Now suppose , and this process of modifying unit-weight edges is applied to both forests. Then if the resultant forests are identical, as illustrated in Fig. 3.
The wald space is defined to be the quotient and it immediately follows that as sets , but the geometry imposed on will be completely different from the geometry of the BHV space.
The boundary rule at infinity requires some explanation. The rule declares that edges of weight (or equivalently length ) can be deleted from a forest F, but unlike the BHV rule for which the vertices at the ends of the edge are merged, edge removal disconnects a tree in F. When resulting degree-2 vertices are removed, the edge length is preserved so that the new edge has length . The corresponding weight is given by . Unlike the BHV boundary rule, in which finitely many trees are identified in each equivalence class, infinitely many combinations of edge weights give rise to the same value . It follows that an uncountable collection of forests can be identified into a single equivalence class in .
In the edge-product space (Moulton and Steel 2004; Gill et al. 2008), an alternative parametrization is used, defining to be the weight of edge e. This parametrization has the advantage that sums of edge lengths become products of edge weights (hence the name ‘edge-product’). The boundary at infinity rule is simpler under this parametrization: the weights in Fig. 3 panel (b) become , and 0 on the left and on the right. However, under the -parametrization, the BHV boundary with lies on faces of cubes with , whereas the boundary at infinity has . We prefer to work with the -parametrization since it gives a more intuitive interpretation of the weights, i.e. corresponds to and corresponds to . Forests which contain more than one connected component lie in the faces of cubes with at least one . Since the pendant edges can be expanded out to infinite length, they are also subject to the boundary at infinity rule, and so the representation of pendant edge lengths in is not via a product geometry, as it is for BHV tree space. While the star trees correspond to all internal edges having zero length, also contains a point which consists of N isolated vertices.
The BHV boundaries enable tree topologies to be changed via nearest neighbour interchange (NNI) operations (as illustrated by Fig. 2). The boundary at infinity corresponds to a different topological operation, called tree bisection and reconnection (TBR) (Allen and Steel 2001). Under this operation, an edge e in a tree can be expanded up to the boundary . Removing the edge bisects the tree, and the two components can be reconnected by an edge with placed arbitrarily between the two trees. Reducing the weight down from 1 then gives a tree with a topology different from the original tree. It follows that there exist continuous paths in the wald space between trees with different topologies, which pass through the boundary at infinity and, as a result, change tree topology via TBR operations. This is in contrast to BHV tree space in which paths between trees of different topologies involve only NNI operations, as edges are contracted down to length zero and alternative edges are expanded out.
While the set was defined above via an equivalence relation on forests, we also need to understand how it parametrizes Markov models and then characterise its elements again as probability mass functions on . The two-state symmetric Markov process extends from being defined on trees to forests by taking the process on each connected component in a forest to be independent of the other components. This defines a distribution on for each . In fact the distribution uniquely determines the equivalence class of F, and vice versa, as the following lemma shows.
Lemma 2.2
Given , then if and only if for all s.
A proof is given in the Appendix.
Note that the forest consisting of N isolated vertices corresponds to the random variables being independent, and this can be obtained from any tree by expanding all edges one after the other, as, by definition, there is at least one edge between any two leaves.
Information geometry for the two-state symmetric model
Information geometry provides methods for constructing metrics and geodesics on parametrized sets of probability distributions. In this section we embed wald space in the space of distributions of two-state characters, and investigate the corresponding information geometry analytically and computationally.
Geometry of embeddings
Suppose that where (Y, d) is a metric space and is injective. We will say that X is embedded in Y, and refer to Y as the ambient space. The embedding can be used to construct certain metrics on X. First, since is injective, d pulls back to define a metric on X which we denote :
for all . The pullback metric is often called the induced extrinsic metric and, it is simply the restriction of d to , and so when the context is clear, it is also denoted d. The probabilistic metrics described in Sect. 3.2 are constructed in this way. A second metric, called the induced intrinsic metric and denoted , is defined as the infimum of the length of all possible paths in X between when path length is measured using the metric d. If no path with finite length exists between and then , in which case is not a metric. Details of this construction of the induced intrinsic metric are given by Bridson and Haefliger (2011). The metrics d and , if well-defined, give X the structure of a length space, which is a space in which the metric between points is the infimum of the lengths of paths between those points. Length spaces are similar to geodesic metric spaces, except that the infimum is not necessarily achieved by a path lying within the space; in a geodesic metric space a minimum length path exists between every pair of points, and so every geodesic metric space is a length space. An example of a length space which is not a geodesic metric space is with the origin removed and the Euclidean metric. Points antipodal to the origin cannot be joined by a geodesic, but the distance between them is the infimum of the lengths of paths joining the points.
In order to illustrate the relationship between d and , consider the example of the embedding of the unit sphere in equipped with the Euclidean metric d. For any two points , is the length of the straight line segment in the ambient space joining the points. This metric is usually called the chordal metric on . However, when we consider paths between which are restricted to lie in , the shortest paths (with respect to d) are great circles, and the induced metric is the arc length metric. In fact, is a geodesic metric space, since the infimum of path length is always achieved by a great circle.
In the following Sect. 3.2, the wald space will be embedded in the space of distributions of characters. Later in Sect. 4 it will be embedded in the space of symmetric positive definite matrices. Each embedding will be used to construct metrics on .
Probabilistic metrics
Here we briefly describe the probabilistic metrics developed by Garba et al. (2018) since these will be used for comparison with other metrics. The Kullback-Leibler divergence is a commonly used measure of the difference between two distributions. Given two probability mass functions p, q on characters , the Kullback–Leibler divergence from q to p is defined as
provided only when . The Kullback–Leibler divergence is not a metric since it is not symmetric. However, metrics can be defined as follows: the Jensen–Shannon metric is defined by
and the Hellinger metric is defined by
Recently, probabilistic metrics have been developed which are based on distributions of gene trees instead of distributions of characters (Adams and Castoe 2020).
The Kullback–Leibler divergence, squared Jensen–Shannon metric and squared Hellinger metric are all examples of a more general class of distances between probability distributions known as f-divergences. Given any convex function f(t) such that , the f-divergence of p from q is defined as
3.1 |
The Kullback–Leibler divergence is obtained by taking , while the reversed divergence is obtained with . The squared Jensen–Shannon metric and squared Hellinger metric can also be obtained by using more complicated functions f, cf. Sason and Verdu (2016).
Now, let be a forest representative of a wald . As described in Sect. 2.4, the distributions at leaves of different trees of F are independent. For two leaves in the same tree in F, some degree of evolution occurs between them since by definition of no two leaves are coincident. Therefore, all characters are possible, giving
3.2 |
It follows that the Kullback–Leibler divergence is always well-defined between distributions of characters corresponding to forest representatives from the wald space. Since by Lemma 2.2 the map is injective for the Jensen–Shannon and Hellinger metrics pull back to define extrinsic metrics on the wald space (analogously to (Garba et al. 2018)). As already mentioned in the introduction, statistical methods rely heavily on the intrinsic nature of metrics, and thus we aim for more geometrical structure in the next section by imposing the Fisher information metric (a Riemannian metric) onto the wald space.
A two-state process geometry for the wald space
BHV tree space and wald space both do not have the structure of a manifold globally, but the interior of each maximal orthant is a manifold parametrized by or . Therefore we consider first the information geometry on the subspaces of wald space corresponding to a fixed fully resolved tree topology—here every wald has only one single tree representative, since every wald corresponding to a forest with more than one component, as well as a wald containing a pendant edge with length zero, lies on the boundary of unit cubes corresponding to fully resolved tree topologies. Secondly, we establish global results about the constructed geometry of .
Thus suppose is a fully resolved tree topology, and that trees with this topology are parametrized by .
Let be the probability mass function associated with tree T determined by . Recalling that for all s, due to (3.2), the Fisher information matrix at is
3.3 |
for where . This defines a Riemannian inner product on the tangent space of at (that is a copy of ) which gives a way to measure the lengths of paths. Specifically, if and lie infinitesimally close on a path, then the squared path length between them is defined to be . Standard results from Riemannian differential geometry show that if is a path in then it is locally a geodesic (i.e. it minimizes path length) if it satisfies the differential equation
3.4 |
where are the Christoffel symbols
The matrix is the inverse of i.e. where is the Kronecker delta. It is important to note that the geodesic equation and loci of solutions are invariant under changes of parametrization, and so the equations can be formulated using lengths or the weights . On the boundary, however, this is no longer necessarily true.
The Riemannian metric defined by the Fisher information matrix is related to the Kullback–Leibler divergence and other f-divergences by the following lemma.
Lemma 3.1
Suppose is an f-divergence given by some convex function f with , as defined by (3.1). Consider a small perturbation of the edge lengths of a tree . Then
3.5 |
where the error term consists of third-order products of the elements of and
3.6 |
In other words, the norm of the perturbation, as measured with respect to the Riemannian inner product, is proportional to the f-divergence of from .
The proof is given in the Appendix. Since the lemma applies to an arbitrary f-divergence, the term on the right-hand side of Eq. (3.5) can be the Kullback–Leibler divergence or the squared Jensen–Shannon metric, for example. Lemma 3.1 gives the fundamental assumption behind the geometries we construct on : that distances are locally measured by the infinitesimal Kullback–Leibler divergence between probability distributions associated with trees, or equivalently, by any f-divergence. As a corollary of Lemma 3.1, it follows that the metric defined by Eq. (3.3) is positive definite (i.e. the metric is not semi-Riemannian). This is because the map from trees to distributions of characters is injective, and since for all , it follows that the right-hand side of Eq. (3.5) is strictly positive for all small non-zero perturbations .
Let be the space of distributions on . By Lemma 2.2, the map from to distributions of characters determines an embedding of in . Given a metric d on , let be the induced intrinsic metric on , as described in Sect. 3.1.
Theorem 3.1
Let d and be metrics on which are the square root of an f- and -divergence, respectively. Then for any :
and thus is well-defined.
for some constant .
Any path which realizes the distance is a solution of Eq. (3.4) at any point in the interior of a maximal orthant.
The proof is given in the Appendix. Valid choices for the metric d in Theorem 3.1 include the Jensen–Shannon metric or Hellinger metric. Theorem 3.1 establishes with metric as a length space. Finiteness of shows, for example, that points in corresponding to disconnected forests (or equivalently, trees with infinite edge lengths) are at a finite distance away from orthant interiors. The second assertion implies a scaling of the induced intrinsic metric under changes of the function f, which, in turn, substantiates our conclusions drawn from Lemma 3.1 that the geometry of induced by d is invariant under the choice of f.
Numerical investigation of the geometry
The geodesic equation (3.4) can be solved numerically on the interior of any maximal orthant given some initial conditions and . As described in Sect. 2.3, the first and second derivatives of with respect to the edge lengths can be computed analytically. Calculation of consists of a sum over all possible characters involving the derivatives of . The Christoffel symbols can similarly be calculated as sums over characters. A fourth-order Runge–Kutta method was used to integrate the ODEs.
This numerical scheme was used to construct and visualize geodesics on a single orthant in . The particular topology and edge lengths for the orthant are represented by the Newick string . Parameters are the pendant edge lengths and the lengths of the two internal edges. We restricted to leaves in order to enable easy visualization of geodesics. Integration was stopped whenever any internal edge was assigned a length , corresponding to the boundary of the orthant. If this occurred for a pendant edge, the pendant edge length was given value zero at that step, and integration was continued.
Figure 4 shows typical results. The figure shows the orthant representing the two internal edge lengths, with geodesics ‘fired’ from some fixed starting point in 24 different directions. Also marked on the plots are contours of distance from the starting point. Each panel shows results for a different initial tree . It is evident that the geodesics are not the same as BHV geodesics, which are straight lines radiating from the initial point with equally spaced circular contours of distance. Figure 4 shows curved geodesics with irregularly spaced contours of distance. Contours appear to stack up towards the origin and codimension-1 BHV boundaries, but are more spaced out as geodesics move out towards the boundaries at infinity. This is more obvious when initial internal edges are long (top row of Fig. 4). On the other hand, the geodesics are more similar to the BHV geodesics when internal edges are short and pendant edges are long, as in the bottom two rows of the right hand column. In all cases, in contrast to BHV tree space, geodesics seem to be slightly attracted toward the star trees. The pendant edges do not behave as they do in BHV tree space: they can change value even when their initial velocity is zero.
Figure 5 provides more detail for certain geodesics in Fig. 4. The graphs in the figure show each edge weight versus time, and the time is proportional to distance travelled. The -parametrization was used for these plots since it shows the results most clearly. Panel (a) shows that, when contours become more and more widely spaced, the boundary at infinity () can be reached in finite time, rather than asymptotically. This shows that points corresponding to trees with infinitely long edges are a finite distance away from the starting point, as established by Theorem 3.1. In the -parametrization, the internal edge lengths rapidly blow up to infinity after time . It follows that the shortest path between two trees with finite edge lengths might involve trees with infinitely long edges. On the other hand, for some panels in Fig. 4, the contours become increasingly close as BHV boundaries are approached, but as panel (b) in Fig. 5 shows, points on BHV boundaries are in fact finitely close to orthant interiors, since the boundary is reached in finite time.
Figure 6 replots the top left and middle left panels from Fig. 4 using the -parametrization, so that the boundary at infinity corresponds to edges of the unit square with weight 1. These plots suggest that trees in which one of the two internal edges is finitely long are ‘repellant’ since the geodesics fired in the North and East compass directions end up passing through the disconnected forest with . Indeed, the points on the boundary with appear to be ‘attractive’, with geodesics pulled round to pass through these points and arriving in finite time.
While these results show how the information geometry on differs substantially from the BHV geometry, the method for constructing geodesics by integrating the geodesic differential equation suffers from several disadvantages. First, it requires summation over the elements of which makes it infeasible for large N (exponential computation time in N). Secondly, only local geodesics are computed by solving the initial value problem (geodesic “shooting” or “firing”) for the differential equation valid only in maximal orthants. Thirdly, for practical applications, an algorithm which takes two points in wald space and joins them by a globally shortest path is more useful, and so it is desirable to solve the boundary value problem for geodesic construction. The next section attempts to deal with some of these shortcomings.
Information geometry for a Gaussian process on trees
In this section we develop the information geometry of a continuous-valued Markov process associated to each tree which is more computationally and analytically tractable than the information geometry for the symmetric two-state Markov process. It has the advantage that the geodesic equation (3.4) can be solved numerically much faster than the corresponding equation for the symmetric two-state model, but the solutions for the two models are very similar.
Definition of the Gaussian process
Our aim is to construct a Gaussian process which is a continuous-valued analogue of the symmetric two-state process by matching the moments specified in Lemma 2.1. Consider the Ornstein–Uhlenbeck process Z(t) on T which satisfies
4.1 |
where is the path length distance between and on T. The stationary distribution is the standard normal distribution N(0, 1), and we assume the process is stationary over T. The Markov process satisfies the detailed balance equation, and so is reversible in its stationary state. As a result, realizations of the process can be simulated by fixing an arbitrary root for the tree, simulating from N(0, 1) and then using Eq. (4.1) to simulate a realization Z(t) for any other . Reversibility of the process ensures the distribution obtained is invariant of the choice of root. A short calculation shows that the covariance matrix of the random variables at the leaves of T is given by . Since for all , this gives . Similarly, the conditional distribution of given has variance . These moments match those in Lemma 2.1 up to the constant factor of 1/4, and we will show later that this factor makes no difference to the geometry obtained. The process Z(t) can be thought of in two ways. First, it approximates the binomial random variables obtained when many independent binary characters evolve under the symmetric two-state process. Secondly, it could be regarded as an evolutionary model of a continuous trait for which the observations are standardized to have zero mean and unit variance in each population. Mean-reverting Gaussian processes like this are sometimes used to model continuous traits for which there is some constant optimal value for survival (Hansen and Martins 1996). The definition of Z(t) extends from trees to forests by taking the process to be independent on each connected component.
Given a forest , the distribution of the random variables at the leaves is multivariate normal with zero mean and covariance matrix where
4.2 |
The path distance is the sum of the lengths on edges e between and , and is taken to be infinite when and are in different components of F. This defines a map from forests to multivariate normal distributions where is the probability density function of . A similar result to Lemma 2.2 holds: whenever , the distributions and are the same, so the map is well-defined on .
A Gaussian process geometry for the wald space
The information geometry of multivariate normal distributions with zero mean has been studied previously (Lenglet et al. 2006) and is analytically tractable. The theory described in Sect. 3 needs adapting to account for the change from discrete to continuous characters. The Fisher information metric of in Eq. (3.3) becomes an integral over rather than a sum, and the mass function is replaced with the density function of the Gaussian corresponding to a fully resolved tree with edge lengths :
4.3 |
The geodesic equation (3.4) remains the same. Although and its derivatives could be evaluated exactly for the two-state symmetric model, evaluation of the metric and Christoffel symbols required a sum over all characters. For the continuous model, the corresponding integrals have closed form, as we describe below.
Gaussian distributions with zero mean are parametrized by their covariance matrices, namely by the space of symmetric positive definite matrices, which we will denote . The set of covariance matrices associated with forests forms a subset within , as determined by the following theorem.
Theorem 4.1
The covariance matrix associated to any , as defined by Eq. (4.2), is positive definite so lies in , and
the map from to is injective and so it determines a well-defined embedding of into .
The proof is given in the Appendix.
For Gaussians with zero mean, it can be shown that the Fisher information metric at a point with covariance matrix S is
where X, Y are matrices in the tangent space at S, i.e. symmetric matrices (Lenglet et al. 2006). This expression is obtained by evaluating the integral in Eq. (4.3). Working in some fixed maximal orthant parametrized by edge lengths , let be the corresponding covariance matrix defined in Eq. (4.2). For each edge define the split matrix by
4.4 |
for . Then the path length between and is where the sum is over all edges in F. Equation (4.2) becomes
4.5 |
An entry above is zero if and are in different connected components, or equivalently, if they are separated by an infinitely long edge.
By differentiating Eq. (4.5), it can be seen that the tangent space at is spanned by matrices of the form for each edge e, where denotes the Hadamard matrix product. The Fisher information metric (4.3) for for a fully resolved tree becomes
4.6 |
where index edges. Algebraic expressions for the first and second derivatives of the Fisher information metric can similarly be obtained, and hence for the Christoffel symbols. Note that scaling by some positive constant has no effect on the metric, and so the factor of 1/4 difference between the covariance matrices obtained from the discrete process X(t) and continuous process Z(t) has no effect on the geometry.
The inner product and its derivatives can be computed in polynomial time and the paths obtained by integrating the geodesic ODE for the continuous Markov model within orthant interiors in resemble those for the two-state model very closely. Namely, results for the same initial conditions as Fig. 4 were obtained but omitted, since the plots were almost indistinguishable from those for the two-state model. However, the inner products defined using the two different models are not identical: both inner products can be written down explicitly in the case , using the transition probabilities for the discrete model and Eq. (4.6) for the continuous model. The two inner products differ when the length of the single edge in the tree is small, but converge as the edge length tends to infinity.
Using Eq. (4.6) and its derivatives, we derived algebraic expressions for the Riemannian curvature tensor and the sectional curvatures at any point in . We implemented these expressions in R, and hence evaluated these quantities for certain trees in . We found that at randomly selected points in , and hence in all spaces with , the sectional curvatures had mixed signs. As a result, there is no global sign condition on curvature like that for BHV tree space, which is globally non-positively curved.
Geometry via embedding in
The information geometry on Gaussians with zero mean can equivalently be regarded as a geometry for the space of symmetric positive definite matrices . This is a useful viewpoint to adopt, first because it highlights the fact that the geometry we develop on is based entirely on the covariance between the leaves induced by the Markov process Z(t), and secondly, because other metrics on have been studied (Dryden et al. 2009). These alternative metrics on could in turn define different metrics on , although they will not be considered any further in this paper. The metric on obtained from the information geometry of Gaussian distributions with zero mean will be denoted . The metric and its associated geodesics in can be computed in polynomial time (Lenglet et al. 2006). The main idea in this section is to use the analytically tractable geometry in , combined with a projection algorithm from the ambient space to the embedded space , to construct approximate geodesics in .
Given the embedding of within , we can consider the intrinsic metric on induced by . By construction, the induced metric corresponds to the information geometry on for the continuous Markov model considered in Sect. 4. The following theorem is analogous to Theorem 3.1 for the discrete Markov substitution models. A proof is given in the appendix.
Theorem 5.1
For any the induced intrinsic metric is finite and therefore well-defined. Any path which realizes the distance is a solution of Eq. (3.4) at any point in the interior of a maximal orthant, where the Riemannian inner product is given by Eq. (4.5).
Lenglet et al. (2006) give formulae for the distance and geodesics between pairs of points in . The distance between is defined by
5.1 |
where denotes the matrix logarithm.
Since the map from to is injective, pulls back to define an extrinsic metric on :
5.2 |
In fact, the space equipped with has globally non-positive curvature (Skovgaard 1984; Ballmann et al. 1985) and so there is a unique geodesic between any two points in the ambient space. The point at proportion along the geodesic between is
5.3 |
Equations (5.1) to (5.3) involve eigen-decompositions of matrices, and so can be computed in steps.
Figure 7 shows a comparison of BHV metric , tropical metric, path distance metric, Jensen–Shannon metric and for every pair of trees in a sample of 100 trees obtained by bootstrap replication during maximum likelihood inference of a phylogenetic tree. The trees were inferred using the MrBayes software (Huelsenbeck and Ronquist 2001), and a sample data set of DNA from 12 primates provided with the software. The path difference metric (Steel and Penny 1993) between trees is . The Jensen–Shannon metric was calculated exactly by summing over all characters for the two-state model, as described in Sect. 3.2. The covariance metric was calculated using Eqs. (5.1) and (5.2). The BHV and tropical metrics are quite closely correlated, as are the Jenson–Shannon and covariance matrices. The path difference metric has a similar correlation with the BHV, tropical and Jensen–Shannon metrics (approximately 0.7–0.8), but is relatively weakly correlated with . This suggests that the BHV and tropical metrics are based on features of the data which are rather distinct from those for the Jensen–Shannon and covariance metrics. The covariance metric has the advantage over the Jensen–Shannon metric of being computable in polynomial time.
Projection into wald space
To approximate geodesics in the extrinsic covariance metric of , we construct a projection from onto , that is, given , we aim to find an element which minimizes . Suppose that F is a fully resolved tree with edge lengths . The expression for can be differentiated with respect to edge lengths of F and gives
where (e.g. Moakher (2005)). Moreover, where denotes the Hadamard or element-wise matrix product and is the split matrix associated with edge i, as defined in Eq. (4.4).
This analytic expression for the derivative can be used to implement a gradient descent algorithm. Within each maximal orthant the edge lengths were updated according to the rule
where denotes the edge lengths at iteration k and the corresponding covariance matrix. The step size was determined using the Barzilai–Borwein method. Two versions of the algorithm were used. In the first, the algorithm was halted whenever an internal edge was assigned a negative length. As a result, the algorithm was constrained to lie within the orthant containing the initial tree. This algorithm was used for by running the algorithm 15 times, each time with an initial tree in one of the 15 maximal orthants of , and the overall tree closest to found. The second version of the algorithm was able to cross codimension-1 BHV boundaries as follows. If an edge length was assigned a negative value, then trees in the two corresponding neighbouring orthants to were considered, taking the absolute value of elements in as edge lengths. The tree at step was taken to be whichever of these two trees was closest to .
In general, the closest point in to a covariance matrix is not necessarily unique as illustrated in Fig. 8. Moreover, the gradient descent algorithm can converge to local minima, and so the result obtained is sensitive to the tree used to initialize the algorithm.
Construction of geodesics in via projection of extrinsic geodesics
Since construction of geodesics in between any two given points and projection from into can both be performed efficiently, we aim to combine these algorithms to give an efficient means of constructing geodesics within between any two given end points. A naive approach, given , is to simply project the extrinsic geodesic between and from into . This approach works for the example of the unit sphere embedded within : the projection of the chord between two points in the sphere is a great circle between those two points. However, this approach fails for since the projected paths are often discontinuous and jump between different orthants, as illustrated in Fig. 8.
The following recursive algorithm for constructing an approximate geo-desic in is therefore proposed, which is intended to overcome this issue. Let for and suppose . The algorithm outputs a sequence where and . For each iteration of the algorithm, the following steps are performed.
Compute the extrinsic geodesic from to using Eq. (5.3).
Find the point at proportion along .
Let be the projection of S into .
The idea is that at each iteration, a new extrinsic geodesic is constructed from the previous point to the destination , a small step is taken along that geodesic, and that point is projected into to give . For the results in this paper, the projection at Step 3 was performed using the second version of the projection algorithm described in Sect. 5.1, rather than using the less efficient search over all orthants. The gradient descent for the projection to obtain at Step 3 was initialized using the edge lengths from the forest .
This algorithm has the disadvantage that it is not symmetric under swapping the end points , whereas the image of the geodesic should be invariant under this operation.
The following symmetrized version of the algorithm overcomes this issue.
The algorithm produces a sequence where the initial values are taken to be and . For each iteration of the algorithm, the following steps are performed.
Compute the extrinsic geodesic from to using Eq. (5.3).
Find the points at proportions and along .
Let and be the projections of R and S into respectively.
The quality of the approximate geodesics produced by the symmetrized algorithm can be assessed by comparing them to geodesics in a single orthant constructed by integrating the geodesic equation as described in Sect. 4.2. Given an initial tree and an initial velocity for , the geodesic equation was integrated until the path obtained reached some specified length. The final point reached was taken to be , and the symmetrized algorithm was then used to obtain an approximate geodesic between and . In all cases, the paths obtained with the two methods matched very closely, with the quality of the match improving for shorter internal edge lengths. Figure 9 shows typical results.
In contrast to the methods presented in Sects. 3 and 4, these algorithms are not based on ‘firing’ geodesics, and can produce approximate geodesics between end points in different orthants. Figures 10 and 11 show results obtained when the end points are separated by 1 or 2 nearest neighbour interchange operations respectively in . More precisely, we consider trees corresponding to Newick strings
5.4 |
In both figures, the approximate geodesics constructed using the recursive algorithm are not symmetric under interchange of the end points, and differ from the paths obtained using the symmetric algorithm. The lengths of the paths can be computed by using large k and summing between successive points in the output. In all the examples we explored, the symmetrized algorithm produced shorter paths. The approximate geodesics shown in the figures are significantly different from BHV geodesics, which consist of straight (or once broken) lines between the end points in both plots.
We apply the symmetrized algorithm to investigate the distance from trees in the interior of a maximal orthant to the star stratum on the boundary of that orthant. If is an approximated geodesic between and computed by the symmetrized algorithm, the intrinsic distance between and can be approximated by . Consider the following setup. For , let be the forest corresponding to the Newick string in -parametrization. This is a fully resolved 4-taxon tree on which each edge has weight . By symmetry, the edges on the tree in the star stratum closest to must all have equal weight . Thus, let be the star tree corresponding to the Newick string , again in -parametrization. Figure 12 shows for each the approximated values of against . Obviously, is closest for slightly larger than with distance decreasing as . This suggests that the star stratum is closer to the tree than the forest consisting of 4 isolated points (obtained from as ), even for values close to 1. Note, though, that the forest is a boundary point of the star stratum. For any the distance to tends to infinity as . Indeed, is not of full rank.
Discussion
In order to do statistical inference on data sets of phylogenetic trees one needs a structure rich enough to enable the use of geometric statistical methods. Recent research has produced geometries such as the BHV and tropical tree spaces and statistical methods adapted to these geometries. Based on a more principled set of underlying assumptions by regarding phylogenetic trees as probability models for genetic sequence data, we have developed a canonical and biologically motivated geometry on tree space by applying tools from information geometry, giving the wald space. In particular, unlike previous related work (Garba et al. 2018) in which various extrinsic metrics were considered, in this paper we have focused on developing intrinsic metrics and their associated geodesics to explore and to enable accessing the geometry, for this is a key ingredient for statistical inference on non-Euclidean spaces.
There are two main difficulties with achieving our aim. First, the discrete-valued Markov process on trees with genetic alphabet characterizes trees as probability models with sample space where N is the number of phylogenetic taxa. Therefore, calculations of distances and construction of geodesics involve summations over terms, which becomes infeasible for large N. In order to establish computational tractability, we generalized the discrete-valued probability model to a continuous-valued Markov process in a canonical way and applied the information geometry again.
Secondly, information geometry is formulated for parametrized probability models that are a manifold, whereas tree space is a union of manifolds having different dimensions due to the orthants (representing forests with different number of edges) being glued together in a certain way. One has to be careful to compare the structure to the one defined in Moulton and Steel (2004), for example, as we are not including forests with coincident leaves and furthermore we consider a different topology induced by the Fisher information metric. We tackled this issue of not having a single connected parametrized manifold by using the continuous-valued Markov models to embed wald space in the ambient space of symmetric positive definite matrices, which has an analytically tractable geometry and thus allows for approximation of geodesics in the embedded space . Our computational results show that the geometry obtained is significantly different from the BHV and tropical geometries, partly due to the inclusion of trees with infinitely long edges in wald space.
Several questions about the geometry of the wald space remain. While we have shown that trees with infinitely long edges are a finite distance away from other trees (Theorems 3.1 and 5.1), computational results suggest that parts of this subspace are repulsive and are avoided by geodesics (see Figs. 6 and 12). An explanation for this behaviour might be obtained via calculations or results about curvature for such points of , but further investigation is required. Secondly, Theorems 3.1 and 5.1 establish as length spaces for the two induced intrinsic metrics we study. It is desirable to strengthen these results and prove that the distance between every pair of points in the space is realized by at least one path, so that our wald space is a geodesic metric space as opposed to a length space. It appears that such a proof requires thorough analysis of the condition on edge weights which excludes trees with coincident leaves. Furthermore, the methods and results presented in Sect. 5.2 represent a first step towards the development of more sophisticated and efficient algorithms for the construction of information geodesics in wald space via the embedding in the space of covariance matrices. A more thorough evaluation of the computational cost as N increases could be carried out, and a more rigorous treatment might establish convergence properties for the symmetrized algorithm. Alternatively, existing algorithms taken from computational Riemannian geometry could be adapted to work in wald space (see Schmidt et al. (2006) for example) and might offer better performance.
The underlying motivation for this work has been to obtain a novel geometric framework for the space of phylogenetic trees which has more principled biological justification than existing geometries, and which can be used to develop statistical methods for analysing data sets of trees. Ultimately, realizing this aim is still some way off. For example, given a sample of points in a metric space (X, d), the Fréchet mean is a point which minimises the sum of squared distances to the data:
In general, the Fréchet mean does not always exist, or it can fail to be unique, but in globally non-positively curved spaces such as and there exists a unique Fréchet mean (Bridson and Haefliger 2011). Development of methods for calculation of a Fréchet mean using an intrinsic information metric in wald space seems very challenging, and the curvature calculations in Sect. 4.2 have implications for the existence and uniqueness of Fréchet means. On the other hand, given any sample of trees in , there is a unique inrinsic Fréchet mean in and an algorithm for computing the mean is given by Lenglet et al. (2006). Our projection algorithm could be used to project this to an extrinsic mean back into . Properties of the projected Fréchet mean tree could be investigated.
In comparison to the BHV and tropical metrics, the intrinsic information metrics have the advantage of interpretability in terms of genetic substitutions and the distributions of characters represented by two trees. This suggests the information metrics might be better suited for statistical tasks such as hypothesis testing. In the BHV and tropical geometries, contraction and expansion of edges offer the means of moving between different topologies. In the wald space, additional topological transformations are possible via expanding edges to infinite length, and these correspond to tree bisection and reconnection (TBR) operations. Many applications in phylogenetics require searches over the space of phylogenetic trees, and movement along information geodesics in the wald space might have advantages over existing methods.
Acknowledgements
The second and the last author express their thanks to the Oberwolfach 1804 meeting “Statistics for Data with Geometric Structure” in which wald space was first discussed. The last two authors gratefully acknowledge support from DFG GRK 2088. The last author was supported by the Niedersachsen Vorab of the Volkswagen Foundation.
Appendix A: Calculation of and its derivatives
The probability of any binary character s on a tree can be computed efficiently via the following algorithm (Semple and Steel 2003), often called the Felsenstein pruning algorithm. First, an arbitrary internal vertex is chosen and used to root the tree. The two-state symmetric model is a reversible Markov process, and so the choice of the root does not affect the value of . The root determines ancestral relations on the tree, and we let denote the subtree of T descended from vertex v. We let denote the leaves of , and given a binary characer s, let denote the restriction of s to . Finally, we let be the probability of on given the letter at v:
since for all . The theorem of total probability gives
6.1 |
For an interior vertex v, if we let , be the vertices immediately descended from v via edges of length , then the transition probabilities in Eq. (2.1) give
6.2 |
where . For a leaf u, we have if and zero otherwise. This and Equation (6.2) can be applied recursively to compute the terms for each vertex , starting at the leaves and working up the tree to the root . Finally, can be computed using Equation (6.1), and it follows from the recursion that is a multivariate polynomial with arguments of the form and , where k ranges over the edges of T. The coefficients of the polynomial depend on the topology of T.
Equations (6.1) and (6.2) can also be used to differentiate with respect to any edge length parameter. These derivatives are required in Sect. 3. Suppose e is an edge of T and we wish to compute the derivative . Since we are free to choose , the calculation is simplified if we let be an internal vertex at one end of edge e. We can order the vertices attached to so that . Equation (6.2) then gives
where the terms can be calculated recursively. Second derivatives of the mass function can be calculated analytically in a similar way.
Appendix B: Proof of Lemma 2.2
First, suppose . The BHV boundary rule does not affect the distribution on characters induced by a tree, because the same distribution is obtained whether an edge of length zero is present in a tree or not. Similarly, if are equal modulo an application of the boundary rule at infinity, then since an edge with weight 1 results in independence between the letters at leaves at either side of the edge under the transition probabilities in Eq. (2.1). Specifically, if are vertices at the ends of an edge e with , then
where , so the conditional distribution of is the same as its marginal. The map from elements of to distributions on characters is therefore well-defined. In fact, the work of Allman et al. (2008) shows the map is injective, and this establishes the lemma.
Appendix C: Proof of Lemma 3.1
The Riemannian metric in Eq. (3.3) can be expanded as
The Taylor expansion of is
Substituting this into the expression for the Riemannian metric gives
since is . On the other hand, a Taylor expansion of f around 1 gives
The first three terms vanish since and since for all . It follows that
and the lemma is established.
Appendix D: Proof of Theorem 3.1
We need to show that the induced intrinsic metric is finite for any . Start by choosing the representative for the equivalence class to be a connected tree with edge weights , some elements of which might have value 1. The tree can be continuously deformed within the orthant corresponding to its topology, to the star tree on which all pendant edges have weight , by changing along the obvious linear path. If the path has finite length, then the first part of the theorem has been established, since any and can be joined to in this way. As shown in Remark 2.1, each for along the path from to is a polynomial in . It follows that if and represent the edge weights at two nearby points on the path then
where for each character s and , is a polynomial in . Then
The path distance between any pair of leaves is continuous along this path, and is strictly positive since the pendant edge lengths are non-zero at all points on the path, apart from potentially at . Pendant edge lengths can be zero on , but by the definition of , the path distance between leaves is non-zero. It follows that is also bound away from zero. Thus there is a constant C(s) such that
where the second line comes from the Cauchy-Schwarz inequality and the norm is the Euclidean norm. Since the are polynomials in and the elements of lie between 0 and 1, the are bounded from above and we obtain
for some constant B(s). Now suppose that is a f-divergence, where is a metric. Applying Eq. (3.6) from Lemma 3.1 with the -parametrization gives
6.3 |
for some constant K. Thus the infinitesimal path length in as measured by the metric d is bounded by some multiple of the Euclidean path length on . The length of the linear path from to measured with d is therefore finite, since the Euclidean length of this path is finite, and hence is finite.
For the second part of the theorem, suppose that is a -divergence, where is a metric. Further, suppose that and are given by and , respectively. It suffices to consider from the same topology and sufficiently close such that the image , and of the geodesic from to in the metric induced by the Riemannian metric lies fully in a convex coordinate patch and has finite Euclidean length there, say L. Hence, for every , there are with , such that . Then the second assertion of the theorem follows from (6.3), setting , as , because
The second equality sign holds because the constants in the individual summands () can be bounded by the supremum of absolute values of the gradient of with respect to , in the coordinate patch, as can be seen from the last lines of the proof of Lemma 3.1.
The third part of the theorem, which states that minimal length paths satisfy the geodesic equation locally, is part of the standard theory for Riemannian geometry on manifolds, e.g. (Lee 1997, Sect. 4).
Appendix E: Proof of Theorem 4.1
The theorem is trivial for , so suppose with and that the first assertion holds for all . The matrix is not changed by inserting edges e with to connect trees in F, or by adding edges with , so without loss of generality we may assume F is a fully resolved tree. We may also assume there is a cherry between leaves and N since each bifurcating tree with has a cherry and permuting the labels of F results in a tree with covariance (with permutation matrix P), where positive definiteness is preserved.
Let and be the edges incident to leaves and N, respectively. Since F is bifurcating, there is exactly one edge, say , incident to and . Let . The tree obtained by deleting and leaf N and merging and to with weight has covariance , which is by induction positive definite. Using this and Sylvester’s criterion (that a matrix is positive if and only if all principal minors have positive determinant) it suffices to show . We have for all that , where
and is defined by Eq. (4.4). Note that for , and do not involve , and that . If denotes the set of permutations of , then the Leibniz formula for determinants gives
so is linear in . By symmetry of the cherry, is also linear in as well. We write . For , we have for and , so for all , and similarly for all . Furthermore, , since in that case the last two rows of coincide. Since g is linear in x and in y, respectively, we have for all , so that for all . If , we would have , but this is not allowed by the definition of .
We also need to show that the map is injective on where [F] denotes the equivalence class of . This is trivial, however, since whenever are in different equivalence classes, the matrix of distances between the leaves is different.
Appendix F: Proof of Theorem 5.1
The proof is similar to that for Theorem 3.1, and so we give a brief sketch. We consider the same path between the trees , and show that each element of is bound from above along the path. Working in the -parametrization of an orthant, Eq. (4.5) becomes
Each element of the matrix is therefore a polynomial in the elements of , and their derivatives with respect to are also polynomials. Recalling that the tangent space of at in a maximal orthant is spanned by , where ranges over the edges in that maximal orthant, Eq. (4.6) becomes
.
Applying the Cauchy-Schwartz inequality gives
The first term in this product is bounded on a geodesic path from to , since is positive definite and its eigenvalues are bound away from zero. The other two terms are also bounded from above, because the derivatives of are polynomials in , Thus for some constant C at all points along that path, and the same argument as for Theorem 3.1 shows that is finite.
Funding
Open Access funding enabled and organized by Projekt DEAL.
Footnotes
This space was first discussed by the authors at the Oberwolfach 1804 meeting “Statistics for Data with Geometric Structure” in the Schwarzwald (Black Forest) in 2018.
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Contributor Information
M. K. Garba, Email: m.k.garba1@ncl.ac.uk
T. M. W. Nye, Email: tom.nye@ncl.ac.uk
J. Lueg, Email: jonas.lueg@stud.uni-goettingen.de, Email: lueg@math.uni-goettingen.de
S. F. Huckemann, Email: huckeman@math.uni-goettingen.de
References
- Adams RH, Castoe TA. Probabilistic species tree distances: implementing the multispecies coalescent to compare species trees within the same model-based framework used to estimate them. Syst Biol. 2020;69(1):194–207. doi: 10.1093/sysbio/syz031. [DOI] [PubMed] [Google Scholar]
- Allen BL, Steel M. Subtree transfer operations and their induced metrics on evolutionary trees. Ann Comb. 2001;5(1):1–15. doi: 10.1007/s00026-001-8006-8. [DOI] [Google Scholar]
- Allman ES, Ané C, Rhodes JA. Identifiability of a Markovian model of molecular evolution with gamma-distributed rates. Adv Appl Probab. 2008;40(1):229–249. doi: 10.1239/aap/1208358894. [DOI] [Google Scholar]
- Ballmann W, Gromov M, Schroeder V. Manifolds of nonpositive curvature. Progress in mathematics. Basel: Birkhäuser; 1985. [Google Scholar]
- Bačák M. Computing medians and means in Hadamard spaces. SIAM J Optim. 2014;24(3):1542–1566. doi: 10.1137/140953393. [DOI] [Google Scholar]
- Billera L, Holmes S, Vogtman K. Geometry of the space of phylogenetic trees. Adv Appl Math. 2001;27:733–767. doi: 10.1006/aama.2001.0759. [DOI] [Google Scholar]
- Bridson MR, Haefliger A. Metric spaces of non-positive curvature. Berlin: Springer; 2011. [Google Scholar]
- Bryant D, Galtier N, Poursat M-A. Likelihood calculation in molecular phylogenetics. In: Gascuel O, editor. Mathematics of evolution and phylogeny. Oxford: Oxford University Press; 2005. pp. 33–62. [Google Scholar]
- Dryden IL, Koloydenko A, Zhou D, et al. Non-Euclidean statistics for covariance matrices, with applications to diffusion tensor imaging. Ann Appl Stat. 2009;3(3):1102–1123. doi: 10.1214/09-AOAS249. [DOI] [Google Scholar]
- Engström A, Hersh P, Sturmfels B. Toric cubes. Rendiconti del Circolo Matematico di Palermo. 2013;62(1):67–78. doi: 10.1007/s12215-013-0115-9. [DOI] [Google Scholar]
- Feragen A, Owen M, Petersen J, Wille M, Thomsen L, Dirksen A, de Bruijne M (2013) Tree-space statistics and approximations for large-scale analysis of anatomical trees. In: 23rd biennial international conference on information processing in medical imaging (IPMI) [DOI] [PubMed]
- Garba MK (2019) Information geometry for phylogenetic trees. Ph.D. thesis, School of Mathematics, Statistics and Physics, Newcastle University
- Garba MK, Nye TMW, Boys RJ. Probabilistic distances between trees. Syst Biol. 2018;67(2):320–327. doi: 10.1093/sysbio/syx080. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Gill J, Linusson S, Moulton V, Steel M. A regular decomposition of the edge-product space of phylogenetic trees. Adv Appl Math. 2008;41(2):158–176. doi: 10.1016/j.aam.2006.07.007. [DOI] [Google Scholar]
- Hansen TF, Martins EP. Translating between microevolutionary process and macroevolutionary patterns: the correlation structure of interspecific data. Evolution. 1996;50(4):1404–1417. doi: 10.1111/j.1558-5646.1996.tb03914.x. [DOI] [PubMed] [Google Scholar]
- Huelsenbeck JP, Ronquist F. MrBayes: Bayesian inference of phylogenetic trees. Bioinformatics. 2001;17(8):754–755. doi: 10.1093/bioinformatics/17.8.754. [DOI] [PubMed] [Google Scholar]
- Kim J. Slicing hyperdimensional oranges: the geometry of phylogenetic estimation. Mol Phylogenet Evol. 2000;17(1):58–75. doi: 10.1006/mpev.2000.0816. [DOI] [PubMed] [Google Scholar]
- Lee JM. Riemannian manifolds: an introduction to curvature. Berlin: Springer; 1997. [Google Scholar]
- Lenglet C, Rousson M, Deriche R, Faugeras O. Statistics on the manifold of multivariate normal distributions: theory and application to diffusion tensor MRI processing. J Math Imaging Vis. 2006;25(3):423–444. doi: 10.1007/s10851-006-6897-z. [DOI] [Google Scholar]
- Lin B, Yoshida R. Tropical Fermat-Weber points. SIAM J Discrete Math. 2018;32(2):1229–1245. doi: 10.1137/16M1071122. [DOI] [Google Scholar]
- Lin B, Monod A, Yoshida R (2018) Tropical foundations for probability and statistics on phylogenetic tree space. arXiv preprint arXiv:1805.12400
- Miller E, Owen M, Provan JS. Polyhedral computational geometry for averaging metric phylogenetic trees. Adv Appl Math. 2015;68:51–91. doi: 10.1016/j.aam.2015.04.002. [DOI] [Google Scholar]
- Moakher M. A differential geometric approach to the geometric mean of symmetric positive-definite matrices. SIAM J Matrix Anal Appl. 2005;26:735–747. doi: 10.1137/S0895479803436937. [DOI] [Google Scholar]
- Moulton V, Steel M. Peeling phylogenetic oranges. Adv Appl Math. 2004;33(4):710–727. doi: 10.1016/j.aam.2004.03.003. [DOI] [Google Scholar]
- Nye TMW. Principal components analysis in the space of phylogenetic trees. Ann Stat. 2011;39(5):2716–2739. doi: 10.1214/11-AOS915. [DOI] [Google Scholar]
- Nye T. An algorithm for constructing principal geodesics in phylogenetic treespace. IEEE ACM Trans Comput Biol. 2014;11(2):304–315. doi: 10.1109/TCBB.2014.2309599. [DOI] [PubMed] [Google Scholar]
- Nye TMW, Tang X, Weyenberg G, Yoshida R. Principal component analysis and the locus of the Fréchet mean in the space of phylogenetic trees. Biometrika. 2017;104(4):901–922. doi: 10.1093/biomet/asx047. [DOI] [PMC free article] [PubMed] [Google Scholar]
- Owen M, Provan JS. A fast algorithm for computing geodesic distances in tree space. IEEE ACM Trans Comput Biol. 2011;8(1):2–13. doi: 10.1109/TCBB.2010.3. [DOI] [PubMed] [Google Scholar]
- Rogers JS. On the consistency of maximum likelihood estimation of phylogenetic trees from nucleotide sequences. Syst Biol. 1997;46(2):354–357. doi: 10.1093/sysbio/46.2.354. [DOI] [PubMed] [Google Scholar]
- Sason I, Verdu S. f-divergence inequalities. IEEE Trans Inf Theory. 2016;62(11):5973–6006. doi: 10.1109/TIT.2016.2603151. [DOI] [Google Scholar]
- Schmidt FR, Clausen M, Cremers D (2006) Shape matching by variational computation of geodesics on a manifold. In: Joint pattern recognition symposium. Springer, pp 142–151
- Semple C, Steel M. Phylogenetics. Oxford lecture series in mathematics and its applications. Oxford: Oxford University Press; 2003. [Google Scholar]
- Skovgaard LT. A Riemannian geometry of the multivariate normal model. Scand J Stat. 1984;11(4):211–223. [Google Scholar]
- Speyer D, Sturmfels B. The tropical Grassmannian. Adv Geom. 2004;4(3):389–411. doi: 10.1515/advg.2004.023. [DOI] [Google Scholar]
- Steel MA, Penny D. Distributions of tree comparison metrics—some new results. Syst Biol. 1993;42(2):126–141. [Google Scholar]
- Willis A. Confidence sets for phylogenetic trees. J Am Stat Assoc. 2019;114(525):235–244. doi: 10.1080/01621459.2017.1395342. [DOI] [Google Scholar]
- Yang Z. Computational molecular evolution. Oxford: Oxford University Press; 2006. [Google Scholar]
- Yoshida R, Zhang L, Zhang X. Tropical principal component analysis and its application to phylogenetics. Bull Math Biol. 2019;81(2):568–597. doi: 10.1007/s11538-018-0493-4. [DOI] [PubMed] [Google Scholar]
- Zwiernik P, Smith JQ. Tree cumulants and the geometry of binary tree models. Bernoulli. 2012;18(1):290–321. doi: 10.3150/10-BEJ338. [DOI] [Google Scholar]