torchtree: Flexible Phylogenetic Model Development and Inference Using PyTorch.
Saved in:
| Title: | torchtree: Flexible Phylogenetic Model Development and Inference Using PyTorch. |
|---|---|
| Authors: | Fourment, Mathieu1 (AUTHOR), Macaulay, Matthew1 (AUTHOR), Swanepoel, Christiaan J2 (AUTHOR), Ji, Xiang3 (AUTHOR), Suchard, Marc A4 (AUTHOR), Matsen IV, Frederick A5 (AUTHOR) |
| Source: | Systematic Biology. Jan2026, Vol. 75 Issue 1, p39-51. 13p. |
| Subjects: | Phylogenetic models, Variational approach (Mathematics), Bayesian analysis, Statistical models, Automatic differentiation, Markov chain Monte Carlo |
| Abstract: | Bayesian inference has predominantly relied on the Markov chain Monte Carlo (MCMC) algorithm for many years. However, MCMC is computationally laborious, especially for complex phylogenetic models of time trees. This bottleneck has led to the search for alternatives, such as variational Bayes, which can scale better to large data sets. In this paper, we introduce torchtree , a framework written in Python that allows developers to easily implement rich phylogenetic models and algorithms using a fixed tree topology. One can either use automatic differentiation or leverage torchtree 's plug-in system to compute gradients analytically for model components for which automatic differentiation is slow. We demonstrate that the torchtree variational inference framework performs similarly to BEAST in terms of speed, and delivers promising approximation results, though accuracy varies across scenarios. Furthermore, we explore the use of the forward Kullback–Leibler (KL) divergence as an optimizing criterion for variational inference, which can handle discontinuous and nondifferentiable models. Our experiments show that inference using the forward KL divergence is frequently faster per iteration compared with the evidence lower bound (ELBO) criterion, although the ELBO-based inference may converge faster in some cases. Overall, torchtree provides a flexible and efficient framework for phylogenetic model development and inference using PyTorch. [ABSTRACT FROM AUTHOR] |
| Copyright of Systematic Biology is the property of Oxford University Press / USA and its content may not be copied or emailed to multiple sites without the copyright holder's express written permission. Additionally, content may not be used with any artificial intelligence tools or machine learning technologies. However, users may print, download, or email articles for individual use. This abstract may be abridged. No warranty is given about the accuracy of the copy. Users should refer to the original published version of the material for the full abstract. (Copyright applies to all Abstracts.) | |
| Database: | Engineering Source |
Be the first to leave a comment!
